57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
from database import postgresql_db_handler
|
|
|
|
|
|
def get_user_by_id(user_id):
|
|
query = """
|
|
SELECT * FROM users WHERE id = %s;
|
|
"""
|
|
params = (user_id,)
|
|
result = postgresql_db_handler.execute_query(query, params, fetchall=True)
|
|
if result:
|
|
user_data = result[0]
|
|
return user_data
|
|
else:
|
|
return None
|
|
|
|
|
|
def get_user_by_email(email):
|
|
query = """
|
|
SELECT * FROM users WHERE email = %s;
|
|
"""
|
|
params = (email,)
|
|
result = postgresql_db_handler.execute_query(query, params, fetchall=True)
|
|
if result:
|
|
user_data = result[0]
|
|
return user_data
|
|
else:
|
|
return None
|
|
|
|
|
|
def create_user_db(username, password, email):
|
|
query = """
|
|
INSERT INTO users (username, password, email)
|
|
VALUES (%s, %s, %s);
|
|
"""
|
|
params = (username, password, email)
|
|
postgresql_db_handler.execute_query(query, params)
|
|
|
|
|
|
def update_user_db(user_id, new_username, new_password, new_email):
|
|
query = """
|
|
UPDATE users
|
|
SET username = %s, password = %s, email = %s
|
|
WHERE id = %s;
|
|
"""
|
|
params = (new_username, new_password, new_email, user_id)
|
|
postgresql_db_handler.execute_query(query, params)
|
|
print(f"User with ID {user_id} updated successfully.")
|
|
|
|
|
|
def delete_user_db(user_id):
|
|
query = """
|
|
DELETE FROM users WHERE id = %s;
|
|
"""
|
|
params = (user_id,)
|
|
postgresql_db_handler.execute_query(query, params)
|
|
print(f"User with ID {user_id} deleted successfully.")
|