Use class instance for downloading

backend/StreamServerApp/database_utils.py

@@ -31,6 +31,7 @@
 logger = logging.getLogger("root")
 
 
+
 def delete_DB_Infos():
     """ delete all videos, movies and series in the db
     """

backend/StreamServerApp/media_management/cover_downloader.py

@@ -8,10 +8,10 @@
 class cover_downloader:
 
     def __init__(self):
-
+        self.auth_key = ""
         env_auth_key = os.getenv('TMBD_KEY')
-        if not auth_key:
-            logger("cover downloader: no auth key")
+        if not env_auth_key:
+            logger.error("cover downloader: no auth key")
             return
         self.auth_key = env_auth_key
 
@@ -20,7 +20,7 @@ def __init__(self):
 
         response = requests.get(configuration_url)
         if response.status_code != 200:
-            logger("cover downloader: Failed to get configuration")
+            logger.error("cover downloader: Failed to get configuration")
             return
         value = response.json()
 
@@ -39,21 +39,21 @@ def download_cover(self, name, outputfile, is_tv_show=False):
                     name.replace(" ", "+"), self.auth_key)
             response = requests.get(api_url)
             if response.status_code != 200:
-                logger("cover downloader: Failed to search movie")
+                logger.error("cover downloader: Failed to search movie")
                 return -1
             value = response.json()
 
             poster_url = "{}/{}{}".format(self.base_url, self.poster_size,
                                         value["results"][0]["poster_path"])
             response = requests.get(poster_url)
             if response.status_code != 200:
-                logger("cover downloader: Failed to download cover")
+                logger.error("cover downloader: Failed to download cover")
                 return -1
 
             with open(outputfile, mode="wb") as file:
                 file.write(response.content)
 
             return 1
         else:
-            logger("cover downloader: No properly initialized")
+            logger.error("cover downloader: No properly initialized")
             return -1

backend/StreamServerApp/tasks.py

@@ -6,6 +6,7 @@
 import logging
 
 logger = logging.getLogger("root")
+cvdwnld = cover_downloader()
 
 
 @shared_task
@@ -27,16 +28,16 @@ def get_subtitles_async(video_id, video_path, remote_url):
 
 @shared_task
 def download_cover_async(id, name, is_tv_show=False):
-    output_file = "/usr/src/static/{}.jpeg".format(name)
-    ret = cover_downloader.download_cover(name, output_file, is_tv_show)
+    output_file = "/usr/static/{}.jpeg".format(name)
+    ret = cvdwnld.download_cover(name, output_file, is_tv_show)
     if ret:
         video = Video.objects.get(id=id)
         if is_tv_show:
             serie = Series.objects.get(id=video.series_id)
-            serie.thumbnail = output_file
+            serie.thumbnail = "/static/{}.jpeg".format(name)
             serie.save()
         else:
-            video.thumbnail = output_file
+            video.thumbnail = "/static/{}.jpeg".format(name)
             video.save()
 
     return 0

backend/StreamServerApp/tests/tests_cover_download.py

@@ -0,0 +1,17 @@
+from django.test import TestCase
+import os
+from unittest import TestCase, mock
+from StreamServerApp.media_management.cover_downloader import cover_downloader
+import optparse
+
+class CoverDownloaderTest(TestCase):
+
+    def setUp(self):
+        pass
+
+    @mock.patch.dict(os.environ, {"TMBD_KEY": ""})
+    def test_ensureErrorWithInvalidToken(self):
+
+        cvdwnld = cover_downloader()
+        self.assertEqual(cvdwnld.download_cover("test", "/test/test.jpeg", True), -1)
+

docker-compose-debug.yml

@@ -79,6 +79,7 @@ services:
     volumes:
         - ./backend/:/usr/src/app/
         - ./Videos/:/usr/src/app/Videos
+        - static_volume:/usr/static/
         - progress_volume:/usr/progress/
         - torrent:/usr/torrent/:rw
         - ./log/:/debug/

docker-compose-prod.yml

@@ -63,6 +63,7 @@ services:
         - ./Videos/:/usr/src/app/Videos
         - torrent:/usr/torrent/:rw
         - progress_volume:/usr/progress/
+        - static_volume:/usr/static/
         - ./log/:/debug/
     env_file:
         - .env