hochbeet/app.py

@@ -20,14 +20,12 @@ log_level = logging.INFO
 LOG_FILE = "/var/log/hochbeet.log"
 LOG_FORMAT = "%(asctime)s %(levelname)s %(message)s"
 
-# logging.basicConfig(format=LOG_FORMAT, level=log_level, filename=LOG_FILE)
-logging.basicConfig(format=LOG_FORMAT, level=log_level)
+logging.basicConfig(format=LOG_FORMAT, level=log_level, filename=LOG_FILE)
+#logging.basicConfig(format=LOG_FORMAT, level=log_level)
 log = logging.getLogger('hochbeet')
 
 
 class GreenControl(Thread):
-    """Green Control"""
-
     def __init__(self):
         super(GreenControl, self).__init__()
         self.__config_file = os.path.join(os.path.expanduser('~'), ".config/hochbeet/config.json")
@@ -43,11 +41,14 @@ class GreenControl(Thread):
         except FileNotFoundError:
             # create default config
             os.makedirs(os.path.dirname(self.__config_file), exist_ok=True)
-            pwd = os.getcwd()
-            shutil.copyfile("config/config.json", self.__config_file)
+            shutil.copyfile("hochbeet/config/config.json", self.__config_file)
             with open(self.__config_file, "r") as handle:
                 self.__config = json.load(handle)
 
+
+    def reload_config(self):
+        self.__trigger_read_config = True
+
     def run(self):
         while self.__run_condition:
             if self.__trigger_read_config:
@@ -59,24 +60,20 @@ class GreenControl(Thread):
             GPIO.setwarnings(False)
             GPIO.setmode(GPIO.BCM)
             GPIO.setup(self.__pin, GPIO.OUT)
+
             now = datetime.datetime.now()
 
+            # Check if auto-water is on
             if self.__config['water'][0]['autostate']:
-                idx = 0
+                index = 0
                 if int(now.hour) >= 12:
-                    idx = 1
-                on_time_pattern = self.__config['water'][0]['times'][idx]['on_time']
+                    index = 1
+                on_time_pattern = self.__config['water'][0]['times'][index]['on_time']
                 on_time_pattern = on_time_pattern.split(':')
-                on_time = now.replace(hour=int(on_time_pattern[0]),
-                                      minute=int(on_time_pattern[1]),
-                                      second=0,
-                                      microsecond=0)
-                off_time_pattern = self.__config['water'][0]['times'][idx]['off_time']
+                on_time = now.replace(hour=int(on_time_pattern[0]), minute=int(on_time_pattern[1]), second=0, microsecond=0)
+                off_time_pattern = self.__config['water'][0]['times'][index]['off_time']
                 off_time_pattern = off_time_pattern.split(':')
-                off_time = now.replace(hour=int(off_time_pattern[0]),
-                                       minute=int(off_time_pattern[1]),
-                                       second=0,
-                                       microsecond=0)
+                off_time = now.replace(hour=int(off_time_pattern[0]), minute=int(off_time_pattern[1]), second=0, microsecond=0)
 
                 if now > on_time and now <= off_time and not self.__water_state:
                     GPIO.output(self.__pin, 1)
@@ -88,60 +85,61 @@ class GreenControl(Thread):
                     log.info("Switch water off by time")
             sleep(1)
 
+
     def state(self):
-        """Return water state"""
         return self.__water_state
 
+
+    def set_state(self, state: bool):
+        if state:
+	    GPIO.output(self.__pin, 1)
+            self.__water_state = True
+        else:
+	    GPIO.output(self.__pin, 0)
+            self.__water_state = False
+
+
     def get_auto_state(self):
-        """Return auto state"""
         state = False
         if self.__config['water'][0]['autostate']:
             state = self.__config['water'][0]['autostate']
         return state
 
+
     def set_auto_state(self, state):
-        """Set the auto state"""
         self.__config['water'][0]['autostate'] = state
         with open(self.__config_file, "w") as handle:
             json.dump(self.__config, handle)
