'Refactored by Sourcery'

cleder · Dec 7, 2022 · bee8bac · bee8bac
1 parent d7d2939
commit bee8bac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/fastkml/times.py b/fastkml/times.py
@@ -71,7 +71,7 @@ def parse_str(self, datestr: str) -> List[Union[datetime, str]]:
             year = int(datestr.split("-")[0])
             month = int(datestr.split("-")[1])
             dt = datetime(year, month, day)
-        elif len(datestr) in [8, 10]:
+        elif len(datestr) in {8, 10}:
             resolution = "date"
             dt = dateutil.parser.parse(datestr)
         elif len(datestr) > 10:

diff --git a/fastkml/views.py b/fastkml/views.py
@@ -100,9 +100,7 @@ def timestamp(self, dt):
 
     @property
     def begin(self):
-        if self._timespan is None:
-            return None
-        return self._timespan.begin[0]
+        return None if self._timespan is None else self._timespan.begin[0]
 
     @begin.setter
     def begin(self, dt) -> None:
@@ -229,11 +227,9 @@ def from_element(self, element):
         if tilt is not None:
             self.tilt = tilt.text
         altitude_mode = element.find(f"{self.ns}altitudeMode")
-        if altitude_mode is not None:
-            self.altitude_mode = altitude_mode.text
-        else:
+        if altitude_mode is None:
             altitude_mode = element.find(f"{gx.NS}altitudeMode")
-            self.altitude_mode = altitude_mode.text
+        self.altitude_mode = altitude_mode.text
         timespan = element.find(f"{self.ns}TimeSpan")
         if timespan is not None:
             s = TimeSpan(self.ns)