import.py

@@ -1,7 +1,6 @@
 import os
 import re
 import sys
-import fnmatch
 from datetime import datetime
 
 sys.path.append('./')
@@ -11,7 +10,7 @@ django.setup()
 from django.contrib.auth.models import User
 from core.models import Video
 
-folder_path = './channel_archiver/UnusualVideos (@UnusualVideos)/'
+folder_path = 'channel_archiver/UnusualVideos (@UnusualVideos)/'
 
 def extract_version(filename):
     # Define a regular expression pattern to match the version number
@@ -27,14 +26,6 @@ def extract_version(filename):
         # If no match is found, return None
         return None
 
-def find_image_file(name_pattern, folder):
-    pattern = re.compile('.*{}.*\.(jpg|png|webp)'.format(name_pattern), re.IGNORECASE)
-    for root, dirs, files in os.walk(folder):
-        for filename in files:
-            if pattern.match(filename):
-                return os.path.join(root, filename)
-    return None
-
 
 for filename in os.listdir(folder_path):
     if filename.endswith('.mp4'):
@@ -54,8 +45,9 @@ for filename in os.listdir(folder_path):
         with open(os.path.join(folder_path, filename), 'rb') as f:
             video.file.save(filename, f)
 
-        filepath = find_image_file("V"+str(id), folder_path)
-        if filepath and os.path.exists(filepath):
+        filename = filename[:-3] + "webp"
+        filepath = os.path.join(folder_path, filename)
+        if os.path.exists(filepath):
             with open(filepath, 'rb') as f:
                 video.thumbnail.save(filename, f)