emhass 0.11.1__py3-none-any.whl → 0.11.3__py3-none-any.whl

emhass/web_server.py

@@ -1,51 +1,78 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-from flask import Flask, request, make_response, render_template
+import argparse
+import json
+import logging
+import os
+import pickle
+import re
+import threading
+from distutils.util import strtobool
+from importlib.metadata import PackageNotFoundError, version
+from pathlib import Path
+
+import yaml
+from flask import Flask, make_response, request
 from jinja2 import Environment, PackageLoader
-from requests import get
 from waitress import serve
-from importlib.metadata import version, PackageNotFoundError
-from pathlib import Path
-import os, json, argparse, pickle, yaml, logging,  re, threading
-from distutils.util import strtobool
 
-from emhass.command_line import set_input_data_dict
-from emhass.command_line import perfect_forecast_optim, dayahead_forecast_optim, naive_mpc_optim
-from emhass.command_line import forecast_model_fit, forecast_model_predict, forecast_model_tune, weather_forecast_cache
-from emhass.command_line import regressor_model_fit, regressor_model_predict
-from emhass.command_line import publish_data, continual_publish
-from emhass.utils import get_injection_dict, get_injection_dict_forecast_model_fit, \
-    get_injection_dict_forecast_model_tune,  build_config, build_secrets, build_params, \
-    param_to_config, build_legacy_config_params
+from emhass.command_line import (
+    continual_publish,
+    dayahead_forecast_optim,
+    forecast_model_fit,
+    forecast_model_predict,
+    forecast_model_tune,
+    naive_mpc_optim,
+    perfect_forecast_optim,
+    publish_data,
+    regressor_model_fit,
+    regressor_model_predict,
+    set_input_data_dict,
+    weather_forecast_cache,
+)
+from emhass.utils import (
+    build_config,
+    build_legacy_config_params,
+    build_params,
+    build_secrets,
+    get_injection_dict,
+    get_injection_dict_forecast_model_fit,
+    get_injection_dict_forecast_model_tune,
+    param_to_config,
+)
 
 # Define the Flask instance
 app = Flask(__name__)
 emhass_conf = {}
 
+
 def checkFileLog(refString=None) -> bool:
     """
     Check logfile for error, anything after string match if provided.
 
-    :param refString: String to reduce log area to check for errors. Use to reduce log to check anything after string match (ie. an action). 
+    :param refString: String to reduce log area to check for errors. Use to reduce log to check anything after string match (ie. an action).
     :type refString: str
     :return: Boolean return if error was found in logs
     :rtype: bool
 
     """
-    if (refString is not None): 
-       logArray = grabLog(refString) #grab reduced log array (everything after string match)
-    else: 
-        if ((emhass_conf['data_path'] / 'actionLogs.txt')).exists():
-            with open(str(emhass_conf['data_path'] / 'actionLogs.txt'), "r") as fp:
-                    logArray = fp.readlines()
+    if refString is not None:
+        logArray = grabLog(
+            refString
+        )  # grab reduced log array (everything after string match)
+    else:
+        if (emhass_conf["data_path"] / "actionLogs.txt").exists():
+            with open(str(emhass_conf["data_path"] / "actionLogs.txt"), "r") as fp:
+                logArray = fp.readlines()
         else:
             app.logger.debug("Unable to obtain actionLogs.txt")
     for logString in logArray:
-            if (logString.split(' ', 1)[0] == "ERROR"):
-                return True     
+        if logString.split(" ", 1)[0] == "ERROR":
+            return True
     return False
 
+
 def grabLog(refString) -> list:
     """
     Find string in logs, append all lines after into list to return.
@@ -58,31 +85,33 @@ def grabLog(refString) -> list:
     """
     isFound = []
     output = []
-    if ((emhass_conf['data_path'] / 'actionLogs.txt')).exists():
-            with open(str(emhass_conf['data_path'] / 'actionLogs.txt'), "r") as fp:
-                    logArray = fp.readlines()
-            # Find all string matches, log key (line Number) in isFound
-            for x in range(len(logArray)-1):
-                if (re.search(refString,logArray[x])):
-                   isFound.append(x)
-            if len(isFound) != 0:
-                # Use last item in isFound to extract action logs  
-                for x in range(isFound[-1],len(logArray)): 
-                    output.append(logArray[x])
+    if (emhass_conf["data_path"] / "actionLogs.txt").exists():
+        with open(str(emhass_conf["data_path"] / "actionLogs.txt"), "r") as fp:
+            logArray = fp.readlines()
+        # Find all string matches, log key (line Number) in isFound
+        for x in range(len(logArray) - 1):
+            if re.search(refString, logArray[x]):
+                isFound.append(x)
+        if len(isFound) != 0:
+            # Use last item in isFound to extract action logs
+            for x in range(isFound[-1], len(logArray)):
+                output.append(logArray[x])
     return output
 
