trustchain-oc1-orchestral
/
devicehub-teal
Archived
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

clean code

This commit is contained in:
Cayo Puigdefabregas 2020-11-23 15:53:25 +01:00
parent 73716a7331
commit 78c1f13260
1 changed files with 22 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
ereuse_devicehub/resources/action/schemas.py
View File

@ -68,43 +68,44 @@ class Allocate(ActionWithMultipleDevices):
agent = NestedOn(s_agent.Agent, only_query='id', required=False, comment=m.Trade.to_comment) agent = NestedOn(s_agent.Agent, only_query='id', required=False, comment=m.Trade.to_comment)
description = SanitizedStr(default='', description=m.Action.description.comment) description = SanitizedStr(default='', description=m.Action.description.comment)
start_time = DateTime(data_key='start_time', description=m.Action.start_time.comment) start_time = DateTime(data_key='start_time', description=m.Action.start_time.comment)
end_time = DateTime(data_key='end_time', description=m.Action.end_time.comment) end_time = DateTime(data_key='end_time', required=False,
description=m.Action.end_time.comment)
code = SanitizedStr(data_key='Transaction', validate=Length(min=1, max=STR_BIG_SIZE), code = SanitizedStr(data_key='Transaction', validate=Length(min=1, max=STR_BIG_SIZE),
required=False, required=False,
description='The code of the agent to assigned.') description='The code of the agent to assigned.')
end_users = Integer(validate=[Range(min=1, error="Value must be greater than 0")], end_users = Integer(validate=[Range(min=1, error="Value must be greater than 0")],
required=True) required=True)
@validates_schema @validates_schema
def validate_allocate(self, data: dict): def validate_allocate(self, data: dict):
txt = "You need deallocate before allocate this device again"
for device in data['devices']: for device in data['devices']:
if device.allocated == False:
device.allocated = True
continue
actions = [a for a in device.actions] actions = [a for a in device.actions]
actions.sort(key=lambda x: x.created) actions.sort(key=lambda x: x.created)
actions.reverse() actions.reverse()
allocate = None
for a in actions: for allocate in actions:
if isinstance(a, m.Allocate): if isinstance(allocate, m.Allocate):
allocate = a same_allocate = [
break allocate.code == data['code'],
if isinstance(a, m.Deallocate): allocate.start_time == data['start_time'],
allocate.end_users == data['end_users']
]
if not all(same_allocate):
raise ValidationError(txt)
if isinstance(allocate, m.Deallocate):
break break
if allocate: device.allocated = True
txt = "You need deallocate before allocate this device again"
same_allocate = [
allocate.code == data['code'],
allocate.start_time == data['start_time'],
allocate.end_users == data['end_users']
]
if not all(same_allocate):
raise ValidationError(txt)
device.allocated = True
class Deallocate(ActionWithMultipleDevices): class Deallocate(ActionWithMultipleDevices):
__doc__ = m.Deallocate.__doc__ __doc__ = m.Deallocate.__doc__
start_time = DateTime(data_key='start_time', description=m.Action.start_time.comment)
@validates_schema @validates_schema
def validate_deallocate(self, data: dict): def validate_deallocate(self, data: dict):