From 7775248c03a8399892bfeb49f686c3b9d6924b428e4084c3dd3c4a7c06ea754d Mon Sep 17 00:00:00 2001 From: n Date: Sat, 15 Feb 2025 19:17:03 -0600 Subject: [PATCH] make function for making accounts probably worst code yet Signed-off-by: n --- src/toastbin/database.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/src/toastbin/database.py b/src/toastbin/database.py index 06c3d36..ec6a1c8 100644 --- a/src/toastbin/database.py +++ b/src/toastbin/database.py @@ -2,7 +2,9 @@ import sqlalchemy import sqlalchemy.orm from sqlalchemy import Table, Column, Integer, String, MetaData from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.sql import text import json +import re with open('config.json', 'r') as file: config = json.load(file) @@ -25,6 +27,31 @@ users = Table( Column('id', Integer, primary_key = True), Column('name', String), Column('password', String), + Column('email', String) ) +class User(Base): + __tablename__ = 'users' + id = Column(Integer, primary_key=True) + name = Column(String) + password = Column(String) + email = Column(String) + metadata.create_all(engine) + +def next_id(): + id = session.execute(text('SELECT MAX(id) FROM users')) + ok = str(id.fetchall()) + id_parsed = ''.join(char for char in ok if char.isdigit()) + id_parsed = int(id_parsed) + id_parsed = id_parsed + 1 + return str(id_parsed) + +print(next_id()) + +def create_user(name, password): + new_user = User(id=next_id(), name=name, password=password) + session.add(new_user) + session.commit() + +# create_user("test3", "test3")