diff --git a/src/toastbin/database.py b/src/toastbin/database.py index b9a4499..06c3d36 100644 --- a/src/toastbin/database.py +++ b/src/toastbin/database.py @@ -1,14 +1,30 @@ import sqlalchemy +import sqlalchemy.orm +from sqlalchemy import Table, Column, Integer, String, MetaData +from sqlalchemy.ext.declarative import declarative_base import json with open('config.json', 'r') as file: config = json.load(file) + username = config['database']['username'] password = config['database']['password'] port = config['database']['port'] location = config['database']['location'] database = config['database']['database'] +metadata = sqlalchemy.MetaData() + engine = sqlalchemy.create_engine('postgresql://' + username + ':' + password + '@' + location +':' + port + '/' + database) # todo: finish this, give users options maybe Session = sqlalchemy.orm.sessionmaker(bind=engine) session = Session() +Base = declarative_base() + +users = Table( + 'users', metadata, + Column('id', Integer, primary_key = True), + Column('name', String), + Column('password', String), +) + +metadata.create_all(engine)