implemented encyption

This commit is contained in:
2022-10-09 15:14:14 +02:00
parent 274549f04b
commit 048eea106e
4 changed files with 56 additions and 51 deletions

View File

@@ -1,5 +1,6 @@
import sqlite3
from datetime import datetime
from services.encryption import Encryption
class User:
def __init__(self, connection : sqlite3.Connection, id = None, username = None, firstname = None, lastname = None, address = None, zipcode = None, city_id = None, email = None, phone = None, password = None, role = None):
@@ -46,7 +47,7 @@ class User:
cur.execute("""
INSERT INTO users
(id, username, firstname, lastname, address, zipcode, city_id, email, phone, password, role, created) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
""", (self.id, self.username, self.firstname, self.lastname, self.address, self.zipcode, self.city_id, self.email, self.phone, self.password, self.role, current_date))
""", (self.id, Encryption.encrypt(self.username), Encryption.encrypt(self.firstname), Encryption.encrypt(self.lastname), Encryption.encrypt(self.address), Encryption.encrypt(self.zipcode), self.city_id, Encryption.encrypt(self.email), Encryption.encrypt(self.phone), Encryption.encrypt(self.password), Encryption.encrypt(self.role), current_date))
self.connection.commit()
cur.close()
@@ -67,7 +68,7 @@ class User:
password = ?,
role = ?
WHERE id = ?
""", (self.username, self.firstname, self.lastname, self.address, self.zipcode, self.city_id, self.email, self.phone, self.password, self.role, self.id))
""", (Encryption.encrypt(self.username), Encryption.encrypt(self.firstname), Encryption.encrypt(self.lastname), Encryption.encrypt(self.address), Encryption.encrypt(self.zipcode), self.city_id, Encryption.encrypt(self.email), Encryption.encrypt(self.phone), self.password, self.role, self.id))
self.connection.commit()
cur.close()
@@ -82,15 +83,15 @@ class User:
def _set_row_values(self, row):
self.id = row[0]
self.username = row[1]
self.firstname = row[2]
self.lastname = row[3]
self.address = row[4]
self.zipcode = row[5]
self.username = Encryption.decrypt(row[1])
self.firstname = Encryption.decrypt(row[2])
self.lastname = Encryption.decrypt(row[3])
self.address = Encryption.decrypt(row[4])
self.zipcode = Encryption.decrypt(row[5])
self.city_id = row[6]
self.email = row[7]
self.phone = row[8]
self.password = row[9]
self.role = row[10]
self.email = Encryption.decrypt(row[7])
self.phone = Encryption.decrypt(row[8])
self.password = Encryption.decrypt(row[9])
self.role = Encryption.decrypt(row[10])
self.created = row[11]
return self