mirror of
https://github.com/Blockstream/satellite-api.git
synced 2025-03-13 11:35:20 +01:00
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
|
import json
|
||
|
|
from datetime import datetime, timedelta
|
||
|
|
|
||
|
|
from marshmallow import fields, Schema, validate, validates, ValidationError
|
||
|
|
|
||
|
|
import constants
|
||
|
|
|
||
|
|
|
||
|
|
class OrderSchema(Schema):
|
||
|
|
uuid = fields.String()
|
||
|
|
bid = fields.Integer()
|
||
|
|
message_size = fields.Integer()
|
||
|
|
bid_per_byte = fields.Float()
|
||
|
|
message_digest = fields.String()
|
||
|
|
status = fields.Function(
|
||
|
|
lambda obj: constants.OrderStatus(obj.status).name)
|
||
|
|
created_at = fields.DateTime()
|
||
|
|
cancelled_at = fields.DateTime()
|
||
|
|
started_transmission_at = fields.DateTime()
|
||
|
|
ended_transmission_at = fields.DateTime()
|
||
|
|
tx_seq_num = fields.Integer()
|
||
|
|
unpaid_bid = fields.Integer()
|
||
|
|
|
||
|
|
|
||
|
|
class OrderUploadReqSchema(Schema):
|
||
|
|
bid = fields.Int(required=True, validate=validate.Range(min=0))
|
||
|
|
message = fields.Str(validate=validate.Length(
|
||
|
|
max=constants.MAX_TEXT_MSG_LEN))
|
||
|
|
|
||
|
|
|
||
|
|
class OrderBumpSchema(Schema):
|
||
|
|
uuid = fields.String()
|
||
|
|
bid_increase = fields.Int(required=True, validate=validate.Range(min=0))
|
||
|
|
auth_token = fields.Str()
|
||
|
|
|
||
|
|
|
||
|
|
class OrdersSchema(Schema):
|
||
|
|
# When 'before' parameter is missing, set it to a time in near
|
||
|
|
# future (e.g. 5 seconds from now) to make sure none of the
|
||
|
|
# existing orders get filtered
|
||
|
|
before = fields.DateTime(
|
||
|
|
missing=lambda: datetime.utcnow() + timedelta(seconds=5), format='iso')
|
||
|
|
limit = fields.Int(missing=lambda: constants.PAGE_SIZE,
|
||
|
|
validate=validate.Range(min=1,
|
||
|
|
max=constants.MAX_PAGE_SIZE))
|
||
|
|
|
||
|
|
|
||
|
|
class TxConfirmationSchema(Schema):
|
||
|
|
regions = fields.String(required=True)
|
||
|
|
|
||
|
|
@validates('regions')
|
||
|
|
def must_be_json_array(self, data):
|
||
|
|
try:
|
||
|
|
js = json.loads(data)
|
||
|
|
if not isinstance(js, list) or len(js) < 1:
|
||
|
|
raise ValidationError("Invalid json array.")
|
||
|
|
except json.JSONDecodeError:
|
||
|
|
raise ValidationError("Invalid json array.")
|
||
|
|
|
||
|
|
|
||
|
|
class RxConfirmationSchema(Schema):
|
||
|
|
region = fields.Int(required=True)
|
||
|
|
|
||
|
|
|
||
|
|
order_schema = OrderSchema()
|
||
|
|
order_upload_req_schema = OrderUploadReqSchema()
|
||
|
|
order_bump_schema = OrderBumpSchema()
|
||
|
|
orders_schema = OrdersSchema()
|
||
|
|
tx_confirmation_schema = TxConfirmationSchema()
|
||
|
|
rx_confirmation_schema = RxConfirmationSchema()
|