+
 # Clear the log file
 def clearFileLog():
     """
     Clear the contents of the log file (actionLogs.txt)
 
     """
-    if ((emhass_conf['data_path'] / 'actionLogs.txt')).exists():
-        with open(str(emhass_conf['data_path'] / 'actionLogs.txt'), "w") as fp:
-            fp.truncate()    
+    if (emhass_conf["data_path"] / "actionLogs.txt").exists():
+        with open(str(emhass_conf["data_path"] / "actionLogs.txt"), "w") as fp:
+            fp.truncate()
 
-@app.route('/')
-@app.route('/index')
+
+@app.route("/")
+@app.route("/index")
 def index():
     """
     Render initial index page and serve to web server.
@@ -91,29 +120,31 @@ def index():
     """
     app.logger.info("EMHASS server online, serving index.html...")
     # Load HTML template
-    file_loader = PackageLoader('emhass', 'templates')
+    file_loader = PackageLoader("emhass", "templates")
     env = Environment(loader=file_loader)
-    #check if index.html exists
-    if 'index.html' not in env.list_templates():
+    # check if index.html exists
+    if "index.html" not in env.list_templates():
         app.logger.error("Unable to find index.html in emhass module")
-        return make_response(["ERROR: unable to find index.html in emhass module"],404)
-    template = env.get_template('index.html')
+        return make_response(["ERROR: unable to find index.html in emhass module"], 404)
+    template = env.get_template("index.html")
     # Load cached dict (if exists), to present generated plot tables
-    if (emhass_conf['data_path'] / 'injection_dict.pkl').exists():
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "rb") as fid:
+    if (emhass_conf["data_path"] / "injection_dict.pkl").exists():
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "rb") as fid:
             injection_dict = pickle.load(fid)
     else:
-        app.logger.info("The data container dictionary is empty... Please launch an optimization task")
-        injection_dict={}
+        app.logger.info(
+            "The data container dictionary is empty... Please launch an optimization task"
+        )
+        injection_dict = {}
 
-    # replace {{basename}} in html template html with path root  
+    # replace {{basename}} in html template html with path root
     # basename = request.headers.get("X-Ingress-Path", "")
     # return make_response(template.render(injection_dict=injection_dict, basename=basename))
-    
+
     return make_response(template.render(injection_dict=injection_dict))
 
 
-@app.route('/configuration')
+@app.route("/configuration")
 def configuration():
     """
     Configuration page actions:
@@ -122,41 +153,46 @@ def configuration():
     """
     app.logger.info("serving configuration.html...")
     # Load HTML template
-    file_loader = PackageLoader('emhass', 'templates')
+    file_loader = PackageLoader("emhass", "templates")
     env = Environment(loader=file_loader)
-    #check if configuration.html exists
-    if 'configuration.html' not in env.list_templates():
+    # check if configuration.html exists
+    if "configuration.html" not in env.list_templates():
         app.logger.error("Unable to find configuration.html in emhass module")
-        return make_response(["ERROR: unable to find configuration.html in emhass module"],404)
-    template = env.get_template('configuration.html')
+        return make_response(
+            ["ERROR: unable to find configuration.html in emhass module"], 404
+        )
+    template = env.get_template("configuration.html")
     return make_response(template.render(config=params))
 
 
-@app.route('/template', methods=['GET'])
+@app.route("/template", methods=["GET"])
 def template_action():
     """
-    template page actions: 
+    template page actions:
     Render and serve template html
 
     """
     app.logger.info(" >> Sending rendered template table data")
-    file_loader = PackageLoader('emhass', 'templates')
+    file_loader = PackageLoader("emhass", "templates")
     env = Environment(loader=file_loader)
     # Check if template.html exists
-    if 'template.html' not in env.list_templates():
+    if "template.html" not in env.list_templates():
         app.logger.error("Unable to find template.html in emhass module")
-        return make_response(["WARNING: unable to find template.html in emhass module"],404)
-    template = env.get_template('template.html')
-    if (emhass_conf['data_path'] / 'injection_dict.pkl').exists():
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "rb") as fid:
+        return make_response(
+            ["WARNING: unable to find template.html in emhass module"], 404
+        )
+    template = env.get_template("template.html")
+    if (emhass_conf["data_path"] / "injection_dict.pkl").exists():
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "rb") as fid:
             injection_dict = pickle.load(fid)
     else:
         app.logger.warning("Unable to obtain plot data from injection_dict.pkl")
         app.logger.warning("Try running an launch an optimization task")
-        injection_dict={}        
+        injection_dict = {}
     return make_response(template.render(injection_dict=injection_dict))
 
-@app.route('/get-config', methods=['GET'])
+
+@app.route("/get-config", methods=["GET"])
 def parameter_get():
     """
     Get request action that builds, formats and sends config as json (config.json format)
@@ -164,21 +200,27 @@ def parameter_get():
     """
     app.logger.debug("Obtaining current saved parameters as config")
     # Build config from all possible sources (inc. legacy yaml config)
