-
Notifications
You must be signed in to change notification settings - Fork 0
/
schemas.py
47 lines (36 loc) · 1.39 KB
/
schemas.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from marshmallow import Schema, fields
# validation for requests (JSON payload) and responses
# POST
class PlainItemSchema(Schema):
id = fields.Int(dump_only=True)
name = fields.Str(required=True)
price = fields.Float(required=True)
# PUT/UPDATE
class ItemUpdateSchema(Schema):
name = fields.Str()
price = fields.Float()
store_id = fields.Int()
# POST
class PlainStoreSchema(Schema):
id = fields.Int(dump_only=True)
name = fields.Str(required=True)
class PlainTagSchema(Schema):
id = fields.Int(dump_only=True)
name = fields.Str()
# ----- inherited from Schema because they are nested ----------
# fields.Nested(....)
class ItemSchema(PlainItemSchema):
store_id = fields.Int(required=True, load_only=True)
store = fields.Nested(PlainStoreSchema(), dump_only=True)
tags = fields.List(fields.Nested(PlainTagSchema()), dump_only=True)
class TagSchema(PlainTagSchema):
store_id = fields.Int(load_only=True)
store = fields.Nested(PlainStoreSchema(), dump_only=True)
items = fields.List(fields.Nested(PlainItemSchema()), dump_only=True)
class StoreSchema(PlainStoreSchema):
items = fields.List(fields.Nested(PlainStoreSchema()), dump_only=True)
tags = fields.List(fields.Nested(PlainTagSchema()), dump_only=True)
class TagAndItemSchema(Schema):
message = fields.Str()
item = fields.Nested(ItemSchema)
tag = fields.Nested(TagSchema)