filter on mimetype - image only

Path-Variable · Dec 24, 2023 · 3458f18 · 3458f18
1 parent bd7df56
commit 3458f18
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -1,7 +1,9 @@
 version: '3.0'
 services:
   api:
-    build: .
+    build:
+      context: .
+      dockerfile: dockerfile-amd64
     ports:
       - "5000:5000"
       - "5678:5678"

diff --git a/main.py b/main.py
@@ -20,7 +20,9 @@ def process(self, content):
             return
         if "event" in content and "files" in content["event"]:
             app.logger.info("Received event contains files")
-            # TODO: switch to manageable list of channel ids
+            if "image" not in content["event"]["files"][0]["mimetype"]:
+                app.logger.info("Received file is not an image")
+                return
             if content["event"]["channel"] in self.read_channels:
                 detector.detect(content["event"]["files"][0]["url_private_download"])
             else: