def put(self, order_id):
# Create or Update
data = request.get_json(force=True)
order = OrdersModel.find_by_order_id(order_id)
if order is None:
order = OrdersModel(data['action'], data['symbol'], data['price'],
data['order_type'], data['order_size'],
data['value'], data['account_id'])
else:
order.price = data['price']
order.order_size = data['order_size']
order.value = data['value']
order.order_type = data['order_type']
order.update_to_db()
return order.json()
def delete(self, order_id):
order = OrdersModel.find_by_order_id(order_id)
if order:
order.delete_from_db()
return {'message': 'order has been deleted'}
def get(self, order_id):
order = OrdersModel.find_by_order_id(order_id)
if order:
return jsonify({'orders': order.json()})
return {'message': 'Order not found'}, 404
