__init__.py 1.68 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
wanliofficial's avatar
wanliofficial committed
4 5
<<<<<<< HEAD
=======
wanli's avatar
wanli committed
6
from sqlalchemy.orm import sessionmaker
wanliofficial's avatar
wanliofficial committed
7
>>>>>>> 735d39eb4d0c3134b62bf4fe1b7a2ca0ea8da1ca
wanli's avatar
wanli committed
8
from sqlalchemy.ext.declarative import declared_attr, declarative_base
wanli's avatar
wanli committed
9

wanli's avatar
wanli committed
10
class BaseModelMixin(object):
wanli's avatar
wanli committed
11
    @declared_attr
wanli's avatar
wanli committed
12 13
    def __tablename__(self):
        return self.__name__.lower()
wanli's avatar
wanli committed
14

wanli's avatar
wanli committed
15
    # __table_args__ = {'mysql_engine': 'InnoDB'}
wanli's avatar
wanli committed
16 17 18

    id = Column(Integer, primary_key=True)

wanli's avatar
wanli committed
19
BaseModel = declarative_base(cls=BaseModelMixin)
wanli's avatar
wanli committed
20 21 22

class MyMixin(object):
    @declared_attr
wanli's avatar
wanli committed
23 24
    def __tablename__(self):
        return self.__name__.lower()
wanli's avatar
wanli committed
25

wanli's avatar
wanli committed
26
    # __table_args__ = {'mysql_engine': 'InnoDB'}
wanli's avatar
wanli committed
27 28 29 30
    __mapper_args__= {'always_refresh': True}

    id = Column(Integer, primary_key=True)

wanli's avatar
wanli committed
31 32
# example:
class MyModel(MyMixin, BaseModel):
wanliofficial's avatar
wanliofficial committed
33
<<<<<<< HEAD
wanli's avatar
wanli committed
34
    name = Column(String(1000))
wanliofficial's avatar
wanliofficial committed
35
=======
wanli's avatar
wanli committed
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
    __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()

63
# our_user = session.query(MyModel).filter_by(name='ed').first()
wanliofficial's avatar
wanliofficial committed
64 65
# print(our_user)
>>>>>>> 735d39eb4d0c3134b62bf4fe1b7a2ca0ea8da1ca