47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
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)
|