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