-    config = build_config(emhass_conf,app.logger,emhass_conf["defaults_path"],emhass_conf["config_path"],emhass_conf["legacy_config_path"])
+    config = build_config(
+        emhass_conf,
+        app.logger,
+        emhass_conf["defaults_path"],
+        emhass_conf["config_path"],
+        emhass_conf["legacy_config_path"],
+    )
     if type(config) is bool and not config:
-        return make_response(["failed to retrieve default config file"],500)
+        return make_response(["failed to retrieve default config file"], 500)
     # Format parameters in config with params (converting legacy json parameters from options.json if any)
-    params = build_params(emhass_conf,{},config,app.logger)
+    params = build_params(emhass_conf, {}, config, app.logger)
     if type(params) is bool and not params:
-        return make_response(["Unable to obtain associations file"],500)
+        return make_response(["Unable to obtain associations file"], 500)
     # Covert formatted parameters from params back into config.json format
-    return_config = param_to_config(params,app.logger)
+    return_config = param_to_config(params, app.logger)
     # Send config
-    return make_response(return_config,201)
+    return make_response(return_config, 201)
 
 
 # Get default Config
-@app.route('/get-config/defaults', methods=['GET'])
+@app.route("/get-config/defaults", methods=["GET"])
 def config_get():
     """
     Get request action, retrieves and sends default configuration
@@ -186,21 +228,21 @@ def config_get():
     """
     app.logger.debug("Obtaining default parameters")
     # Build config, passing only default file
-    config = build_config(emhass_conf,app.logger,emhass_conf["defaults_path"])
+    config = build_config(emhass_conf, app.logger, emhass_conf["defaults_path"])
     if type(config) is bool and not config:
-        return make_response(["failed to retrieve default config file"],500)
+        return make_response(["failed to retrieve default config file"], 500)
     # Format parameters in config with params
-    params = build_params(emhass_conf,{},config,app.logger)
+    params = build_params(emhass_conf, {}, config, app.logger)
     if type(params) is bool and not params:
-        return make_response(["Unable to obtain associations file"],500)
+        return make_response(["Unable to obtain associations file"], 500)
     # Covert formatted parameters from params back into config.json format
-    return_config = param_to_config(params,app.logger)
+    return_config = param_to_config(params, app.logger)
     # Send params
-    return make_response(return_config,201)
+    return make_response(return_config, 201)
 
 
 # Get YAML-to-JSON config
