Add files via upload

Author: mozancetin

SQLiteImageHandler/SQLiteImageHandler.py

@@ -1,7 +1,7 @@
 import sqlite3
 import os
 
-class SQLiteImageHandler():
+class ImageHandler:
 
     def __init__(self, databasePath : str = "database.db", tableName : str = "images"):
         self.databasePath = databasePath
@@ -80,6 +80,11 @@ def getSaveImage(self, imageName : str = None, savePath : str = "savedImage"):
         if imageName == None:
             raise ValueError("Give me an image boss. (imageName parameter has not declared.)")
 
+        path, extension = os.path.splitext(savePath)
+
+        if extension != "":
+            savePath = path
+
         try:
             self.cursor.execute(f"SELECT * FROM {self.tableName} WHERE Name = ?", (imageName,))
             data = self.cursor.fetchone()
@@ -172,4 +177,4 @@ def updateImage(self, imageName : str = None, newImageBytes : bytes = None, newE
                 self.connection.commit()
             print("Image updated successfully")
         except Exception as error:
-            print(error)
+            print(error)

SQLiteImageHandler/__init__.py

@@ -0,0 +1 @@
+from SQLiteImageHandler.SQLiteImageHandler import ImageHandler