diff --git a/src/aki_prj23_transparenzregister/utils/mongo/company_mongo_service.py b/src/aki_prj23_transparenzregister/utils/mongo/company_mongo_service.py index cce8a25..2c7d1db 100644 --- a/src/aki_prj23_transparenzregister/utils/mongo/company_mongo_service.py +++ b/src/aki_prj23_transparenzregister/utils/mongo/company_mongo_service.py @@ -71,6 +71,15 @@ class CompanyMongoService: self.collection.find({"$or": [{"yearly_results": {"$exists": False}}]}) ) + def get_where_yearly_results(self) -> list[dict]: + """Get a list of all companies with valid yearly_results (interesting entries for data loader). + + Returns: + list[dict]: List of companies + """ + with self.lock: + return list(self.collection.find({"yearly_results": {"$gt": {}}})) + def insert(self, company: Company) -> InsertOneResult: """_summary_.