diff --git a/src/utils/sqlite.py b/src/utils/sqlite.py index 7707999..c69bff0 100644 --- a/src/utils/sqlite.py +++ b/src/utils/sqlite.py @@ -1,5 +1,5 @@ from os import mkdir -from sqlite3 import connect +from sqlite3 import Cursor, connect from utils.misc import exist @@ -7,7 +7,7 @@ from utils.misc import exist class Database: """Handle SQLite3 database""" - def __init__(self, path: str, filename: str): + def __init__(self, path: str, filename: str) -> None: if not exist(path): mkdir(path) @@ -17,7 +17,7 @@ class Database: self.fullpath = fullpath - def request(self, request: str, valeurs=None): + def request(self, request: str, valeurs=None) -> tuple[Cursor, int | None]: """Send a request to the database""" connection = connect(self.fullpath) cursor = connection.cursor() @@ -35,7 +35,7 @@ class Database: def format(self, keys, cursor: tuple) -> dict: """Format sqlite request's result as dict""" values = [] - if cursor != None: + if cursor[0] != None: datas = cursor[0].fetchall() for data in datas[0]: values.append(data) @@ -53,7 +53,7 @@ class Database: class FilesDB(Database): """Handle files in sqlite3 database""" - def __init__(self, path: str, filename: str): + def __init__(self, path: str, filename: str) -> None: super().__init__(path, filename) self.table_name = "files" @@ -61,7 +61,7 @@ class FilesDB(Database): f"CREATE TABLE IF NOT EXISTS {self.table_name} \ (filename TEXT, date INTEGER);") - def add_file(self, filename: str, date: int): + def add_file(self, filename: str, date: int) -> None: """Add a file""" self.request( f"INSERT INTO {self.table_name} (filename, date) VALUES (?, ?);",