Add basic sqlalchemy model.
Signed-off-by: Jan Losinski <losinski@wh2.tu-dresden.de>
This commit is contained in:
parent
9404d574dc
commit
a202328fba
|
|
@ -0,0 +1,46 @@
|
|||
from sqlalchemy import create_engine
|
||||
from sqlalchemy import Integer, String, DateTime, Column, ForeignKey, Enum
|
||||
from sqlalchemy.orm import scoped_session, sessionmaker, relationship
|
||||
from sqlalchemy.ext.declarative import declarative_base
|
||||
|
||||
Base = declarative_base()
|
||||
|
||||
|
||||
class User(Base):
|
||||
__tablename__ = "user"
|
||||
|
||||
name = Column(String, primary_key=True)
|
||||
token = Column(String())
|
||||
client_id = Column(String())
|
||||
client_secret = Column(String())
|
||||
auth_token = Column(String)
|
||||
refresh_token = Column(String)
|
||||
token_valid = Column(DateTime)
|
||||
last_check = Column(DateTime)
|
||||
email = Column(String)
|
||||
kindle_mail = Column(String)
|
||||
|
||||
jobs = relationship('Job', backref='user')
|
||||
|
||||
|
||||
class Job(Base):
|
||||
__tablename__ = "job"
|
||||
|
||||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||||
article = Column(Integer)
|
||||
title = Column(String)
|
||||
user_name = Column(Integer, ForeignKey("user.name"))
|
||||
format = Column(Enum('pdf', 'mobi'))
|
||||
|
||||
|
||||
def make_session(uri):
|
||||
Session = sessionmaker(autocommit=False,
|
||||
autoflush=False,
|
||||
bind=create_engine(uri))
|
||||
return scoped_session(Session)
|
||||
|
||||
|
||||
def re_create_db(uri):
|
||||
engine = create_engine(uri)
|
||||
Base.metadata.drop_all(engine)
|
||||
Base.metadata.create_all(engine)
|
||||
Loading…
Reference in New Issue