Add database structure

This commit is contained in:
Simon V. Lejel 2022-03-11 18:09:49 +01:00
parent 28e4ae478e
commit 925164e21d
2 changed files with 35 additions and 0 deletions

17
db/connector.py Normal file
View file

@ -0,0 +1,17 @@
import os
from sqlalchemy import create_engine
user = os.environ['DB_USER']
password = os.environ['DB_PASSWORD']
host = os.environ['DB_HOST']
port = os.environ['DB_PORT']
database = os.environ['DB_DATABASE']
def get_connection():
return create_engine(
url="mysql+pymysql://{0}:{1}@{2}:{3}/{4}".format(
user, password, host, port, database
)
)

18
db/model/image_table.py Normal file
View file

@ -0,0 +1,18 @@
import sqlalchemy as db
from db import connector
engine = connector.create_engine
metadata_obj = db.MetaData()
profile = db.Table(
'images',
metadata_obj,
db.Column('id', db.Integer, primary_key=True),
db.Column('location', db.String),
db.Column('copyright', db.String),
db.Column('date_added', db.TIMESTAMP),
)
metadata_obj.create_all(engine)