mirror of
https://github.com/LukeHagar/connexion.git
synced 2025-12-06 12:27:45 +00:00
1012721d2d
* Upgrade isort==5.9.1 * isort --thirdparty connexion,aiohttp ./tests * isort --thirdparty aiohttp,connexion --check-only --diff . * cd tests before running isort
64 lines
1.6 KiB
Python
Executable File
64 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import datetime
|
|
import logging
|
|
|
|
import connexion
|
|
import orm
|
|
from connexion import NoContent
|
|
|
|
db_session = None
|
|
|
|
|
|
def get_pets(limit, animal_type=None):
|
|
q = db_session.query(orm.Pet)
|
|
if animal_type:
|
|
q = q.filter(orm.Pet.animal_type == animal_type)
|
|
return [p.dump() for p in q][:limit]
|
|
|
|
|
|
def get_pet(pet_id):
|
|
pet = db_session.query(orm.Pet).filter(orm.Pet.id == pet_id).one_or_none()
|
|
return pet.dump() if pet is not None else ('Not found', 404)
|
|
|
|
|
|
def put_pet(pet_id, pet):
|
|
p = db_session.query(orm.Pet).filter(orm.Pet.id == pet_id).one_or_none()
|
|
pet['id'] = pet_id
|
|
if p is not None:
|
|
logging.info('Updating pet %s..', pet_id)
|
|
p.update(**pet)
|
|
else:
|
|
logging.info('Creating pet %s..', pet_id)
|
|
pet['created'] = datetime.datetime.utcnow()
|
|
db_session.add(orm.Pet(**pet))
|
|
db_session.commit()
|
|
return NoContent, (200 if p is not None else 201)
|
|
|
|
|
|
def delete_pet(pet_id):
|
|
pet = db_session.query(orm.Pet).filter(orm.Pet.id == pet_id).one_or_none()
|
|
if pet is not None:
|
|
logging.info('Deleting pet %s..', pet_id)
|
|
db_session.query(orm.Pet).filter(orm.Pet.id == pet_id).delete()
|
|
db_session.commit()
|
|
return NoContent, 204
|
|
else:
|
|
return NoContent, 404
|
|
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
db_session = orm.init_db('sqlite:///:memory:')
|
|
app = connexion.FlaskApp(__name__)
|
|
app.add_api('openapi.yaml')
|
|
|
|
application = app.app
|
|
|
|
|
|
@application.teardown_appcontext
|
|
def shutdown_session(exception=None):
|
|
db_session.remove()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(port=8081, use_reloader=False, threaded=False)
|