Migrate envertech portalmonitor from optparse to argparse module.

[sven/scripts.git] / home / portalmonitor.py

diff --git a/home/portalmonitor.py b/home/portalmonitor.py
index 12a088e1e6970c7baaca183993bb02a6af82c609..fdae4ba21e67bd9c84458cf3a570cb7e00877d86 100755 (executable)
--- a/home/portalmonitor.py
+++ b/home/portalmonitor.py
@@ -1,26 +1,26 @@
 #!/usr/bin/env python3
 
+import argparse
 import requests
 import time
 import sys
 import configparser
 from suntime import Sun
-from optparse import OptionParser
 
-parser = OptionParser(usage="usage: %prog [options]")
-parser.add_option("-s",
+parser = argparse.ArgumentParser()
+parser.add_argument("-s",
                   "--status",
                   action="store_true",
                   dest="printStatus",
                   help="Print Status Information",
                   default=False)
-parser.add_option("-f",
+parser.add_argument("-f",
                   "--force",
                   action="store_true",
                   dest="force",
                   help="Force retrieval of Power value",
                   default=False)
-(options, args) = parser.parse_args()
+args = parser.parse_args()
 
 
 def isDaylight(lat, lon, toleranceSeconds):
@@ -46,10 +46,7 @@ def getCurrentPower(userName, password, stationId):
                            'pwd': password
                        },
                        timeout=(10, 30))
-        except requests.exceptions.RequestException as e:
-            raise SystemExit(e)
 
-        try:
             r = s.post(
                 'https://www.envertecportal.com/ApiStations/getStationInfo',
                 data={
@@ -57,10 +54,7 @@ def getCurrentPower(userName, password, stationId):
                 },
                 timeout=(10, 60)).json()
             power = r['Data']['Power']
-        except requests.exceptions.RequestException as e:
-            raise SystemExit(e)
 
-        try:
             r = s.post('https://www.envertecportal.com/apiAccount/Logout',
                        timeout=(10, 30))
         except requests.exceptions.RequestException as e:
@@ -69,20 +63,43 @@ def getCurrentPower(userName, password, stationId):
     return float(power)
 
 
+# use our stateFile to determine if we have a state change
+# used to decide if we print something - thus generate a mail - later on
+def stateCheck(newState, stateFile):
+    try:
+        with open(stateFile, "r") as f:
+            oldState = f.read()
+    except FileNotFoundError:
+        oldState = "NULL"
+
+    if newState == oldState:
+        change = False
+    else:
+        change = True
+        with open(stateFile, "w") as f:
+            f.write(newState)
+
+    return change
+
+
 # read configuration file
 conf = configparser.ConfigParser()
 conf.read('portalmonitor.ini')
 
 # retrieve current power value as reported by envertecportal
 if isDaylight(conf['config'].getfloat('lat'), conf['config'].getfloat('lon'),
-              conf['config'].getint('toleranceSeconds')) or options.force:
+              conf['config'].getint('toleranceSeconds')) or args.force:
     currentPower = getCurrentPower(conf['config']['userName'],
                                    conf['config']['password'],
                                    conf['config']['stationId'])
 
-    if options.printStatus:
-        print('Current Power: ' + str(currentPower))
+    if args.printStatus:
+        print(f"Current Power: {currentPower}")
 
     if currentPower == 0:
-        print('Error: Power dropped to 0 but we should have daylight!')
+        if stateCheck('FAILED', conf['config']['stateFile']):
+            print('Error: Power dropped to 0 but we should have daylight!')
         sys.exit(1)
+    else:
+        if stateCheck('OK', conf['config']['stateFile']):
+            print(f"Resolved: Inverter reports {currentPower}W")