def lambda_handler(event, context):
"""
crud lambda application...
"""
headers = {
"Content-Type": "application/json",
"Access-Allow-Credentials": "false",
"Access-Control-Allow-Headers": "Content-Type",
"Access-Control-Allow-Origin": "*",
"Access-Control-Allow-Methods": "POST,GET,PUT,DELETE,OPTIONS",
"Access-Control-Max-Age": "86400",
}
filters = {
"/get_by_list": "todolist",
"/get_by_owner": "owner",
"/get_by_status": "status",
"/get_by_user": "******",
"/get_by_access": "access",
}
path = event["path"]
querystring_parameters = event["queryStringParameters"]
query_filter = querystring_parameters[filters[path]]
if filters[path] == "todolist":
queryset = ToDo.objects(todolist=query_filter).all()
elif filters[path] == "owner":
owner = query_filter.lower()
if "list" in querystring_parameters.keys():
list_mode = querystring_parameters["list"].lower() == "true"
queryset = (ToDoList.objects(owner=owner).all()
if list_mode else ToDo.objects(owner=owner).all())
else:
queryset = list(ToDoList.objects(owner=owner).all()) + list(
ToDo.objects(owner=owner).all())
elif filters[path] == "user":
queryset = ToDoList.objects(owner=query_filter).all()
tmp = ToDoList.objects.all()
contrib = list(
filter(
lambda x: query_filter in x.access["contributors"] or
query_filter in x.access["readers"],
tmp,
))
queryset = list(queryset) + contrib
elif filters[path] == "access":
raw_data = ToDoList.objects.all(
) # raw_data é a coleção de todos as listas
username = querystring_parameters["username"].lower(
) # username é um param que passamos
if query_filter == "all":
processed_data = {
"reader": [
todolist.serialize() for todolist in list(
filter(
lambda x: username in x.access["readers"],
raw_data,
))
],
"contributor": [
todolist.serialize() for todolist in list(
filter(
lambda x: username in x.access["contributors"],
raw_data,
))
],
}
else:
processed_data = [
x.serialize() for x in
list( # se o parametro for qualquer outra coisa, ele vai procurar dentro de access pelo nome no param
filter(
lambda x: username in x.access[
query_filter
], # isso diminui o tamanho da prog e tb deixa aberto pra novas roles, caso aconteça
raw_data,
))
]
return {"statusCode": 200, "headers": headers, "body": processed_data}
else:
status = query_filter.lower()
if "list" in querystring_parameters.keys():
list_mode = querystring_parameters["list"].lower() == "true"
queryset = (ToDoList.objects(status=status).all()
if list_mode else ToDo.objects(status=status).all())
else:
queryset = list(ToDoList.objects(status=status).all()) + list(
ToDo.objects(status=status).all())
queryset = [x.serialize() for x in queryset]
return {
"statusCode": 200,
"headers": headers,
"body": json.dumps(queryset, ),
}
def lambda_handler(event, context):
"""
crud lambda application...
"""
db = config("MONGO_DB")
host = config("MONGO_HOST")
conn = mongoengine.connect(host=host, db=db)
headers = {
"Content-Type": "application/json",
"Access-Allow-Credentials": "false",
"Access-Control-Allow-Headers": "Content-Type",
"Access-Control-Allow-Origin": "*",
"Access-Control-Allow-Methods": "POST,GET,PUT,DELETE,OPTIONS",
"Access-Control-Max-Age": "86400",
}
method = event["httpMethod"]
path = event["path"]
filter_paths = [
"/get_by_status", "/get_by_owner", "/get_by_list", "/get_by_user",
"/get_by_access"
]
status = {"get": 200, "post": 200, "put": 200, "delete": 200}
body = event["body"]
f = json.loads(body) if body else {}
querystring_parameters = (event["queryStringParameters"]
if event["queryStringParameters"] else {})
if method == "OPTIONS":
return {"statusCode": 204, "headers": headers}
if path in filter_paths:
return filters.lambda_handler(event, context)
if path == "/todolist" or path == "/add_user":
return app_todolist.lambda_handler(event, context)
if method == "GET":
if "id" in querystring_parameters.keys():
message = (ToDo.objects(
id=ObjectId(querystring_parameters["id"])).first().serialize())
else:
message = [i.serialize() for i in ToDo.objects.all()]
if method == "POST":
try:
deadline = [int(i) for i in f["deadline"].split("-")]
deadline = date(year=deadline[0],
month=deadline[1],
day=deadline[2])
ToDo.objects.create(
name=f["name"],
description=f["description"],
owner=f["owner"],
todolist=f["todolist"],
assignment=f["assignment"],
created_at=datetime.now(),
last_update={
"user": f["owner"],
"date": datetime.now()
},
deadline=deadline,
notification=deadline - timedelta(days=3),
attachments=f["attachments"],
status=None,
).save()
message = [ToDo.objects(name=f["name"]).first().serialize()]
except (ValidationError, KeyError):
return {
"statusCode": 500,
"headers": headers,
"body": json.dumps("Invalid request body"),
}
elif method == "PUT":
try:
_id = ObjectId(querystring_parameters["id"])
obj = ToDo.objects(id=_id).first()
d = obj.last_update
# d["user"] = f["user"]
d["date"] = datetime.now()
# f.pop("user")
f["last_update"] = d
obj.update(**f)
d["todo"] = str(obj.pk)
ToDoList.objects(id=ObjectId(obj.todolist)).update(last_update=d)
obj = ToDo.objects(id=_id).first()
message = obj.serialize()
except (ValidationError, KeyError) as e:
return {
"statusCode": 500,
"headers": headers,
"body": json.dumps(e)
}
elif method == "DELETE":
_id = ObjectId(querystring_parameters["id"])
ToDo.objects(id=_id).first().delete()
message = "Object deleted successfully"
return {
"headers": headers,
"statusCode": status[method.lower()],
"body": json.dumps(message),
}