From b970416641c21c54b70ea2a9b727513118470303 Mon Sep 17 00:00:00 2001
From: TrisNol <tristan.nolde@yahoo.de>
Date: Tue, 29 Aug 2023 18:50:33 +0200
Subject: [PATCH] checkpoint(data-ingestion): Add method to mongo service to
 fetch companies with yearly results

---
 .../utils/mongo/company_mongo_service.py                 | 9 +++++++++
 1 file changed, 9 insertions(+)

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_.