-@app.route('/get-json', methods=['POST'])
+@app.route("/get-json", methods=["POST"])
 def json_convert():
     """
     Post request action, receives yaml config (config_emhass.yaml or EMHASS-Add-on config page) and converts to config json format.
@@ -212,268 +254,318 @@ def json_convert():
 
     # If filed to Parse YAML
     if yaml_config is None:
-        return make_response(["failed to Parse YAML from data"],400)
+        return make_response(["failed to Parse YAML from data"], 400)
     # Test YAML is legacy config format (from config_emhass.yaml)
-    test_legacy_config = build_legacy_config_params(emhass_conf,yaml_config, app.logger)    
+    test_legacy_config = build_legacy_config_params(
+        emhass_conf, yaml_config, app.logger
+    )
     if test_legacy_config:
         yaml_config = test_legacy_config
     # Format YAML to params (format params. check if params match legacy option.json format)
-    params = build_params(emhass_conf,{},yaml_config,app.logger)
+    params = build_params(emhass_conf, {}, yaml_config, app.logger)
     if type(params) is bool and not params:
-        return make_response(["Unable to obtain associations file"],500)
+        return make_response(["Unable to obtain associations file"], 500)
     # Covert formatted parameters from params back into config.json format
-    config = param_to_config(params,app.logger)
+    config = param_to_config(params, app.logger)
     # convert json to str
     config = json.dumps(config)
 
     # Send params
-    return make_response(config,201)
+    return make_response(config, 201)
+
 
-@app.route('/set-config', methods=['POST'])
+@app.route("/set-config", methods=["POST"])
 def parameter_set():
     """
     Receive JSON config, and save config to file (config.json and param.pkl)
 
     """
     config = {}
-    if not emhass_conf['defaults_path']:
-        return make_response(["Unable to Obtain defaults_path from emhass_conf"],500)
-    if not emhass_conf['config_path']:
-        return make_response(["Unable to Obtain config_path from emhass_conf"],500)
-    
+    if not emhass_conf["defaults_path"]:
+        return make_response(["Unable to Obtain defaults_path from emhass_conf"], 500)
+    if not emhass_conf["config_path"]:
+        return make_response(["Unable to Obtain config_path from emhass_conf"], 500)
+
     # Load defaults as a reference point (for sorting) and a base to override
-    if os.path.exists(emhass_conf['defaults_path']) and Path(emhass_conf['defaults_path']).is_file():
-        with emhass_conf['defaults_path'].open('r') as data:
+    if (
+        os.path.exists(emhass_conf["defaults_path"])
+        and Path(emhass_conf["defaults_path"]).is_file()
+    ):
+        with emhass_conf["defaults_path"].open("r") as data:
             config = json.load(data)
     else:
-        app.logger.warning("Unable to obtain default config. only parameters passed from request will be saved to config.json")
+        app.logger.warning(
+            "Unable to obtain default config. only parameters passed from request will be saved to config.json"
+        )
 
     # Retrieve sent config json
     request_data = request.get_json(force=True)
 
     # check if data is empty
     if len(request_data) == 0:
-        return make_response(["failed to retrieve config json"],400)
-    
+        return make_response(["failed to retrieve config json"], 400)
+
     # Format config by converting to params (format params. check if params match legacy option.json format. If so format)
-    params = build_params(emhass_conf,params_secrets,request_data,app.logger)
+    params = build_params(emhass_conf, params_secrets, request_data, app.logger)
     if type(params) is bool and not params:
-        return make_response(["Unable to obtain associations file"],500)
-    
+        return make_response(["Unable to obtain associations file"], 500)
+
     # Covert formatted parameters from params back into config.json format.
     # Overwrite existing default parameters in config
-    config.update(param_to_config(params,app.logger))
+    config.update(param_to_config(params, app.logger))
 
     # Save config to config.json
-    if os.path.exists(emhass_conf['config_path'].parent):
-        with emhass_conf['config_path'].open('w') as f:
+    if os.path.exists(emhass_conf["config_path"].parent):
+        with emhass_conf["config_path"].open("w") as f:
             json.dump(config, f, indent=4)
-    else: 
-        return make_response(["Unable to save config file"],500)
+    else:
+        return make_response(["Unable to save config file"], 500)
     request_data
 
-    # Save params with updated config 
-    if os.path.exists(emhass_conf['data_path']):
-        with open(str(emhass_conf['data_path'] / 'params.pkl'), "wb") as fid:
-            pickle.dump((config_path, build_params(emhass_conf,params_secrets,config,app.logger)), fid)
-    else: 
-        return make_response(["Unable to save params file, missing data_path"],500)
-    
+    # Save params with updated config
+    if os.path.exists(emhass_conf["data_path"]):
+        with open(str(emhass_conf["data_path"] / "params.pkl"), "wb") as fid:
+            pickle.dump(
+                (
+                    config_path,
+                    build_params(emhass_conf, params_secrets, config, app.logger),
+                ),
+                fid,
+            )
+    else:
+        return make_response(["Unable to save params file, missing data_path"], 500)
+
     app.logger.info("Saved parameters from webserver")
-    return make_response({},201)
+    return make_response({}, 201)
 
-@app.route('/action/<action_name>', methods=['POST'])
+
+@app.route("/action/<action_name>", methods=["POST"])
 def action_call(action_name):
     """
     Receive Post action, run action according to passed slug(action_name) (e.g. /action/publish-data)
 
     :param action_name: Slug/Action string corresponding to which action to take
     :type action_name: String
