28 lines
632 B
Python
28 lines
632 B
Python
|
from peewee import Model,CharField
|
||
|
|
from playhouse.db_url import connect
|
||
|
|
|
||
|
|
|
||
|
|
# TODO: create connection to bot database
|
||
|
|
db = connect("sqlite:///db.db")
|
||
|
|
|
||
|
|
|
||
|
|
class WebUser(Model):
|
||
|
|
username = CharField()
|
||
|
|
password_hash = CharField()
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
db_table = "webusers"
|
||
|
|
database = db
|
||
|
|
|
||
|
|
def userExists(self,username) -> bool:
|
||
|
|
"""Check if the username exists in a database."""
|
||
|
|
query = WebUser.select().where(WebUser.username == username)
|
||
|
|
|
||
|
|
if (not query) or (not query.exists()):
|
||
|
|
return False
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
def build_models():
|
||
|
|
db.create_tables([WebUser])
|