fix(sia): handles files with final 'a', 'b' and 'c'

AlertaDengue · Sep 20, 2023 · 3e73ece · 3e73ece
1 parent 0300efb
commit 3e73ece
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/docs/source/tutorials/Dengue.ipynb b/docs/source/tutorials/Dengue.ipynb
@@ -108,7 +108,7 @@
     }
    ],
    "source": [
-    "munrj = gpd.read_file('../data/rj-municipalities.json')\n",
+    "munrj = gpd.read_file('data/rj-municipalities.json')\n",
     "munrj.plot();"
    ]
   },

diff --git a/pysus/ftp/databases/sia.py b/pysus/ftp/databases/sia.py
@@ -66,8 +66,10 @@ def describe(self, file: File) -> dict:
 
     def format(self, file: File) -> tuple:
         if file.extension.upper() in [".DBC", ".DBF"]:
-            year, month = file.name[-4:-2], file.name[-2:]
-            group, _uf = file.name[:-6].upper(), file.name[-6:-4].upper()
+            digits = ''.join([d for d in file.name if d.isdigit()])
+            chars, _ = file.name.split(digits)
+            year, month = digits[:2], digits[2:]
+            group, _uf = chars[:-2].upper(), chars[-2:].upper()
             return group, _uf, zfill_year(year), month
         return ()