fix: include group on sinan listing

Author: luabida

pysus/api/_impl/databases.py

@@ -267,6 +267,7 @@ def _query():
                     "name": r.path.split("/")[-1],
                     "path": r.path,
                     "dataset": r.dataset.name if r.dataset else None,
+                    "group": r.group.name if r.group else None,
                     "year": r.year,
                     "month": r.month,
                     "state": r.state,

pysus/api/ftp/databases.py

@@ -403,12 +403,55 @@ class SINAN(Dataset):
     ]
 
     group_definitions: dict[str, str] = {
-        "DENG": "Dengue",
-        "ZIKA": "Zika Vírus",
+        "ACBI": "Acidente de trabalho com material biológico",
+        "ACGR": "Acidente de trabalho",
+        "ANIM": "Acidente por Animais Peçonhentos",
+        "ANTR": "Atendimento Antirrabico",
+        "BOTU": "Botulismo",
+        "CANC": "Cancêr relacionado ao trabalho",
+        "CHAG": "Doença de Chagas Aguda",
         "CHIK": "Febre de Chikungunya",
+        "COLE": "Cólera",
+        "COQU": "Coqueluche",
+        "DENG": "Dengue",
+        "DERM": "Dermatoses ocupacionais",
+        "DIFT": "Difteria",
+        "ESQU": "Esquistossomose",
+        "EXAN": "Doença exantemáticas",
+        "FMAC": "Febre Maculosa",
+        "FTIF": "Febre Tifóide",
         "HANS": "Hanseníase",
+        "HANT": "Hantavirose",
+        "HEPA": "Hepatites Virais",
+        "IEXO": "Intoxicação Exógena",
+        "INFL": "Influenza Pandêmica",
+        "LEIV": "Leishmaniose Visceral",
+        "LEPT": "Leptospirose",
+        "LERD": "LER/Dort",
+        "LTAN": "Leishmaniose Tegumentar Americana",
+        "MALA": "Malária",
+        "MENI": "Meningite",
+        "MENT": "Transtornos mentais relacionados ao trabalho",
+        "NTRA": "Notificação de Tracoma",
+        "PAIR": "Perda auditiva por ruído relacionado ao trabalho",
+        "PEST": "Peste",
+        "PFAN": "Paralisia Flácida Aguda",
+        "PNEU": "Pneumoconioses realacionadas ao trabalho",
+        "RAIV": "Raiva",
+        "SDTA": "Surto Doenças Transmitidas por Alimentos",
+        "SIFA": "Sífilis Adquirida",
+        "SIFC": "Sífilis Congênita",
+        "SIFG": "Sífilis em Gestante",
+        "SRC": "Síndrome da Rubéola Congênia",
+        "TETA": "Tétano Acidental",
+        "TETN": "Tétano Neonatal",
+        "TOXC": "Toxoplasmose Congênita",
+        "TOXG": "Toxoplasmose Gestacional",
+        "TRAC": "Inquérito de Tracoma",
         "TUBE": "Tuberculose",
-        "ANIM": "Acidente por Animais Peçonhentos",
+        "VARC": "Varicela",
+        "VIOL": "Violência doméstica, sexual e/ou outras violências",
+        "ZIKA": "Zika Vírus",
     }
 
     @property

pysus/api/models.py

@@ -327,6 +327,7 @@ async def search(self, **kwargs) -> list[BaseRemoteFile]:
 
 class BaseRemoteDataset(BaseRemoteObject, SearchableMixin, ABC):
     client: BaseRemoteClient = Field(exclude=True)
+    group_definitions: dict[str, str] = {}
     _content: Sequence[BaseRemoteGroup | BaseRemoteFile] | None = PrivateAttr(
         default=None
     )