feat: add max_date to Home Assistant WS import

config.exemple.yaml

@@ -17,6 +17,7 @@ home_assistant_ws: # FOR ENERGY TAB
   ssl: true
   token: HOME_ASSISTANT_TOKEN_GENERATE_IN_PROFILE_TABS_(BOTTOM)
   url: myhomeassistant.domain.fr
+  max_date: "2021-06-01"
 ssl:
   gateway: true
   certfile: ""

src/models/database.py

@@ -1131,10 +1131,26 @@
             table = ProductionDetail
             relation = UsagePoints.relation_production_detail
         sort = asc("date") if order_dir == "desc" else desc("date")
-        if begin is None or end is None:
+        if begin is None and end is None:
             return self.session.scalars(
                 select(table).join(relation).where(table.usage_point_id == usage_point_id).order_by(sort)
             ).all()
+        elif begin is not None and end is None:
+            return self.session.scalars(
+                select(table)
+                .join(relation)
+                .where(table.usage_point_id == usage_point_id)
+                .filter(table.date >= begin)
+                .order_by(sort)
+            ).all()
+        elif end is not None and begin is None:
+            return self.session.scalars(
+                select(table)
+                .join(relation)
+                .where(table.usage_point_id == usage_point_id)
+                .filter(table.date <= end)
+                .order_by(sort)
+            ).all()
         else:
             return self.session.scalars(
                 select(table)

src/models/export_home_assistant_ws.py

@@ -143,7 +143,12 @@
             if self.usage_point_id_config.consumption_detail:
                 logging.info("Consommation")
                 measure_type = "consumption"
-                detail = DB.get_detail_all(self.usage_point_id, order_dir="desc")
+                if "max_date" in self.config:
+                    logging.warn(f"WARNING : Max date détecter {self.config['max_date']}")
+                    begin = datetime.strptime(self.config["max_date"], "%Y-%m-%d")
+                    detail = DB.get_detail_all(begin=begin, usage_point_id=self.usage_point_id, order_dir="desc")
+                else:
+                    detail = DB.get_detail_all(self.usage_point_id, order_dir="desc")
 
                 cost = 0
                 last_year = None