lionel/cryptAES
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

SQL formatting

Browse Source
This commit is contained in:
lionel 2022-08-14 01:30:48 +02:00
parent da91616a23
commit 22f15b91c8
1 changed files with 12 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
backup.py
View File

@ -339,10 +339,8 @@ class DataBase:
params = {'id': id, params = {'id': id,
'name': ', '.join([f"'{file['name']}'" for file in list_file]), 'name': ', '.join([f"'{file['name']}'" for file in list_file]),
'path': ', '.join([f"'{file['path']}'" for file in list_file])} 'path': ', '.join([f"'{file['path']}'" for file in list_file])}
cursor.execute("""SELECT 1 FROM files cursor.execute("""SELECT 1 FROM files WHERE crypt_id=? AND name NOT IN (?) AND path NOT IN (?)""",
WHERE crypt_id={id} (params['id'], params['name'], params['path']))
AND name NOT IN ({name})
AND path NOT IN ({path})""".format(**params))
neighbour = cursor.fetchall() neighbour = cursor.fetchall()
# if they have a neighbour don't overwrite it # if they have a neighbour don't overwrite it
if len(neighbour) > 0: if len(neighbour) > 0:
@ -351,22 +349,21 @@ class DataBase:
else: else:
# if they are different, define the same id for the files of this archive # if they are different, define the same id for the files of this archive
if len(set(crypt_id_list)) > 1: if len(set(crypt_id_list)) > 1:
cursor.execute("""UPDATE files SET crypt_id={id} cursor.execute("""UPDATE files SET crypt_id=? WHERE name IN (?) AND path IN (?)""",
WHERE name IN ({name}) (params['id'], params['name'], params['path']))
AND path IN ({path})""".format(**params))
return id return id
def get_files(self, path): def get_files(self, path):
cursor = self.conn.cursor() cursor = self.conn.cursor()
cursor.execute("""SELECT id, name, path FROM files WHERE path LIKE '{path}%'""" cursor.execute("""SELECT id, name, path FROM files WHERE path LIKE ?""",
.format(path=path)) (path + "%", ))
list_file = cursor.fetchall() list_file = cursor.fetchall()
return list_file return list_file
def delete_file(self, file): def delete_file(self, file):
cursor = self.conn.cursor() cursor = self.conn.cursor()
cursor.execute("""DELETE FROM files WHERE id='{id}' AND name='{name}' AND path='{path}'""" cursor.execute("""DELETE FROM files WHERE id=? AND name=? AND path=?""",
.format(**file)) (file['id'], file['name'], file['path']))
self.conn.commit() self.conn.commit()
def get_orphan_crypt(self): def get_orphan_crypt(self):
@ -402,8 +399,8 @@ class DataBase:
def set_crypt_attr(self, crypt_id, compress_mode=None, sha1sum=None): def set_crypt_attr(self, crypt_id, compress_mode=None, sha1sum=None):
cursor = self.conn.cursor() cursor = self.conn.cursor()
cursor.execute("""UPDATE crypt SET compress_mode='{mode}', sha1sum='{sum}' WHERE id='{id}'""" cursor.execute("""UPDATE crypt SET compress_mode=?, sha1sum=? WHERE id=?""",
.format(id=crypt_id, mode=compress_mode, sum=sha1sum)) (compress_mode, sha1sum, crypt_id))
self.conn.commit() self.conn.commit()
def add(self, list_file): def add(self, list_file):
@ -419,8 +416,8 @@ class DataBase:
(file['size'], file['m_date'], file['c_date'], crypt_id, file['name'], file['path'])) (file['size'], file['m_date'], file['c_date'], crypt_id, file['name'], file['path']))
proceed = True proceed = True
else: else:
cursor.execute("""INSERT INTO files VALUES({id}, '{name}', '{path}', '{size}', '{m_date}', '{c_date}', {crypt_id})""".format( cursor.execute("""INSERT INTO files VALUES(?, ?, ?, ?, ?, ?, ?)""",
id=file_id, crypt_id=crypt_id, **file)) (file_id, file['name'], file['path'], file['size'], file['m_date'], file['c_date'], crypt_id))
file_id += 1 file_id += 1
proceed = True proceed = True
if proceed: if proceed: