Feat add organization and catalog fixtures in init data

ops/ansible/environments/staging/assets/initdata.yml

@@ -97,3 +97,6 @@ datasets:
       last_updated_at: null
       url: "https://www.data.gouv.fr/fr/datasets/catalogue-des-enquetes-realisees-par-la-dares-2012-572205/"
       tag_ids: []
+# créer une organization
+# créer un cataloge // Pour les relier utiliser un siret qui existe
+# Lier un dataset au catalogue

tests/tools/test_initdata.py

@@ -25,6 +25,9 @@ async def test_initdata_empty(tmp_path: Path) -> None:
         users: []
         tags: []
         datasets: []
+        organizations: []
+        catalogs: []
+
         """
     )
     await initdata.main(path)
@@ -53,6 +56,9 @@ async def test_initdata_env_password_invalid(
               email: [email protected]
               password: __env__
         datasets: []
+        organizations: []
+        catalogs: []
+
         """
     )
 
@@ -78,6 +84,9 @@ async def test_initdata_env_password(
               password: __env__
         tags: []
         datasets: []
+        organizations: []
+        catalogs: []
+
         """
     )
 
@@ -115,7 +124,9 @@ async def test_repo_initdata(
     num_users = 2
     num_tags = 7
     num_datasets = 4
-    num_entities = num_users + num_tags + num_datasets
+    num_organization = 1
+    num_catalog = 1
+    num_entities = num_users + num_tags + num_datasets + num_catalog + num_organization
 
     await initdata.main(path, no_input=True)
     captured = capsys.readouterr()

tools/initdata.py

@@ -12,13 +12,17 @@
 from pydantic import BaseModel, ValidationError, parse_raw_as
 
 from server.application.auth.commands import CreatePasswordUser
+from server.application.catalogs.commands import CreateCatalog
 from server.application.datasets.commands import CreateDataset, UpdateDataset
+from server.application.organizations.commands import CreateOrganization
 from server.application.tags.commands import CreateTag
 from server.config.di import bootstrap, resolve
 from server.domain.auth.entities import UserRole
 from server.domain.auth.repositories import PasswordUserRepository
+from server.domain.catalogs.repositories import CatalogRepository
 from server.domain.datasets.entities import Dataset
 from server.domain.datasets.repositories import DatasetRepository
+from server.domain.organizations.repositories import OrganizationRepository
 from server.domain.tags.repositories import TagRepository
 from server.seedwork.application.messages import MessageBus
 
@@ -130,6 +134,48 @@ def _get_dataset_attr(dataset: Dataset, attr: str) -> Any:
     print(f"{success('created')}: {create_command!r}")
 
 
+async def handle_organizations(item: dict) -> None:
+    bus = resolve(MessageBus)
+    repository = resolve(OrganizationRepository)
+
+    siret = item["params"]["siret"]
+
+    existing_organization = await repository.get_by_siret(siret)
+
+    if existing_organization is not None:
+
+        organization_repr = (
+            f"Organization(siret={siret!r}, name={item['params']['name']!r}, ...)"
+        )
+        print(f"{info('ok')}: {organization_repr}")
+        return
+
+    create_command = CreateOrganization(**item["params"])
+
+    await bus.execute(create_command)
+    print(f"{success('created')}: {create_command!r}")
+
+
+async def handle_catalogs(item: dict) -> None:
+    bus = resolve(MessageBus)
+    repository = resolve(CatalogRepository)
+
+    siret = item["params"]["organization_siret"]
+
+    existing_organization = await repository.get_by_siret(siret)
+
+    if existing_organization is not None:
+
+        organization_repr = f"Catalog(siret={siret!r}, ...)"
+        print(f"{info('ok')}: {organization_repr}")
+        return
+
+    create_command = CreateCatalog(**item["params"])
+
+    await bus.execute(create_command)
+    print(f"{success('created')}: {create_command!r}")
+
+
 async def main(path: pathlib.Path, reset: bool = False, no_input: bool = False) -> None:
     with path.open() as f:
         spec = yaml.safe_load(f)
@@ -151,6 +197,16 @@ async def main(path: pathlib.Path, reset: bool = False, no_input: bool = False)
     for item in spec["datasets"]:
         await handle_dataset(item, reset=reset)
 
+    print("\n", ruler("Organizations"))
+
+    for item in spec["organizations"]:
+        await handle_organizations(item)
+
+    print("\n", ruler("Catalogs"))
+
+    for item in spec["catalogs"]:
+        await handle_catalogs(item)
+
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()

tools/initdata.yml

@@ -89,7 +89,7 @@ datasets:
       description: |
         Par région, les ressources ci-dessous recensent les restaurants, brasseries et cafétérias dans le réseau des CROUS.
       service: Direction des données du CROUS
-      geographical_coverage: Monde  # Fake, for E2E testing purposes
+      geographical_coverage: Monde # Fake, for E2E testing purposes
       formats:
         - file_tabular
       technical_source: Système d'information central du CROUS
@@ -120,3 +120,11 @@ datasets:
       url: "https://www.data.gouv.fr/fr/datasets/catalogue-des-enquetes-realisees-par-la-dares-2012-572205/"
       license: Licence Ouverte
       tag_ids: []
+organizations:
+  - params:
+      name: "Ministère de la Culture"
+      siret: "44229377500031" # Siret of https://www.societe.com/societe/fake-442293775.html
+
+catalogs:
+  - params:
+      organization_siret: "44229377500031" # See "Ministère de la Culture"