Switch Flask server in production mode

Makefile

@@ -1,4 +1,4 @@
-COMPOSE=docker compose -f docker-compose.dev.yml
+COMPOSE=docker compose -f dev/docker-compose.dev.yaml
 
 .DEFAULT_GOAL := wizard
 ## Run wizard
@@ -17,6 +17,12 @@ run:
 	$(COMPOSE) rm -f myelectricaldata_import
 	python3 main.py action=run
 
+## Connect to EnedisGateway container : DEV
+run-production:
+	$(COMPOSE) stop myelectricaldata_import
+	$(COMPOSE) rm -f myelectricaldata_import
+	python3 main.py action=run env=production
+
 ## Start docker conatiners for dev
 up:
 	@echo "Start docker container for dev"
@@ -64,9 +70,9 @@ git_push:
 generate-dependencies:
 	cd app; pip-compile -o requirements.txt pyproject.toml; cd -
 
-## Create github pre release (dev)
-create-release-dev:
-	python3 main.py action=create_pre_release
+clean: generate-dependencies
+	docker image rm -f myelectricaldata_import_myelectricaldata_import
+#	docker build ./
 
 ## Create github release (prod)
 create-release:

app/main.py

@@ -1,22 +1,26 @@
-import sys
-
 from flask import Flask, request, send_file
 from flask_apscheduler import APScheduler
+from waitress import serve
 
 from config import cycle_minimun
 from dependencies import *
-from models.config import Config, get_version
-from models.jobs import Job
 from models.ajax import Ajax
+from models.config import Config, get_version
 from models.database import Database
 from models.influxdb import InfluxDB
+from models.jobs import Job
 from models.log import Log
 from models.mqtt import Mqtt
 from templates.index import Index
 from templates.usage_point_id import UsagePointId
 
 LOG = Log()
 
+if "DEV" in os.environ and os.getenv("DEV"):
+    LOG.title_warning("Run in Development mode")
+else:
+    LOG.title("Run in production mode")
+
 if "APPLICATION_PATH_DATA" in os.environ:
     APPLICATION_PATH_DATA = os.getenv("APPLICATION_PATH_DATA")
 else:
@@ -143,6 +147,7 @@ def usage_point_id(usage_point_id):
     def lock():
         return str(DB.lock_status())
 
+
     @APP.route("/gateway_status", methods=['GET'])
     @APP.route("/gateway_status/", methods=['GET'])
     def gateway_status():
@@ -172,6 +177,7 @@ def account_status(usage_point_id):
     def import_all_data(usage_point_id):
         return Ajax(usage_point_id).import_data()
 
+
     @APP.route("/import/<usage_point_id>/<target>", methods=['GET'])
     @APP.route("/import/<usage_point_id>/<target>", methods=['GET'])
     def import_data(usage_point_id, target):
@@ -207,5 +213,7 @@ def whitelist_data(usage_point_id, target, date):
     def fetch_data(usage_point_id, target, date):
         return Ajax(usage_point_id).fetch(target, date)
 
-
-    APP.run(host="0.0.0.0", port=5000, debug=False, use_reloader=True)
+    if "DEV" in os.environ and os.getenv("DEV"):
+        APP.run(host="0.0.0.0", port=5000, debug=False, use_reloader=True)
+    else:
+        serve(APP, host="0.0.0.0", port=5000)

app/models/jobs.py

@@ -1,12 +1,12 @@
 import __main__ as app
-import sys
+import time
 import traceback
 
 from models.config import get_version
 from models.database import Database
-from models.export_mqtt import ExportMqtt
 from models.export_home_assistant import HomeAssistant
 from models.export_influxdb import ExportInfluxDB
+from models.export_mqtt import ExportMqtt
 from models.log import Log
 from models.query_address import Address
 from models.query_contract import Contract
