implement user and login
This commit is contained in:
87
obj/user.py
Normal file
87
obj/user.py
Normal file
@@ -0,0 +1,87 @@
|
||||
import sqlite3
|
||||
import os
|
||||
|
||||
class User:
|
||||
def __init__(self, user_id, name, email, password):
|
||||
self._user_id = user_id
|
||||
self._name = name
|
||||
self._email = email
|
||||
self._password = password
|
||||
self._db_path = os.path.join("db", "user.db")
|
||||
|
||||
# Getter
|
||||
def get_id(self):
|
||||
return self._user_id
|
||||
|
||||
def get_name(self):
|
||||
return self._name
|
||||
|
||||
def get_email(self):
|
||||
return self._email
|
||||
|
||||
def get_password(self):
|
||||
return self._password
|
||||
|
||||
# Setter
|
||||
def set_name(self, name):
|
||||
self._name = name
|
||||
self._update_db()
|
||||
|
||||
def set_email(self, email):
|
||||
self._email = email
|
||||
self._update_db()
|
||||
|
||||
def set_password(self, password):
|
||||
self._password = password
|
||||
self._update_db()
|
||||
|
||||
def _update_db(self):
|
||||
"""Update the user's data in the DB"""
|
||||
with sqlite3.connect(self._db_path) as conn:
|
||||
c = conn.cursor()
|
||||
c.execute('UPDATE users SET name = ?, email = ?, password = ? WHERE id = ?',
|
||||
(self._name, self._email, self._password, self._user_id))
|
||||
conn.commit()
|
||||
|
||||
def save_to_db(self):
|
||||
"""Speichert den aktuellen Benutzer in die Datenbank (Insert oder Update)"""
|
||||
with sqlite3.connect(self._db_path) as conn:
|
||||
c = conn.cursor()
|
||||
c.execute("SELECT id FROM users WHERE id = ?", (self._user_id,))
|
||||
if c.fetchone():
|
||||
# Benutzer existiert bereits → aktualisieren
|
||||
c.execute('UPDATE users SET name = ?, email = ?, password = ? WHERE id = ?',
|
||||
(self._name, self._email, self._password, self._user_id))
|
||||
else:
|
||||
# Neuer Benutzer → einfügen
|
||||
c.execute('INSERT INTO users (id, name, email, password) VALUES (?, ?, ?, ?)',
|
||||
(self._user_id, self._name, self._email, self._password))
|
||||
conn.commit()
|
||||
|
||||
|
||||
@staticmethod
|
||||
def authenticate_user(username, password):
|
||||
"""Prüft, ob ein Benutzer mit E-Mail + Passwort existiert"""
|
||||
db_path = os.path.join("db", "user.db")
|
||||
with sqlite3.connect(db_path) as conn:
|
||||
c = conn.cursor()
|
||||
c.execute("SELECT id, name, email FROM users WHERE name = ? AND password = ?", (username, password))
|
||||
result = c.fetchone()
|
||||
if result:
|
||||
user_id, name, email = result
|
||||
return User(user_id, name, email, password)
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def validate_user(username, userid):
|
||||
"""Prüft, ob ein Benutzer mit E-Mail + Passwort existiert"""
|
||||
db_path = os.path.join("db", "user.db")
|
||||
with sqlite3.connect(db_path) as conn:
|
||||
c = conn.cursor()
|
||||
c.execute("SELECT id, name, email FROM users WHERE name = ? AND id = ?", (username, userid))
|
||||
result = c.fetchone()
|
||||
if result:
|
||||
user_id, name, email= result
|
||||
return True
|
||||
return None
|
||||
|
||||
Reference in New Issue
Block a user