(feat): Insertion to mongoDB done

utilsv2/nordic.py

@@ -10,7 +10,11 @@ from utilsv2.log import logger
 
 logger = logging.getLogger(__name__)
 
+type evtype = dict[str, Any]
+type sttype = dict[str, Any]
 
+
+# INFO: Don't think we really need this
 class Mag:
     def __init__(self, mag: float, type: str, agency: str):
         self.mag = mag
@@ -24,7 +28,7 @@ class Mag:
         return json.dumps(self, default=lambda o: o.__dict__, sort_keys=True, indent=4)
 
 
-def parse(event: list[str]) -> dict[str, Any]:
+def parse_event(event: list[str]) -> evtype:
     # nordic must always have the first line a type 1 line
     # but a type 1 line can have the id ommited if it's the first line
     # if event[0][-1] != "1" or event[0][-1] != " ":
@@ -35,16 +39,52 @@ def parse(event: list[str]) -> dict[str, Any]:
     for line in event:
         toParse[line[-1]].append(line)
 
+    _ret = {}
+
     for k, v in toParse.items():
         match k:
             case "1":
-                parse_type_1(v)
+                aux = parse_type_1(v)
+                if aux:
+                    _ret.update(aux)
             case "3":
-                parse_type_3(v)
+                _ret.update(parse_type_3(v))
+            case "6":
+                _ret.update(parse_type_6(v))
+            case "E":
+                _ret.update(parse_type_e(v))
+            case "I":
+                _ret.update(parse_type_i(v))
             case _:
                 pass
+    print(_ret)
+    return _ret
 
-    return {}
+
+def parse_stations_V1(lines: list[str], event_id: int) -> sttype:
+    _ret = {"ID": event_id, "stations": {}}
+    for st in lines:
+        try:
+            ampl = float(st[35:40])
+        except ValueError:
+            ampl = None
+
+        station = st[1:6].strip()
+        if station not in _ret["stations"].keys():
+            _ret["stations"][station] = []
+
+        _ret["stations"][station].append(
+            {
+                "Component": st[6:9].strip(),
+                "I": None if st[9] == " " else st[9],
+                "Time": parse_dt(st[18:30], True).strftime("%H:%M:%S.%f%z"),
+                "Phase": st[10:15].strip(),
+                "Weigth": None if st[15] == " " else st[15],
+                "Amplitude": ampl,
+            }
+        )
+
+    return _ret
 
 
 def parse_type_1(lines: list[str]) -> dict[str, Any] | None:
@@ -52,8 +92,8 @@ def parse_type_1(lines: list[str]) -> dict[str, Any] | None:
     for line in lines:
         if "Date" not in line1.keys():
             dt = parse_dt(line[:21])
-            dist_ind = line[20]
-            event_id = line[21]
+            dist_ind = line[21]
+            event_id = line[22]
             lat = float(line[24:31])
             long = float(line[30:39])
             depth = float(line[38:44])
@@ -78,6 +118,7 @@ def parse_type_1(lines: list[str]) -> dict[str, Any] | None:
 
 def parse_type_3(lines: list[str]) -> dict[str, Any]:
     comments = {"Sentido": "", "Regiao": "", "VZ": None, "SZ": None, "FE": None}
+
     for line in lines:
         if line.startswith(" SENTIDO"):
             aux = line[:-2].split(":", maxsplit=1)
@@ -94,20 +135,31 @@ def parse_type_3(lines: list[str]) -> dict[str, Any]:
                 else:
                     comments["Regiao"] = item[1:]
 
-    print(comments)
     return comments
 
 
-def parse_type_6():
-    pass
+def parse_type_6(lines: list[str]) -> dict[str, list[str]]:
+    _ret = {"Wavename": []}
+    for line in lines:
+        _ret["Wavename"].append(line[:-2].strip())
+    return _ret
 
 
-def parse_type_e():
-    pass
+def parse_type_e(lines: list[str]) -> dict[str, int]:
+    err = {}
+    for line in lines:
+        gap = int(line[5:8])
+        err["GAP"] = gap
+
+    return err
 
 
-def parse_type_7():
-    pass
+def parse_type_i(lines: list[str]) -> dict[str, int]:
+    aux = {}
+    for line in lines:
+        aux["ID"] = int(line[60:75])
+
+    return aux
 
 
 def parse_dt(_text: str, isStation=False) -> datetime | time:
@@ -125,9 +177,9 @@ def parse_dt(_text: str, isStation=False) -> datetime | time:
         )
         return dt
     else:
-        h = int(_text[11:13])
-        m = int(_text[13:15])
-        s_ms = int(float(_text[16:20]) * 1000)
+        h = int(_text[:2])
+        m = int(_text[2:4])
+        s_ms = int(float(_text[5:]) * 1000)
         s = s_ms // 1000
         s_ms = s_ms % 1000
         dt = time(hour=h, minute=m, second=s, microsecond=s_ms)