Merge pull request #3 from netz39/pictures

Add picture delivery for state.png

Dockerfile

@@ -16,6 +16,7 @@ COPY requirements.txt /
 RUN pip install -r requirements.txt
 
 COPY src/*.py /
+COPY assets/ /assets/
 
 COPY --from=install /git-version.txt /
 

src/PictureManager.py

@@ -0,0 +1,13 @@
+class PictureManager:
+    def __init__(self, is_open_func, open_image_path, closed_image_path):
+        self.is_open_func = is_open_func
+        self.open_image_path = open_image_path
+        self.closed_image_path = closed_image_path
+
+    def get_image(self):
+        if self.is_open_func():
+            with open(self.open_image_path, "rb") as image_file:
+                return image_file.read()
+        else:
+            with open(self.closed_image_path, "rb") as image_file:
+                return image_file.read()

src/SpaceApiEntry.py

@@ -69,3 +69,6 @@ def set_is_open(self, is_open):
 
     def set_lastchange(self, lastchange):
         self.data["state"]["lastchange"] = lastchange
+
+    def is_open(self):
+        return self.data["state"]["open"]

src/app.py

@@ -15,8 +15,9 @@
 
 import json
 
-from src.SpaceApiEntry import SpaceApiEntry
-from src.SpaceStatusObserver import SpaceStatusObserver
+from PictureManager import PictureManager
+from SpaceApiEntry import SpaceApiEntry
+from SpaceStatusObserver import SpaceStatusObserver
 
 startup_timestamp = datetime.now()
 
@@ -84,12 +85,24 @@ def get(self):
         self.finish()
 
 
-def make_app(observer):
+class PictureHandler(tornado.web.RequestHandler, ABC):
+    # noinspection PyAttributeOutsideInit
+    def initialize(self, picture_manager):
+        self.picture_manager = picture_manager
+
+    def get(self):
+        self.set_header("Content-Type", "image/png")
+        self.write(self.picture_manager.get_image())
+        self.finish()
+
+
+def make_app(observer, picture_manager):
     version_path = r"/v[0-9]"
     return tornado.web.Application([
         (version_path + r"/health", HealthHandler),
         (version_path + r"/oas3", Oas3Handler),
         (r"/", SpaceAPIHandler, dict(observer=observer)),
+        (r"/state.png", PictureHandler, dict(picture_manager=picture_manager)),
     ])
 
 
@@ -121,7 +134,13 @@ def main():
     )
     observer.start()
 
-    app = make_app(observer)
+    picture_manager = PictureManager(
+        is_open_func=observer.space_api_entry.is_open,
+        open_image_path="../assets/open.png",
+        closed_image_path="../assets/closed.png"
+    )
+
+    app = make_app(observer, picture_manager)
     sockets = tornado.netutil.bind_sockets(arg_port, '')
     server = tornado.httpserver.HTTPServer(app)
     server.add_sockets(sockets)