-            self.__trigger_read_config = True
+        self.reload_config()
+
 
     def get_times(self):
-        """Return the times for auto state"""
         times = None
         if self.__config['water'][0]['times']:
             times = self.__config['water'][0]['times']
         return times
 
+
     def set_times(self, times):
-        """Set the times for auto state"""
         self.__config['water'][0]['times'] = times
         with open(self.__config_file, "w") as handle:
             json.dump(self.__config, handle)
-            self.__trigger_read_config = True
-
-    def switch_water(self, state):
-        """Switch the water on/off"""
-        if state:
-            GPIO.output(self.__pin, 1)
-        else:
-            GPIO.output(self.__pin, 0)
-
+        self.reload_config()
 
 
 green_ctrl = GreenControl()
 green_ctrl.start()
+
 app = Flask(__name__)
 
 
 @app.route('/')
 def index():
-    """Handle index.html"""
     return render_template('index.html')
 
+
 @app.route('/sample', methods=['GET'])
 def get_sample():
-    """Handle GET request for /sample"""
     global green_ctrl
 
     sample = {}
@@ -149,32 +147,32 @@ def get_sample():
     sample["state"] = green_ctrl.state()
     return make_response(jsonify(sample), 200)
 
+
 @app.route('/sample', methods=['PATCH'])
 def patch_sample():
-    """Handle PATCH request for sample"""
     global green_ctrl
     record = json.loads(request.data)
     if "waterstate" in record:
         if record["waterstate"]:
+            green_ctrl.set_state(True)
             log.info("Switch water on by button: %s", datetime.datetime.now())
-            green_ctrl.switch_water(True)
         else:
+            green_ctrl.set_state(False)
             log.info("Switch water off by button: %s", datetime.datetime.now())
-            green_ctrl.switch_water(False)
     return make_response("", 204)
 
+
 @app.route('/config', methods=['GET'])
 def get_config():
-    """Handle GER request for /config"""
     global green_ctrl
     config = {}
     config["autostate"] = green_ctrl.get_auto_state()
     config["times"] = green_ctrl.get_times()
     return make_response(jsonify(config), 200)
 
+
 @app.route('/config', methods=['PATCH'])
 def patch_config():
-    """Handle PATCH request for sample"""
     global green_ctrl
     record = json.loads(request.data)
     if "id" in record and record['id'] == '1':
@@ -187,5 +185,6 @@ def patch_config():
     config["times"] = green_ctrl.get_times()
     return make_response(jsonify(config), 200)
 
+
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=8000)

setup.py

@@ -6,11 +6,12 @@ from setuptools import setup
 from setuptools.command.install import install
 
 NAME = 'Hochbeet'
-VERSION = '1'
+VERSION = '2'
 AUTHOR = 'Thomas Klaehn'
 EMAIL = 'tkl@blackfinn.de'
 PACKAGES = ['hochbeet']
-REQUIRES = ['Flask', 'gunicorn', 'RPi.GPIO']
+REQUIRES = ['RPi.GPIO']
+
 CONFIG_FILE = 'config.json'
 PACKAGE_DATA = {
     'hochbeet': [
@@ -22,8 +23,8 @@ PACKAGE_DATA = {
 }
 
 SERVICEDIR = "/lib/systemd/system"
-START_SCRIPT = 'hochbeet.service'
-DAEMON_START_SCRIPT = os.path.join(SERVICEDIR, START_SCRIPT)
+DAEMON_START_SCRIPT = os.path.join(SERVICEDIR, 'hochbeet.service')
+
 LOGFILE = "/var/log/hochbeet.log"
 
 
@@ -31,7 +32,7 @@ class Install(install):
     def run(self):
         install.run(self)
         os.makedirs(SERVICEDIR, exist_ok=True)
-        shutil.copyfile(START_SCRIPT, DAEMON_START_SCRIPT)
+        shutil.copyfile('hochbeet.service', os.path.join(SERVICEDIR, DAEMON_START_SCRIPT))
         os.chmod(DAEMON_START_SCRIPT, stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IROTH)
 
         try: