You are on page 1of 1

class UserScreen(Screen):

pass

class LeaseApp(MDApp):
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.init_database()

def build(self):
self.theme_cls.primary_palette = "Blue"
self.theme_cls.primary_hue = "500"
self.theme_cls.theme_style = "Light"
return Builder.load_string(KV)

def init_database(self):
conn = sqlite3.connect('users.db')
cursor = conn.cursor()

cursor.execute('''CREATE TABLE IF NOT EXISTS users


(username TEXT PRIMARY KEY NOT NULL, password TEXT
NOT NULL);''')
conn.commit()
conn.close()

def login(self, username, password):


if username == 'admin' and password == 'admin':
self.root.current = 'admin'
elif self.validate_user(username, password):
self.root.get_screen('user').ids.user_label.text = f"Welcome,
{username}!"
self.root.current = 'user'
else:
self.show_alert_dialog("Invalid username or password")

def register(self, username, password):


conn = sqlite3.connect('users.db')
cursor = conn.cursor()

try:
cursor.execute("INSERT INTO users (username, password) VALUES
(?, ?)", (username, password))
conn.commit()
self.show_alert_dialog("Registration successful!")
self.root.current = 'login'
except sqlite3.IntegrityError:
self.show_alert_dialog("Username already exists")

You might also like