295 lines
9.2 KiB
Python
295 lines
9.2 KiB
Python
import copy
|
|
|
|
from flask import g
|
|
from sqlalchemy.util import OrderedSet
|
|
from teal.marshmallow import ValidationError
|
|
|
|
from ereuse_devicehub.db import db
|
|
from ereuse_devicehub.resources.action.models import Trade, Confirm, ConfirmRevoke, Revoke
|
|
from ereuse_devicehub.resources.user.models import User
|
|
from ereuse_devicehub.resources.lot.views import delete_from_trade
|
|
|
|
|
|
class TradeView():
|
|
"""Handler for manager the trade action register from post
|
|
|
|
request_post = {
|
|
'type': 'Trade',
|
|
'devices': [device_id],
|
|
'documents': [document_id],
|
|
'userFrom': user2.email,
|
|
'userTo': user.email,
|
|
'price': 10,
|
|
'date': "2020-12-01T02:00:00+00:00",
|
|
'lot': lot['id'],
|
|
'confirm': True,
|
|
}
|
|
|
|
"""
|
|
|
|
def __init__(self, data, resource_def, schema):
|
|
self.schema = schema
|
|
a = resource_def.schema.load(data)
|
|
a.pop('user_to_email', '')
|
|
a.pop('user_from_email', '')
|
|
self.trade = Trade(**a)
|
|
self.create_phantom_account()
|
|
db.session.add(self.trade)
|
|
self.create_automatic_trade()
|
|
self.create_confirmations()
|
|
|
|
def post(self):
|
|
db.session().final_flush()
|
|
ret = self.schema.jsonify(self.trade)
|
|
ret.status_code = 201
|
|
db.session.commit()
|
|
return ret
|
|
|
|
def create_confirmations(self) -> None:
|
|
"""Do the first confirmation for the user than do the action"""
|
|
|
|
# if the confirmation is mandatory, do automatic confirmation only for
|
|
# owner of the lot
|
|
if self.trade.confirm:
|
|
if self.trade.devices:
|
|
confirm_devs = Confirm(user=g.user,
|
|
action=self.trade,
|
|
devices=self.trade.devices)
|
|
db.session.add(confirm_devs)
|
|
|
|
if self.trade.documents:
|
|
confirm_docs = Confirm(user=g.user,
|
|
action=self.trade,
|
|
documents=self.trade.documents)
|
|
db.session.add(confirm_docs)
|
|
return
|
|
|
|
# check than the user than want to do the action is one of the users
|
|
# involved in the action
|
|
if not g.user in [self.trade.user_from, self.trade.user_to]:
|
|
txt = "You do not participate in this trading"
|
|
raise ValidationError(txt)
|
|
|
|
if self.trade.user_from == g.user or self.trade.user_from.phantom:
|
|
confirm_from = Confirm(user=self.trade.user_from,
|
|
action=self.trade,
|
|
devices=self.trade.devices)
|
|
db.session.add(confirm_from)
|
|
|
|
if self.trade.user_to == g.user or self.trade.user_to.phantom:
|
|
confirm_to = Confirm(user=self.trade.user_to,
|
|
action=self.trade,
|
|
devices=self.trade.devices)
|
|
db.session.add(confirm_to)
|
|
|
|
def create_phantom_account(self) -> None:
|
|
"""
|
|
If exist both users not to do nothing
|
|
If exist from but not to:
|
|
search if exist in the DB
|
|
if exist use it
|
|
else create new one
|
|
The same if exist to but not from
|
|
|
|
"""
|
|
if self.trade.user_from and self.trade.user_to:
|
|
return
|
|
|
|
if self.trade.user_from and not self.trade.user_to:
|
|
assert g.user == self.trade.user_from
|
|
email = "{}_{}@dhub.com".format(str(self.trade.user_from.id), self.trade.code)
|
|
users = User.query.filter_by(email=email)
|
|
if users.first():
|
|
user = users.first()
|
|
self.trade.user_to = user
|
|
return
|
|
|
|
user = User(email=email, password='', active=False, phantom=True)
|
|
db.session.add(user)
|
|
self.trade.user_to = user
|
|
|
|
if not self.trade.user_from and self.trade.user_to:
|
|
email = "{}_{}@dhub.com".format(str(self.trade.user_to.id), self.trade.code)
|
|
users = User.query.filter_by(email=email)
|
|
if users.first():
|
|
user = users.first()
|
|
self.trade.user_from = user
|
|
return
|
|
|
|
user = User(email=email, password='', active=False, phantom=True)
|
|
db.session.add(user)
|
|
self.trade.user_from = user
|
|
|
|
def create_automatic_trade(self) -> None:
|
|
# not do nothing if it's neccesary confirmation explicity
|
|
if self.trade.confirm:
|
|
return
|
|
|
|
# Change the owner for every devices
|
|
for dev in self.trade.devices:
|
|
dev.owner = self.trade.user_to
|
|
if hasattr(dev, 'components'):
|
|
for c in dev.components:
|
|
c.owner = self.trade.user_to
|
|
|
|
|
|
class ConfirmMixin():
|
|
"""
|
|
Very Important:
|
|
==============
|
|
All of this Views than inherit of this class is executed for users
|
|
than is not owner of the Trade action.
|
|
|
|
The owner of Trade action executed this actions of confirm and revoke from the
|
|
lot
|
|
|
|
"""
|
|
|
|
Model = None
|
|
|
|
def __init__(self, data, resource_def, schema):
|
|
self.schema = schema
|
|
a = resource_def.schema.load(data)
|
|
self.validate(a)
|
|
if not a['devices'] and not a['documents']:
|
|
raise ValidationError('Devices not exist.')
|
|
self.model = self.Model(**a)
|
|
|
|
def post(self):
|
|
db.session().final_flush()
|
|
ret = self.schema.jsonify(self.model)
|
|
ret.status_code = 201
|
|
db.session.commit()
|
|
return ret
|
|
|
|
|
|
class ConfirmView(ConfirmMixin):
|
|
"""Handler for manager the Confirmation register from post
|
|
|
|
request_confirm = {
|
|
'type': 'Confirm',
|
|
'action': trade.id,
|
|
'devices': [device_id]
|
|
}
|
|
"""
|
|
|
|
Model = Confirm
|
|
|
|
def validate(self, data):
|
|
"""If there are one device than have one confirmation,
|
|
then remove the list this device of the list of devices of this action
|
|
"""
|
|
real_devices = []
|
|
for dev in data['devices']:
|
|
ac = dev.last_action_trading
|
|
if ac.type == Confirm.t and not ac.user == g.user:
|
|
real_devices.append(dev)
|
|
|
|
data['devices'] = OrderedSet(real_devices)
|
|
|
|
# Change the owner for every devices
|
|
for dev in data['devices']:
|
|
user_to = data['action'].user_to
|
|
dev.change_owner(user_to)
|
|
|
|
|
|
class RevokeView(ConfirmMixin):
|
|
"""Handler for manager the Revoke register from post
|
|
|
|
request_revoke = {
|
|
'type': 'Revoke',
|
|
'action': trade.id,
|
|
'devices': [device_id],
|
|
}
|
|
|
|
"""
|
|
|
|
Model = Revoke
|
|
|
|
def __init__(self, data, resource_def, schema):
|
|
self.schema = schema
|
|
a = resource_def.schema.load(data)
|
|
self.validate(a)
|
|
|
|
def validate(self, data):
|
|
"""All devices need to have the status of DoubleConfirmation."""
|
|
|
|
### check ###
|
|
if not data['devices']:
|
|
raise ValidationError('Devices not exist.')
|
|
|
|
for dev in data['devices']:
|
|
if not dev.trading == 'TradeConfirmed':
|
|
txt = 'Some of devices do not have enough to confirm for to do a revoke'
|
|
ValidationError(txt)
|
|
### End check ###
|
|
|
|
ids = {d.id for d in data['devices']}
|
|
lot = data['action'].lot
|
|
# import pdb; pdb.set_trace()
|
|
self.model = delete_from_trade(lot, ids)
|
|
|
|
# devices = set(data['devices'])
|
|
# without_confirms = set() # set of devs without confirms of user2
|
|
|
|
# if g.user == lot.trade.author:
|
|
# for dev in devices:
|
|
# ac = dev.last_action_trading
|
|
# if ac.type == 'Confirm' and ac.user == g.user:
|
|
# without_confirms.add(dev)
|
|
|
|
# # we need to mark one revoke for every devs
|
|
# revoke = Revoke(action=lot.trade, user=g.user, devices=devices)
|
|
# db.session.add(revoke)
|
|
|
|
# if without_confirms:
|
|
# confirm_revoke = ConfirmRevoke(
|
|
# action=revoke,
|
|
# user=g.user,
|
|
# devices=without_confirms
|
|
# )
|
|
# db.session.add(confirm_revoke)
|
|
|
|
# lot.devices.difference_update(without_confirms)
|
|
# lot.trade.devices = lot.devices
|
|
|
|
# self.model = revoke
|
|
|
|
|
|
class ConfirmRevokeView(ConfirmMixin):
|
|
"""Handler for manager the Confirmation register from post
|
|
|
|
request_confirm_revoke = {
|
|
'type': 'ConfirmRevoke',
|
|
'action': action_revoke.id,
|
|
'devices': [device_id]
|
|
}
|
|
|
|
"""
|
|
|
|
Model = ConfirmRevoke
|
|
|
|
def validate(self, data):
|
|
"""All devices need to have the status of revoke."""
|
|
|
|
if not data['action'].type == 'Revoke':
|
|
txt = 'Error: this action is not a revoke action'
|
|
ValidationError(txt)
|
|
|
|
for dev in data['devices']:
|
|
if not dev.trading == 'Revoke':
|
|
txt = 'Some of devices do not have revoke to confirm'
|
|
ValidationError(txt)
|
|
|
|
devices = OrderedSet(data['devices'])
|
|
data['devices'] = devices
|
|
|
|
# Change the owner for every devices
|
|
# data['action'] == 'Revoke'
|
|
|
|
trade = data['action'].action
|
|
for dev in devices:
|
|
dev.reset_owner()
|
|
|
|
trade.lot.devices.difference_update(devices)
|