-    
+
     """
     # Setting up parameters
     # Params
-    if (emhass_conf['data_path'] / 'params.pkl').exists():
-        with open(str(emhass_conf['data_path'] / 'params.pkl'), "rb") as fid:
-            emhass_conf['config_path'], params = pickle.load(fid)
+    ActionStr = " >> Obtaining params: "
+    app.logger.info(ActionStr)
+    if (emhass_conf["data_path"] / "params.pkl").exists():
+        with open(str(emhass_conf["data_path"] / "params.pkl"), "rb") as fid:
+            emhass_conf["config_path"], params = pickle.load(fid)
+            # Set local costfun variable
+            if params.get("optim_conf", None) is not None:
+                costfun = params["optim_conf"].get("costfun", "profit")
             params = json.dumps(params)
     else:
         app.logger.error("Unable to find params.pkl file")
         return make_response(grabLog(ActionStr), 400)
     # Runtime
-    runtimeparams = request.get_json(force=True,silent=True)
+    runtimeparams = request.get_json(force=True, silent=True)
     if runtimeparams is not None:
-        if runtimeparams != '{}':
+        if runtimeparams != "{}":
             app.logger.info("Passed runtime parameters: " + str(runtimeparams))
     else:
         app.logger.warning("Unable to parse runtime parameters")
-        runtimeparams = {} 
+        runtimeparams = {}
     runtimeparams = json.dumps(runtimeparams)
 
     # weather-forecast-cache (check before set_input_data_dict)
-    if action_name == 'weather-forecast-cache':
+    if action_name == "weather-forecast-cache":
         ActionStr = " >> Performing weather forecast, try to caching result"
         app.logger.info(ActionStr)
         weather_forecast_cache(emhass_conf, params, runtimeparams, app.logger)
-        msg = f'EMHASS >> Weather Forecast has run and results possibly cached... \n'
+        msg = "EMHASS >> Weather Forecast has run and results possibly cached... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
 
     ActionStr = " >> Setting input data dict"
     app.logger.info(ActionStr)
-    input_data_dict = set_input_data_dict(emhass_conf, costfun, 
-        params, runtimeparams, action_name, app.logger)
+    app.logger.warning(costfun)
+    input_data_dict = set_input_data_dict(
+        emhass_conf, costfun, params, runtimeparams, action_name, app.logger
+    )
     if not input_data_dict:
         return make_response(grabLog(ActionStr), 400)
-    
+
     # If continual_publish is True, start thread with loop function
-    if len(continual_publish_thread) == 0 and input_data_dict['retrieve_hass_conf'].get('continual_publish',False):
+    if len(continual_publish_thread) == 0 and input_data_dict["retrieve_hass_conf"].get(
+        "continual_publish", False
+    ):
         # Start Thread
-        continualLoop = threading.Thread(name='continual_publish',target=continual_publish,args=[input_data_dict,entity_path,app.logger])
+        continualLoop = threading.Thread(
+            name="continual_publish",
+            target=continual_publish,
+            args=[input_data_dict, entity_path, app.logger],
+        )
         continualLoop.start()
-        continual_publish_thread.append(continualLoop)      
+        continual_publish_thread.append(continualLoop)
 
     # Run action based on POST request
     # If error in log when running action, return actions log (list) as response. (Using ActionStr as a reference of the action start in the log)
     # publish-data
-    if action_name == 'publish-data':
+    if action_name == "publish-data":
         ActionStr = " >> Publishing data..."
         app.logger.info(ActionStr)
         _ = publish_data(input_data_dict, app.logger)
-        msg = f'EMHASS >> Action publish-data executed... \n'
+        msg = "EMHASS >> Action publish-data executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
     # perfect-optim
-    elif action_name == 'perfect-optim':
+    elif action_name == "perfect-optim":
         ActionStr = " >> Performing perfect optimization..."
         app.logger.info(ActionStr)
         opt_res = perfect_forecast_optim(input_data_dict, app.logger)
         injection_dict = get_injection_dict(opt_res)
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "wb") as fid:
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "wb") as fid:
             pickle.dump(injection_dict, fid)
-        msg = f'EMHASS >> Action perfect-optim executed... \n'
+        msg = "EMHASS >> Action perfect-optim executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
     # dayahead-optim
-    elif action_name == 'dayahead-optim':
+    elif action_name == "dayahead-optim":
         ActionStr = " >> Performing dayahead optimization..."
         app.logger.info(ActionStr)
         opt_res = dayahead_forecast_optim(input_data_dict, app.logger)
         injection_dict = get_injection_dict(opt_res)
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "wb") as fid:
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "wb") as fid:
             pickle.dump(injection_dict, fid)
-        msg = f'EMHASS >> Action dayahead-optim executed... \n'
+        msg = "EMHASS >> Action dayahead-optim executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
     # naive-mpc-optim
-    elif action_name == 'naive-mpc-optim':
+    elif action_name == "naive-mpc-optim":
         ActionStr = " >> Performing naive MPC optimization..."
         app.logger.info(ActionStr)
         opt_res = naive_mpc_optim(input_data_dict, app.logger)
         injection_dict = get_injection_dict(opt_res)
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "wb") as fid:
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "wb") as fid:
             pickle.dump(injection_dict, fid)
-        msg = f'EMHASS >> Action naive-mpc-optim executed... \n'
+        msg = "EMHASS >> Action naive-mpc-optim executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
     # forecast-model-fit
-    elif action_name == 'forecast-model-fit':
+    elif action_name == "forecast-model-fit":
         ActionStr = " >> Performing a machine learning forecast model fit..."
         app.logger.info(ActionStr)
         df_fit_pred, _, mlf = forecast_model_fit(input_data_dict, app.logger)
-        injection_dict = get_injection_dict_forecast_model_fit(
-            df_fit_pred, mlf)
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "wb") as fid:
+        injection_dict = get_injection_dict_forecast_model_fit(df_fit_pred, mlf)
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "wb") as fid:
             pickle.dump(injection_dict, fid)
-        msg = f'EMHASS >> Action forecast-model-fit executed... \n'
+        msg = "EMHASS >> Action forecast-model-fit executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
     # forecast-model-predict
-    elif action_name == 'forecast-model-predict':
+    elif action_name == "forecast-model-predict":
         ActionStr = " >> Performing a machine learning forecast model predict..."
         app.logger.info(ActionStr)
         df_pred = forecast_model_predict(input_data_dict, app.logger)
         if df_pred is None:
             return make_response(grabLog(ActionStr), 400)
-        table1 = df_pred.reset_index().to_html(classes='mystyle', index=False)
+        table1 = df_pred.reset_index().to_html(classes="mystyle", index=False)
         injection_dict = {}
-        injection_dict['title'] = '<h2>Custom machine learning forecast model predict</h2>'
-        injection_dict['subsubtitle0'] = '<h4>Performed a prediction using a pre-trained model</h4>'
-        injection_dict['table1'] = table1
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "wb") as fid:
+        injection_dict["title"] = (
+            "<h2>Custom machine learning forecast model predict</h2>"
+        )
+        injection_dict["subsubtitle0"] = (
+            "<h4>Performed a prediction using a pre-trained model</h4>"
+        )
+        injection_dict["table1"] = table1
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "wb") as fid:
             pickle.dump(injection_dict, fid)
-        msg = f'EMHASS >> Action forecast-model-predict executed... \n'
+        msg = "EMHASS >> Action forecast-model-predict executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
     # forecast-model-tune
-    elif action_name == 'forecast-model-tune':
+    elif action_name == "forecast-model-tune":
         ActionStr = " >> Performing a machine learning forecast model tune..."
         app.logger.info(ActionStr)
-        df_pred_optim, mlf = forecast_model_tune(input_data_dict, app.logger)    
-        if df_pred_optim is None or  mlf is None:
+        df_pred_optim, mlf = forecast_model_tune(input_data_dict, app.logger)
+        if df_pred_optim is None or mlf is None:
             return make_response(grabLog(ActionStr), 400)
-        injection_dict = get_injection_dict_forecast_model_tune(
-            df_pred_optim, mlf)
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "wb") as fid:
+        injection_dict = get_injection_dict_forecast_model_tune(df_pred_optim, mlf)
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "wb") as fid:
             pickle.dump(injection_dict, fid)
-        msg = f'EMHASS >> Action forecast-model-tune executed... \n'
+        msg = "EMHASS >> Action forecast-model-tune executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
     # regressor-model-fit
-    elif action_name == 'regressor-model-fit':
+    elif action_name == "regressor-model-fit":
         ActionStr = " >> Performing a machine learning regressor fit..."
         app.logger.info(ActionStr)
         regressor_model_fit(input_data_dict, app.logger)
-        msg = f'EMHASS >> Action regressor-model-fit executed... \n'
+        msg = "EMHASS >> Action regressor-model-fit executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
-   # regressor-model-predict
-    elif action_name == 'regressor-model-predict':
+    # regressor-model-predict
+    elif action_name == "regressor-model-predict":
         ActionStr = " >> Performing a machine learning regressor predict..."
         app.logger.info(ActionStr)
         regressor_model_predict(input_data_dict, app.logger)
-        msg = f'EMHASS >> Action regressor-model-predict executed... \n'
+        msg = "EMHASS >> Action regressor-model-predict executed... \n"
         if not checkFileLog(ActionStr):
             return make_response(msg, 201)
         return make_response(grabLog(ActionStr), 400)
     # Else return error
     else:
         app.logger.error("ERROR: passed action is not valid")
-        msg = f'EMHASS >> ERROR: Passed action is not valid... \n'
+        msg = "EMHASS >> ERROR: Passed action is not valid... \n"
         return make_response(msg, 400)
 
+
 if __name__ == "__main__":
     # Parsing arguments
     parser = argparse.ArgumentParser()
-    parser.add_argument('--url', type=str, help='The URL to your Home Assistant instance, ex the external_url in your hass configuration')
-    parser.add_argument('--key', type=str, help='Your access key. If using EMHASS in standalone this should be a Long-Lived Access Token')
-    parser.add_argument('--no_response', type=strtobool, default='False', help='This is set if json response errors occur')
+    parser.add_argument(
+        "--url",
+        type=str,
+        help="The URL to your Home Assistant instance, ex the external_url in your hass configuration",
+    )
+    parser.add_argument(
+        "--key",
+        type=str,
+        help="Your access key. If using EMHASS in standalone this should be a Long-Lived Access Token",
+    )
+    parser.add_argument(
+        "--no_response",
+        type=strtobool,
+        default="False",
+        help="This is set if json response errors occur",
+    )
     args = parser.parse_args()
 
     # Pre formatted config parameters
-    config = {} 
+    config = {}
     # Secrets
     params_secrets = {}
     # Built parameters (formatted config + secrets)
-    params = None 
-    
-    # Find env's, not not set defaults 
+    params = None
+
+    # Find env's, not not set defaults
     DATA_PATH = os.getenv("DATA_PATH", default="/app/data/")
     ROOT_PATH = os.getenv("ROOT_PATH", default=str(Path(__file__).parent))
-    CONFIG_PATH = os.getenv('CONFIG_PATH', default="/share/config.json")
-    OPTIONS_PATH = os.getenv('OPTIONS_PATH', default="/data/options.json") 
-    DEFAULTS_PATH = os.getenv('DEFAULTS_PATH', default=ROOT_PATH +"/data/config_defaults.json")
-    ASSOCIATIONS_PATH = os.getenv('ASSOCIATIONS_PATH', default=ROOT_PATH + "/data/associations.csv")
-    LEGACY_CONFIG_PATH = os.getenv("LEGACY_CONFIG_PATH", default="/app/config_emhass.yaml")
+    CONFIG_PATH = os.getenv("CONFIG_PATH", default="/share/config.json")
+    OPTIONS_PATH = os.getenv("OPTIONS_PATH", default="/data/options.json")
+    DEFAULTS_PATH = os.getenv(
+        "DEFAULTS_PATH", default=ROOT_PATH + "/data/config_defaults.json"
+    )
+    ASSOCIATIONS_PATH = os.getenv(
+        "ASSOCIATIONS_PATH", default=ROOT_PATH + "/data/associations.csv"
+    )
+    LEGACY_CONFIG_PATH = os.getenv(
+        "LEGACY_CONFIG_PATH", default="/app/config_emhass.yaml"
+    )
 
     # Define the paths
     config_path = Path(CONFIG_PATH)
@@ -484,50 +576,62 @@ if __name__ == "__main__":
     data_path = Path(DATA_PATH)
     root_path = Path(ROOT_PATH)
     # Add paths to emhass_conf
-    emhass_conf['config_path'] = config_path
-    emhass_conf['options_path'] = options_path
-    emhass_conf['defaults_path'] = defaults_path
-    emhass_conf['associations_path'] = associations_path
-    emhass_conf['legacy_config_path'] = legacy_config_path 
-    emhass_conf['data_path'] = data_path
-    emhass_conf['root_path'] = root_path 
+    emhass_conf["config_path"] = config_path
+    emhass_conf["options_path"] = options_path
+    emhass_conf["defaults_path"] = defaults_path
+    emhass_conf["associations_path"] = associations_path
+    emhass_conf["legacy_config_path"] = legacy_config_path
+    emhass_conf["data_path"] = data_path
+    emhass_conf["root_path"] = root_path
 
     # Combine parameters from configuration sources (if exists)
-    config.update(build_config(emhass_conf,app.logger,defaults_path,config_path,legacy_config_path))
+    config.update(
+        build_config(
+            emhass_conf, app.logger, defaults_path, config_path, legacy_config_path
+        )
+    )
     if type(config) is bool and not config:
         raise Exception("Failed to find default config")
 
-    costfun = os.getenv('LOCAL_COSTFUN', config.get('costfun', 'profit'))
-    logging_level = os.getenv('LOGGING_LEVEL', config.get('logging_level','INFO'))
+    # Set local variables
+    costfun = os.getenv("LOCAL_COSTFUN", config.get("costfun", "profit"))
+    logging_level = os.getenv("LOGGING_LEVEL", config.get("logging_level", "INFO"))
     # Temporary set logging level if debug
     if logging_level == "DEBUG":
         app.logger.setLevel(logging.DEBUG)
-        
+
     ## Secrets
     argument = {}
     if args.url:
-        argument['url'] = args.url
+        argument["url"] = args.url
     if args.key:
-        argument['key'] = args.key
+        argument["key"] = args.key
     # Combine secrets from ENV, Arguments/ARG, Secrets file (secrets_emhass.yaml), options (options.json from addon configuration file) and/or Home Assistant Standalone API (if exist)
-    emhass_conf, secrets = build_secrets(emhass_conf,app.logger,argument,options_path,os.getenv('SECRETS_PATH', default='/app/secrets_emhass.yaml'), bool(args.no_response))
+    emhass_conf, secrets = build_secrets(
+        emhass_conf,
+        app.logger,
+        argument,
+        options_path,
+        os.getenv("SECRETS_PATH", default="/app/secrets_emhass.yaml"),
+        bool(args.no_response),
+    )
     params_secrets.update(secrets)
 
-    server_ip = params_secrets.get("server_ip","0.0.0.0")
+    server_ip = params_secrets.get("server_ip", "0.0.0.0")
 
-    # Check if data path exists 
-    if not os.path.isdir(emhass_conf['data_path']):
-        app.logger.warning("Unable to find data_path: " + str(emhass_conf['data_path']))
+    # Check if data path exists
+    if not os.path.isdir(emhass_conf["data_path"]):
+        app.logger.warning("Unable to find data_path: " + str(emhass_conf["data_path"]))
         if os.path.isdir(Path("/app/data/")):
-            emhass_conf['data_path'] = Path("/app/data/")
+            emhass_conf["data_path"] = Path("/app/data/")
         else:
-            Path(root_path / "data/").mkdir(parents=True, exist_ok=True) 
-            emhass_conf['data_path'] = root_path / "data/"
-        app.logger.info("data_path has been set to " + str(emhass_conf['data_path']))
+            Path(root_path / "data/").mkdir(parents=True, exist_ok=True)
+            emhass_conf["data_path"] = root_path / "data/"
+        app.logger.info("data_path has been set to " + str(emhass_conf["data_path"]))
 
     # Initialize this global dict
-    if (emhass_conf['data_path'] / 'injection_dict.pkl').exists():
-        with open(str(emhass_conf['data_path'] / 'injection_dict.pkl'), "rb") as fid:
+    if (emhass_conf["data_path"] / "injection_dict.pkl").exists():
+        with open(str(emhass_conf["data_path"] / "injection_dict.pkl"), "rb") as fid:
             injection_dict = pickle.load(fid)
     else:
         injection_dict = None
@@ -535,22 +639,28 @@ if __name__ == "__main__":
     # Build params from config and param_secrets (migrate params to correct config catagories), save result to params.pkl
     params = build_params(emhass_conf, params_secrets, config, app.logger)
     if type(params) is bool:
-        raise Exception("A error has occurred while building params")   
-    
-    if os.path.exists(str(emhass_conf['data_path'])): 
-        with open(str(emhass_conf['data_path'] / 'params.pkl'), "wb") as fid:
+        raise Exception("A error has occurred while building params")
+    # Update params with local variables
+    params["optim_conf"]["costfun"] = costfun
+    params["optim_conf"]["logging_level"] = logging_level
+
+    # Save params to file for later reference
+    if os.path.exists(str(emhass_conf["data_path"])):
+        with open(str(emhass_conf["data_path"] / "params.pkl"), "wb") as fid:
             pickle.dump((config_path, params), fid)
-    else: 
-        raise Exception("missing: " + str(emhass_conf['data_path']))   
+    else:
+        raise Exception("missing: " + str(emhass_conf["data_path"]))
 
     # Define loggers
-    ch = logging.StreamHandler() 
-    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+    ch = logging.StreamHandler()
+    formatter = logging.Formatter(
+        "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    )
     ch.setFormatter(formatter)
     # Action file logger
-    fileLogger = logging.FileHandler(str(emhass_conf['data_path'] / 'actionLogs.txt')) 
-    formatter = logging.Formatter('%(levelname)s - %(name)s - %(message)s')
-    fileLogger.setFormatter(formatter) # add format to Handler
+    fileLogger = logging.FileHandler(str(emhass_conf["data_path"] / "actionLogs.txt"))
+    formatter = logging.Formatter("%(levelname)s - %(name)s - %(message)s")
+    fileLogger.setFormatter(formatter)  # add format to Handler
     if logging_level == "DEBUG":
         app.logger.setLevel(logging.DEBUG)
         ch.setLevel(logging.DEBUG)
@@ -573,30 +683,34 @@ if __name__ == "__main__":
         fileLogger.setLevel(logging.DEBUG)
     app.logger.propagate = False
     app.logger.addHandler(ch)
-    app.logger.addHandler(fileLogger)   
+    app.logger.addHandler(fileLogger)
     # Clear Action File logger file, ready for new instance
     clearFileLog()
 
-    # If entity_path exists, remove any entity/metadata files 
-    entity_path = emhass_conf['data_path'] / "entities"
-    if os.path.exists(entity_path): 
+    # If entity_path exists, remove any entity/metadata files
+    entity_path = emhass_conf["data_path"] / "entities"
+    if os.path.exists(entity_path):
         entity_pathContents = os.listdir(entity_path)
         if len(entity_pathContents) > 0:
             for entity in entity_pathContents:
                 os.remove(entity_path / entity)
-        
+
     # Initialise continual publish thread list
     continual_publish_thread = []
-    
+
     # Launch server
-    port = int(os.environ.get('PORT', 5000))
-    app.logger.info("Launching the emhass webserver at: http://"+server_ip+":"+str(port))
-    app.logger.info("Home Assistant data fetch will be performed using url: "+params_secrets['hass_url'])
-    app.logger.info("The data path is: "+str(emhass_conf['data_path']))
-    app.logger.info("The logging is: "+str(logging_level))
+    port = int(os.environ.get("PORT", 5000))
+    app.logger.info(
+        "Launching the emhass webserver at: http://" + server_ip + ":" + str(port)
+    )
+    app.logger.info(
+        "Home Assistant data fetch will be performed using url: "
+        + params_secrets["hass_url"]
+    )
+    app.logger.info("The data path is: " + str(emhass_conf["data_path"]))
+    app.logger.info("The logging is: " + str(logging_level))
     try:
-        app.logger.info("Using core emhass version: "+version('emhass'))
+        app.logger.info("Using core emhass version: " + version("emhass"))
     except PackageNotFoundError:
         app.logger.info("Using development emhass version")
     serve(app, host=server_ip, port=port, threads=8)
-