__init__.py 1.51 KB
Newer Older
wanli's avatar
wanli committed
1
# -*- coding: utf-8 -*-
wanli's avatar
wanli committed
2
from sqlalchemy import create_engine
wanli's avatar
wanli committed
3
from sqlalchemy import Column, Integer, String
wanli's avatar
wanli committed
4
from sqlalchemy.orm import sessionmaker
wanli's avatar
wanli committed
5
from sqlalchemy.ext.declarative import declared_attr, declarative_base
wanli's avatar
wanli committed
6

wanli's avatar
wanli committed
7
class BaseModelMixin(object):
wanli's avatar
wanli committed
8
    @declared_attr
wanli's avatar
wanli committed
9 10
    def __tablename__(self):
        return self.__name__.lower()
wanli's avatar
wanli committed
11

wanli's avatar
wanli committed
12
    # __table_args__ = {'mysql_engine': 'InnoDB'}
wanli's avatar
wanli committed
13 14 15

    id = Column(Integer, primary_key=True)

wanli's avatar
wanli committed
16
BaseModel = declarative_base(cls=BaseModelMixin)
wanli's avatar
wanli committed
17 18 19

class MyMixin(object):
    @declared_attr
wanli's avatar
wanli committed
20 21
    def __tablename__(self):
        return self.__name__.lower()
wanli's avatar
wanli committed
22

wanli's avatar
wanli committed
23
    # __table_args__ = {'mysql_engine': 'InnoDB'}
wanli's avatar
wanli committed
24 25 26 27
    __mapper_args__= {'always_refresh': True}

    id = Column(Integer, primary_key=True)

wanli's avatar
wanli committed
28 29
# example:
class MyModel(MyMixin, BaseModel):
wanli's avatar
wanli committed
30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
    __tablename__ = 'mymodel'
    name = Column(String(20))
    fullname = Column(String(32))
    password = Column(String(32))


# if __name__ == "__main__":
engine = create_engine('sqlite:///test.db')

BaseModel.metadata.create_all(engine, checkfirst=True)

Session = sessionmaker(bind=engine)
session = Session()

ed_user = MyModel(name='ed', fullname='Ed Jones', password='edspassword')
session.add(ed_user)


session.add_all(
    [
        MyModel(name='wendy', fullname='Wendy Williams', password='foobar'),
        MyModel(name='mary', fullname='Mary Contrary', password='xxg527'),
        MyModel(name='fred', fullname='Fred Flinstone', password='blah')
    ]
)
session.commit()

57
# our_user = session.query(MyModel).filter_by(name='ed').first()
wanliofficial's avatar
wanliofficial committed
58
# print(our_user)