db_model

db_model.py

from sqlalchemy import Column, Integer, String, DateTime
from sqlalchemy.dialects.sqlite import DATETIME

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

Base = declarative_base()


class TestTable(Base):
    __tablename__ = 'test_table'

    id = Column(Integer,
                Sequence('id', start=1, increment=1),
                primary_key=True)
    col1 = Column(Integer)
    col2 = Column(String(200))
    created_at = Column(DATETIME)
    updated_at = Column(DATETIME)

    def __init__(self, GroupId, AccountId, createdAt, updatedAt):
        self.GroupId = GroupId
        self.AccountId = AccountId
        self.createdAt = createdAt
        self.updatedAt = updatedAt
        
        
def db_init(db_engine):
    db_session = sessionmaker(bind=db_engine)
    session = db_session()
    return session


def db_drop(db_engine):
    Base.metadata.drop_all(db_engine)
    return None