def create(event, context):
data = json.loads(event['body'])
customer_id = data['customer_id']
product_id = data['product_id']
basket = None
# check if the item is in the basket already, if so we're gonna just add to it
try:
basket = Basket.get(customer_id, product_id)
if basket is not None:
basket.quantity = basket.quantity + data['quantity']
except:
pass
if basket is None:
basket = Basket(customer_id=customer_id,
product_id=product_id,
quantity=data['quantity'],
price=data['price'])
basket.save()
return {"statusCode": 200, "body": basket.body()}
def get(event, context):
customer_id = event['pathParameters']['id']
item_id = event['pathParameters']['product_id']
try:
basket = Basket.get(customer_id, item_id)
return {'statusCode': 200, 'body': basket.body()}
except:
raise Exception("Invalid request")
def update(event, context):
# if 'requestContext' in event:
# if 'identity' in event['requestContext']:
# if 'cognitoIdentityId' in event['requestContext']['identity']:
# id = event['requestContext']['identity']['cognitoIdentityId']
# else:
id = event['pathParameters']['id']
data = json.loads(event['body'])
product_id = data['product_id']
try:
basket = Basket.get(id, product_id)
basket.quantity = data['quantity']
basket.save()
return {'statusCode': 200, 'body': {basket.body()}}
except Exception as ex:
logger.error(ex)
raise Exception("Invalid request")
