From 77711d8a2f2feba66bfb569755909ef7e645bf23 Mon Sep 17 00:00:00 2001
From: TrisNol <tristan.nolde@yahoo.de>
Date: Mon, 25 Sep 2023 19:34:10 +0200
Subject: [PATCH] feat: Add simple wrapper to update particual financial
 entries

---
 .../apps/fix_company_financials.py            | 23 +++++++++++++++++++
 tests/apps/fix_company_financials_test.py     |  6 +++++
 2 files changed, 29 insertions(+)
 create mode 100644 src/aki_prj23_transparenzregister/apps/fix_company_financials.py
 create mode 100644 tests/apps/fix_company_financials_test.py

diff --git a/src/aki_prj23_transparenzregister/apps/fix_company_financials.py b/src/aki_prj23_transparenzregister/apps/fix_company_financials.py
new file mode 100644
index 0000000..0fb2aed
--- /dev/null
+++ b/src/aki_prj23_transparenzregister/apps/fix_company_financials.py
@@ -0,0 +1,23 @@
+"""Fix fincancial data of particular companies identified by their ID."""
+from aki_prj23_transparenzregister.apps.enrich_company_financials import work
+from aki_prj23_transparenzregister.config.config_providers import JsonFileConfigProvider
+from aki_prj23_transparenzregister.utils.mongo.company_mongo_service import (
+    CompanyMongoService,
+)
+from aki_prj23_transparenzregister.utils.mongo.connector import MongoConnector
+
+if __name__ == "__main__":
+    config_provider = JsonFileConfigProvider("./secrets.json")
+
+    mongo_connector = MongoConnector(config_provider.get_mongo_connection_string())
+    company_service = CompanyMongoService(mongo_connector)
+
+    entries = [
+        "649f16a4e198338c3b442ab1",
+        "649f16a5e198338c3b442b0a",
+        "649f16a5e198338c3b442ac6",
+    ]
+
+    companies = [company_service.get_by_object_id(entry) for entry in entries]
+    for company in companies:
+        work(company, company_service)
diff --git a/tests/apps/fix_company_financials_test.py b/tests/apps/fix_company_financials_test.py
new file mode 100644
index 0000000..3684269
--- /dev/null
+++ b/tests/apps/fix_company_financials_test.py
@@ -0,0 +1,6 @@
+"""test aki_prj23_transparenzregister.apps.fix_company_financials."""
+from aki_prj23_transparenzregister.apps import fix_company_financials
+
+
+def test_main() -> None:
+    assert fix_company_financials