@@ -26,8 +26,15 @@ def __init__(self, usage_point_id=None):
         self.mqtt_config = self.config.mqtt_config()
         self.home_assistant_config = self.config.home_assistant_config()
         self.influxdb_config = self.config.influxdb_config()
+        self.wait_job_start = 10
 
     def job_import_data(self, target=None):
+        app.LOG.title("Démarrage du job d'importation dans 10s")
+        i = self.wait_job_start
+        while i > 0:
+            app.LOG.log(f" => {i}s")
+            time.sleep(1)
+            i = i - 1
         if app.DB.lock_status():
             return {
                 "status": False,
@@ -99,14 +106,16 @@ def job_import_data(self, target=None):
                             if target == "mqtt" or target is None:
                                 ExportMqtt(self.usage_point_id, "consumption").contract()
                                 ExportMqtt(self.usage_point_id, "consumption").address()
-                                if hasattr(self.usage_point_config, "consumption") and self.usage_point_config.consumption:
+                                if hasattr(self.usage_point_config,
+                                           "consumption") and self.usage_point_config.consumption:
                                     ExportMqtt(self.usage_point_id, "consumption").daily_annual(
                                         self.usage_point_config.consumption_price_base
                                     )
                                     ExportMqtt(self.usage_point_id, "consumption").daily_linear(
                                         self.usage_point_config.consumption_price_base
                                     )
-                                if hasattr(self.usage_point_config, "production") and self.usage_point_config.production:
+                                if hasattr(self.usage_point_config,
+                                           "production") and self.usage_point_config.production:
                                     ExportMqtt(self.usage_point_id, "production").daily_annual(
                                         self.usage_point_config.production_price
                                     )
@@ -155,21 +164,25 @@ def job_import_data(self, target=None):
                         if "enable" in self.influxdb_config and self.influxdb_config["enable"]:
                             # app.INFLUXDB.purge_influxdb()
                             if target == "influxdb" or target is None:
-                                if hasattr(self.usage_point_config, "consumption") and self.usage_point_config.consumption:
+                                if hasattr(self.usage_point_config,
+                                           "consumption") and self.usage_point_config.consumption:
                                     ExportInfluxDB(self.usage_point_id).daily(
                                         self.usage_point_config.consumption_price_base,
                                     )
-                                if hasattr(self.usage_point_config, "production") and self.usage_point_config.production:
+                                if hasattr(self.usage_point_config,
+                                           "production") and self.usage_point_config.production:
                                     ExportInfluxDB(self.usage_point_id).daily(
                                         self.usage_point_config.production_price,
                                         "production"
                                     )
-                                if hasattr(self.usage_point_config, "consumption_detail") and self.usage_point_config.consumption_detail:
+                                if hasattr(self.usage_point_config,
+                                           "consumption_detail") and self.usage_point_config.consumption_detail:
                                     ExportInfluxDB(self.usage_point_id).detail(
                                         self.usage_point_config.consumption_price_hp,
                                         self.usage_point_config.consumption_price_hc
                                     )
-                                if hasattr(self.usage_point_config, "production_detail") and self.usage_point_config.production_detail:
+                                if hasattr(self.usage_point_config,
+                                           "production_detail") and self.usage_point_config.production_detail:
                                     ExportInfluxDB(self.usage_point_id).detail(
                                         self.usage_point_config.production_price,
                                         "production_detail"

app/models/stat.py

@@ -38,6 +38,7 @@ def __init__(self, usage_point_id):
         self.value_yesterday_evolution = 0
         self.value_current_month_evolution = 0
         self.value_peak_offpeak_percent_hp_vs_hc = 0
+        self.value_monthly_evolution = 0
 
     def daily(self, index=0, measurement_direction="consumption"):
         begin = datetime.combine(self.yesterday_date - timedelta(days=index), datetime.min.time())

app/pyproject.toml

@@ -18,7 +18,8 @@ dependencies = [
     "jinja2",
     "art",
     "SQLAlchemy",
-    "alembic"
+    "alembic",
+    "waitress"
 ]
 
 [project.optional-dependencies]

app/requirements.txt

@@ -83,6 +83,8 @@ urllib3==1.26.12
     # via
     #   influxdb-client
     #   requests
+waitress==2.1.2
+    # via myelectricaldata_import (pyproject.toml)
 werkzeug==2.2.2
     # via flask
 

dependencies.py

@@ -6,7 +6,7 @@
 from InquirerPy.base import Choice
 from packaging.version import parse as parse_version
 
-docker_compose = "docker-compose -f docker-compose.dev.yml"
+docker_compose = "docker-compose -f dev/docker-compose.dev.yaml"
 
 
 def cmd(cmd, path="./"):
@@ -56,17 +56,20 @@ def wizard():
         app.LOG.error("Good bye!!")
 
 
-def run(debug=False):
+def run(dev=False, debug=False):
     if debug:
         app.LOG.title(["Boot DynAPI in debug mode", "CTRL + C to exit"])
     else:
         app.LOG.title(["Boot DynAPI", "CTRL + C to exit"])
     mode_debug = ""
+    mode_dev = ""
     if debug:
         mode_debug = "-e DEBUG=true"
+    if dev:
+        mode_dev = "-e DEV=true"
     command = (
-        f"{docker_compose} run -p 5000:5000 "
-        f"{mode_debug} myelectricaldata_import"
+        f"{docker_compose} run -p 5000:5000"
+        f"{mode_debug} {mode_dev} myelectricaldata_import"
     )
     app.LOG.log(command)
     os.system(command)

docker-compose.dev.yml → dev/docker-compose.dev.yaml

@@ -1,7 +1,7 @@
 version: "3.9"
 services:
   myelectricaldata_import:
-    build: .
+    build: ./../
     environment:
       TZ: Europe/Paris
     ports:
@@ -10,15 +10,15 @@ services:
       - influxdb
       - mosquitto
     volumes:
-      - ./data:/data
-      - ./app:/app
+      - ./../data:/data
+      - ./../app:/app
 
   influxdb:
     image: influxdb:2.4
     ports:
       - '8086:8086'
     volumes:
-      - ./data/influxdb:/var/lib/influxdb2
+      - ./../data/influxdb:/var/lib/influxdb2
     healthcheck:
       test: ["CMD", "curl", "-f", "http://influxdb:8086"]
       interval: 25s
@@ -40,8 +40,8 @@ services:
     volumes:
       - /etc/localtime:/etc/localtime:ro
       - /etc/timezone:/etc/timezone:ro
-      - ./data/mosquitto:/data
-      - ./mosquitto:/mosquitto/config/
+      - ./../data/mosquitto:/data
+      - ./../mosquitto:/mosquitto/config/
     ports:
       - 1883:1883
       - 9001:9001
@@ -53,7 +53,7 @@ services:
     links:
       - mosquitto
     volumes:
-      - ./data/:/mqtt-explorer/config
+      - ./../data/:/mqtt-explorer/config
     ports:
       - 4000:4000
 

docker-compose.yaml

@@ -0,0 +1,11 @@
+version: "3.9"
+services:
+  myelectricaldata:
+    image: m4dm4rtig4n/myelectricaldata:latest
+    restart: unless-stopped
+    volumes:
+      - ./data:/data
+    environment:
+      TZ: Europe/Paris
+    ports:
+      - '5000:5000'

main.py

@@ -14,15 +14,19 @@
 
 LOG = Log()
 
+production = False
+if "env" in PARAMS and PARAMS["env"] == "production":
+    production = True
+
 if __name__ == "__main__":
     if ACTION == "wizard":
         wizard()
     elif ACTION == "run":
-        run()
-        # run()
+        if production:
+            run()
+        else:
+            run(dev=True)
     elif ACTION == "debug":
         run(debug=True)
-    elif ACTION == "create_pre_release":
-        create_release("preprod")
     elif ACTION == "create_release":
         create_release()