fixing confirm and code in schema

This commit is contained in:
Cayo Puigdefabregas 2021-06-07 15:45:04 +02:00
parent efbc74f7fb
commit 8289f24b69
1 changed files with 13 additions and 7 deletions

View File

@ -570,23 +570,29 @@ class Trade(ActionWithMultipleDevices):
* without confirmation * without confirmation
""" """
if not (data['user_from_email'] or data['user_to_email']):
txt = "you need one user from or user to for to do a offer"
raise ValidationError(txt)
if data['user_from_email']: if data['user_from_email']:
user_from = User.query.filter_by(email=data['user_from_email']).one() user_from = User.query.filter_by(email=data['user_from_email']).one()
data['user_from'] = user_from data['user_from'] = user_from
else:
data['confirm'] = False @validates_schema
def validate_email_users(self, data: dict):
"""We need at least one user"""
if not (data['user_from_email'] or data['user_to_email']):
txt = "you need one user from or user to for to do a trade"
raise ValidationError(txt)
if not g.user.email in [data['user_from_email'], data['user_to_email']]:
txt = "you need to be one of participate of the action"
raise ValidationError(txt)
@validates_schema @validates_schema
def validate_code(self, data: dict): def validate_code(self, data: dict):
"""If the user not exist, you need a code to be able to do the traceability""" """If the user not exist, you need a code to be able to do the traceability"""
if data['user_from_email'] and data['user_to_email']: if data['user_from_email'] and data['user_to_email']:
data['confirm'] = True
return return
if not data.get('code'): if not data['confirm'] not data.get('code'):
txt = "you need a code to be able to do the traceability" txt = "you need a code to be able to do the traceability"
raise ValidationError(txt) raise ValidationError(txt)