feat: Add simple wrapper to update particual financial entries

This commit is contained in:
TrisNol 2023-09-25 19:34:10 +02:00
parent 7b5cf16e49
commit 77711d8a2f
2 changed files with 29 additions and 0 deletions

View File

@ -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)

View File

@ -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