82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
from .database import DatabaseController
|
|
from models.users.user import User
|
|
from models.users.customer import Customer
|
|
from models.users.seller import Seller
|
|
from models.users.admin import Admin
|
|
|
|
|
|
class UserController(DatabaseController):
|
|
FIELDS = ['id', 'username', 'password', 'firstName',
|
|
'lastName', 'email', 'phone', 'role']
|
|
TYPE = User
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def convert_type(self, user: User | None
|
|
) -> User | Customer | Seller | Admin | None:
|
|
""" Function to convert a given user to their correct subtype """
|
|
if user is None:
|
|
return None
|
|
|
|
# Set the object to the correct type
|
|
type = Customer
|
|
if user.role == "Seller":
|
|
type = Seller
|
|
if user.role == "Admin":
|
|
type = Admin
|
|
|
|
obj = self.new_instance(type, user.__dict__)
|
|
return obj
|
|
|
|
def create(self, user: User):
|
|
params = [
|
|
user.username,
|
|
user.password,
|
|
user.firstName,
|
|
user.lastName,
|
|
user.email,
|
|
user.phone,
|
|
user.role
|
|
]
|
|
query = """
|
|
INSERT INTO Users
|
|
(username, password, first_name, last_name, email, phone, role)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
"""
|
|
|
|
self.do(query, params)
|
|
|
|
def read(self, username: str) -> User | None:
|
|
params = [
|
|
username
|
|
]
|
|
query = "SELECT * FROM Users WHERE Username = ?"
|
|
|
|
return self.convert_type(self.get_one(query, params))
|
|
|
|
def read_id(self, id: int) -> User | None:
|
|
params = [
|
|
id
|
|
]
|
|
query = "SELECT * FROM Users WHERE id = ?"
|
|
|
|
return self.convert_type(self.get_one(query, params))
|
|
|
|
def read_all(self) -> list[User] | None:
|
|
params = []
|
|
query = """ SELECT * FROM Users """
|
|
|
|
return self.get_many(query, params)
|
|
|
|
def update(self):
|
|
print("Doing work")
|
|
|
|
def delete(self, id: int):
|
|
params = [
|
|
id
|
|
]
|
|
query = """ DELETE FROM Users WHERE id = ? """
|
|
|
|
return self.do(query, params)
|