apiflask

admin

1. from apiflask import APIFlask, Schema, abort
   
2. from apiflask.fields import Integer, String
   
3. from apiflask.validators import Length, OneOf
   
4. 
   
5. app = APIFlask(__name__)
   
6. 
   
7. pets = [
   
8.     {'id': 0, 'name': 'Kitty', 'category': 'cat'},
   
9.     {'id': 1, 'name': 'Coco', 'category': 'dog'},
   
10.     {'id': 2, 'name': 'Flash', 'category': 'cat'}
    
11. ]
    
12. 
    
13. 
    
14. class PetIn(Schema):
    
15.     name = String(required=True, validate=Length(0, 10))
    
16.     category = String(required=True, validate=OneOf(['dog', 'cat']))
    
17. 
    
18. 
    
19. class PetOut(Schema):
    
20.     id = Integer()
    
21.     name = String()
    
22.     category = String()
    
23. 
    
24. 
    
25. @app.get('/')
    
26. def say_hello():
    
27.     return {'message': 'Hello!'}
    
28. 
    
29. 
    
30. @app.get('/pets/<int:pet_id>')
    
31. @app.output(PetOut)
    
32. def get_pet(pet_id):
    
33.     if pet_id > len(pets) - 1 or pets[pet_id].get('deleted'):
    
34.         abort(404)
    
35.     return pets[pet_id]
    
36. 
    
37. 
    
38. @app.get('/pets')
    
39. @app.output(PetOut(many=True))
    
40. def get_pets():
    
41.     return pets
    
42. 
    
43. 
    
44. @app.post('/pets')
    
45. @app.input(PetIn, location='json')
    
46. @app.output(PetOut, status_code=201)
    
47. def create_pet(json_data):
    
48.     pet_id = len(pets)
    
49.     json_data['id'] = pet_id
    
50.     pets.append(json_data)
    
51.     return pets[pet_id]
    
52. 
    
53. 
    
54. @app.patch('/pets/<int:pet_id>')
    
55. @app.input(PetIn(partial=True), location='json')
    
56. @app.output(PetOut)
    
57. def update_pet(pet_id, json_data):
    
58.     if pet_id > len(pets) - 1:
    
59.         abort(404)
    
60.     for attr, value in json_data.items():
    
61.         pets[pet_id][attr] = value
    
62.     return pets[pet_id]
    
63. 
    
64. 
    
65. @app.delete('/pets/<int:pet_id>')
    
66. @app.output({}, status_code=204)
    
67. def delete_pet(pet_id):
    
68.     if pet_id > len(pets) - 1:
    
69.         abort(404)
    
70.     pets[pet_id]['deleted'] = True
    
71.     pets[pet_id]['name'] = 'Ghost'
    
72.     return ''

复制代码

admin

数据库结合

1. from apiflask import APIFlask, Schema
   
2. from apiflask.fields import Integer, String
   
3. from apiflask.validators import Length, OneOf
   
4. from flask_sqlalchemy import SQLAlchemy
   
5. 
   
6. app = APIFlask(__name__)
   
7. app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'
   
8. app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
   
9. 
   
10. db = SQLAlchemy(app)
    
11. 
    
12. 
    
13. class PetModel(db.Model):
    
14.     id = db.Column(db.Integer, primary_key=True)
    
15.     name = db.Column(db.String(10))
    
16.     category = db.Column(db.String(10))
    
17. 
    
18. 
    
19. def init_database():
    
20.     db.create_all()
    
21. 
    
22.     pets = [
    
23.         {'name': 'Kitty', 'category': 'cat'},
    
24.         {'name': 'Coco', 'category': 'dog'},
    
25.         {'name': 'Flash', 'category': 'cat'}
    
26.     ]
    
27.     for pet_data in pets:
    
28.         pet = PetModel(**pet_data)
    
29.         db.session.add(pet)
    
30.     db.session.commit()
    
31. 
    
32. 
    
33. class PetIn(Schema):
    
34.     name = String(required=True, validate=Length(0, 10))
    
35.     category = String(required=True, validate=OneOf(['dog', 'cat']))
    
36. 
    
37. 
    
38. class PetOut(Schema):
    
39.     id = Integer()
    
40.     name = String()
    
41.     category = String()
    
42. 
    
43. 
    
44. @app.get('/')
    
45. def say_hello():
    
46.     return {'message': 'Hello!'}
    
47. 
    
48. 
    
49. @app.get('/pets/<int:pet_id>')
    
50. @app.output(PetOut)
    
51. def get_pet(pet_id):
    
52.     return db.get_or_404(PetModel, pet_id)
    
53. 
    
54. 
    
55. @app.get('/pets')
    
56. @app.output(PetOut(many=True))
    
57. def get_pets():
    
58.     return PetModel.query.all()
    
59. 
    
60. 
    
61. @app.post('/pets')
    
62. @app.input(PetIn, location='json')
    
63. @app.output(PetOut, status_code=201)
    
64. def create_pet(json_data):
    
65.     pet = PetModel(**json_data)
    
66.     db.session.add(pet)
    
67.     db.session.commit()
    
68.     return pet
    
69. 
    
70. 
    
71. @app.patch('/pets/<int:pet_id>')
    
72. @app.input(PetIn(partial=True), location='json')
    
73. @app.output(PetOut)
    
74. def update_pet(pet_id, json_data):
    
75.     pet = db.get_or_404(PetModel, pet_id)
    
76.     for attr, value in json_data.items():
    
77.         setattr(pet, attr, value)
    
78.     db.session.commit()
    
79.     return pet
    
80. 
    
81. 
    
82. @app.delete('/pets/<int:pet_id>')
    
83. @app.output({}, status_code=204)
    
84. def delete_pet(pet_id):
    
85.     pet = db.get_or_404(PetModel, pet_id)
    
86.     db.session.delete(pet)
    
87.     db.session.commit()
    
88.     return ''
    
89. 
    
90. 
    
91. with app.app_context():
    
92.     init_database()

复制代码