add ssl support

print errors to stderror
update README

Author: chocomega

README.md

@@ -25,6 +25,7 @@ pip install -r requirements.txt
 Rename the `script_config.example.yaml` file to `script_config.yaml` and edit it to suit your needs. All the following keys are mandatory:
 
 - `ha_url`: URL of your Home Assistant instance
+- `ha_use_ssl`: Specify if Home Assistant uses SSL or not
 - `ha_access_token`: Long-lived access token for your Home Assistant instance
 - `med_cache_db_path`: Path to the cache database of MyElectricalData
 - `med_config_path`: Path to the configuration file of MyElectricalData
@@ -51,7 +52,7 @@ shell_command:
     statistics_delete_all: python statistics_importer/statistics_importer.py -d
 ```
 
-You can then create an automation to call the service `shell_command.statistics_import` periodically or when MyElectricalData cache is updated.
+You can then create an automation calling the service `shell_command.statistics_import` periodically or when MyElectricalData cache is updated.
 
 ## Warning
 - Home Assistant may take some time to display the newly created statistics. Please wait, they will eventually show up in the UI.

script_config.example.yaml

@@ -3,6 +3,9 @@
 # URL of Home Assistant
 ha_url: "localhost:8123"
 
+# Specify if Home Assistant uses SSL or not
+ha_use_ssl: false
+
 # Long-Lived access token from Home Assistant
 ha_access_token: "xxxxxx"
 

statistics_importer.py

@@ -1,4 +1,4 @@
-__version__ = '1.0.0'
+__version__ = '2.0.0'
 
 import aiohttp
 import argparse
@@ -33,6 +33,7 @@ class Config:
     ha_access_token: str
     med_cache_db_path: str
     med_config_path: str
+    ha_use_ssl: bool = False
 
     @classmethod
     def load(cls, path: str = os.path.abspath("script_config.yaml")) -> "Config":
@@ -423,15 +424,15 @@ async def main(args: argparse.Namespace) -> int:
     start_time = time.time()
 
     try:
-        # Read config.yaml
+        # Read script_config.yaml
         config = Config.load()
 
         # Read MyElectricalData config.yaml
         with open(os.path.abspath(config.med_config_path)) as file:
             med_config = yaml.safe_load(file)
 
         # Create the WebSocket connection
-        url = f"ws://{config.ha_url}/api/websocket"
+        url = f"{'wss' if config.ha_use_ssl else 'ws'}://{config.ha_url}/api/websocket"
         print("Connecting to websocket at", url)
 
         async with aiohttp.ClientSession() as session:
@@ -451,8 +452,8 @@ async def main(args: argparse.Namespace) -> int:
 
     except Exception as e:
         tb = traceback.extract_tb(e.__traceback__)
-        print(f"ERROR on line {tb[0].lineno}: {tb[0].line}")
-        print(f"{type(e).__name__} - {e}")
+        print(f"ERROR on line {tb[0].lineno}: {tb[0].line}", file=sys.stderr)
+        print(f"{type(e).__name__} - {e}", file=sys.stderr)
         return 1
 
     finally: