LukeParke/connexion
1
0
Fork 0
mirror of https://github.com/LukeHagar/connexion.git synced 2025-12-10 04:19:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix flake8 formatting of SQLAlchemy example

Browse Source
This commit is contained in:
Henning Jacobs
2016-06-03 23:52:45 +02:00
parent 34430a4660
commit a30681992f
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/sqlalchemy/app.py
View File

@@ -9,12 +9,14 @@ import orm
db_session = None db_session = None
def get_pets(limit, animal_type=None): def get_pets(limit, animal_type=None):
q = db_session.query(orm.Pet) q = db_session.query(orm.Pet)
if animal_type: if animal_type:
q = q.filter(orm.Pet.animal_type == animal_type) q = q.filter(orm.Pet.animal_type == animal_type)
return [p.dump() for p in q][:limit] return [p.dump() for p in q][:limit]
def get_pet(pet_id): def get_pet(pet_id):
pet = db_session.query(orm.Pet).filter(orm.Pet.id == pet_id).one_or_none() pet = db_session.query(orm.Pet).filter(orm.Pet.id == pet_id).one_or_none()
return pet.dump() or ('Not found', 404) return pet.dump() or ('Not found', 404)
@@ -51,6 +53,7 @@ app.add_api('swagger.yaml')
application = app.app application = app.app
@application.teardown_appcontext @application.teardown_appcontext
def shutdown_session(exception=None): def shutdown_session(exception=None):
db_session.remove() db_session.remove()

9
examples/sqlalchemy/orm.py
View File

@@ -12,8 +12,7 @@ class Pet(Base):
animal_type = Column(String(20)) animal_type = Column(String(20))
created = Column(DateTime()) created = Column(DateTime())
def update(self, id=None, name=None, animal_type=None, tags=None, def update(self, id=None, name=None, animal_type=None, tags=None, created=None):
created=None):
if name is not None: if name is not None:
self.name = name self.name = name
if animal_type is not None: if animal_type is not None:
@@ -22,14 +21,12 @@ class Pet(Base):
self.created = created self.created = created
def dump(self): def dump(self):
return dict([(k,v) for k,v in vars(self).items() return dict([(k, v) for k, v in vars(self).items() if not k.startswith('_')])
if not k.startswith('_')])
def init_db(uri): def init_db(uri):
engine = create_engine(uri, convert_unicode=True) engine = create_engine(uri, convert_unicode=True)
db_session = scoped_session(sessionmaker(autocommit=False, autoflush=False, db_session = scoped_session(sessionmaker(autocommit=False, autoflush=False, bind=engine))
bind=engine))
Base.query = db_session.query_property() Base.query = db_session.query_property()
Base.metadata.create_all(bind=engine) Base.metadata.create_all(bind=engine)
return db_session return db_session