Refactor methods

sportorg/libs/sfr/sfrximporter.py

@@ -9,12 +9,12 @@
     ResultManual,
     ResultStatus,
     Split,
-    SystemType,
+    SystemType, Person,
 )
 from sportorg.utils.time import hhmmss_to_time
 
 
-def import_sfrx(source):
+def import_sfrx(source: str):
     sfr_csv = sfrxparser.parse(source)
     obj = memory.race()
 
@@ -176,7 +176,7 @@ def import_sfrx(source):
         )
 
 
-def convert_bib(bib) -> int:
+def convert_bib(bib: str) -> int:
     if not bib.isdigit():
         return 0
 
@@ -186,7 +186,7 @@ def convert_bib(bib) -> int:
         return int(bib[0] + str(int(bib[1:])))
 
 
-def sfr_qual_to_sportorg(value) -> int:
+def sfr_qual_to_sportorg(value: str) -> int:
     return {
         "": 0,
         "0": 0,
@@ -203,7 +203,7 @@ def sfr_qual_to_sportorg(value) -> int:
     }[value]
 
 
-def set_property(person, key, value) -> None:
+def set_property(person: Person, key: str, value: str) -> None:
     if value == "":
         return
 

sportorg/libs/sfr/sfrxparser.py

@@ -14,7 +14,7 @@ def __init__(self, data=None):
         self._data = [] if data is None else data
         self._splits = []
 
-    def parse(self, source):
+    def parse(self, source: str):
         f = codecs.open(source, "r", "UTF-8")
 
         for row in f:
@@ -111,6 +111,6 @@ def settings(self):
         return self._settings
 
 
-def parse(source) -> SFRXParser:
+def parse(source: str) -> SFRXParser:
     parser = SFRXParser()
     return parser.parse(source)