Beispiel #1
0
def post_post_meta(event, context):
    table = __get_table_client()
    company_id = event["pathParameters"]["company_id"]
    # post_id = event["pathParameters"]["post_id"]
    payload = json.loads(event["body"])
    payload.update({"company_id": company_id})
    post = Post(payload)

    keys = post.get_keys()
    PK, SK = keys["PK"], keys["SK"]
    print("Key: ", json.dumps({"PK": PK, "SK": SK}, indent=4))
    table_record = post.get_record()

    try:
        table.put_item(
            Item=table_record,
            ConditionExpression=
            "attribute_not_exists(PK) and attribute_not_exists(SK)",
            ReturnConsumedCapacity="TOTAL",
        )
    except ClientError as e:
        if e.response["Error"]["Code"] == "ConditionalCheckFailedException":
            return _response(409, {"status": "Item already exists"})
        print(e.response["Error"]["Message"])
        return _response(500, {"status": "DynamoDB Client Error"})
    else:
        print("PutItem succeeded:")
        print(json.dumps(table_record, indent=4, cls=DecimalEncoder))
    item = post.get_item()
    return _response(201, item)
Beispiel #2
0
def put_post_meta(event, context):

    # table = __get_table_client()

    # payload = json.loads(event["body"])

    # (
    #     PK,
    #     SK,
    #     company_id,
    #     user_id,
    #     post_id,
    #     post_title,
    #     post_content,
    #     can_share_on,
    #     points_map,
    #     created_at,
    #     updated_at,
    # ) = _get_post_meta(payload)

    # print(
    #     PK,
    #     SK,
    #     company_id,
    #     user_id,
    #     post_id,
    #     post_title,
    #     post_content,
    #     can_share_on,
    #     points_map,
    #     created_at,
    #     updated_at,
    # )
    # import time

    # post_created_at = int(time.time())
    # try:
    #     table.update_item(
    #         Key={"PK": PK, "SK": SK},
    #         UpdateExpression="SET #company_id = :company_id, #user_id = :user_id, #post_id = :post_id, #post_title = :post_title, #post_content = :post_content, #can_share_on = :can_share_on, #points_map = :points_map, #created_at = if_not_exists(#created_at, :created_at), #post_created_at = if_not_exists(#post_created_at, :post_created_at), #updated_at = :updated_at",
    #         ExpressionAttributeNames={
    #             "#company_id": "company_id",
    #             "#user_id": "user_id",
    #             "#post_id": "post_id",
    #             "#post_title": "post_title",
    #             "#post_content": "post_content",
    #             "#can_share_on": "can_share_on",
    #             "#points_map": "points_map",
    #             "#post_created_at": "post_created_at",
    #             "#created_at": "created_at",
    #             "#updated_at": "updated_at",
    #         },
    #         ExpressionAttributeValues={
    #             ":company_id": company_id,
    #             ":user_id": user_id,
    #             ":post_id": post_id,
    #             ":post_title": post_title,
    #             ":post_content": post_content,
    #             ":can_share_on": can_share_on,
    #             ":points_map": points_map,
    #             ":created_at": created_at,
    #             ":updated_at": updated_at,
    #             ":post_created_at": post_created_at,
    #         },
    #         ReturnConsumedCapacity="TOTAL",
    #     )
    #     payload["post_id"] = post_id

    # except ClientError as e:
    #     print(e.response["Error"]["Message"])
    #     return _response(500, {"status": "DynamoDB Client Error"})
    # except KeyError as e:
    #     print(e)
    #     return _response(404, {"status": "ITEM NOT FOUND"})
    # else:
    #     print("PutItem succeeded:")
    #     print(json.dumps(payload, indent=4, cls=DecimalEncoder))

    # if json.loads(event["body"]).get("post_id"):
    #     return _response(200, payload)
    # return _response(201, payload)
    table = __get_table_client()
    company_id = event["pathParameters"]["company_id"]
    post_id = event["pathParameters"]["post_id"]
    created_at = event["queryStringParameters"]["created_at"]

    payload = json.loads(event["body"])
    payload.update({
        "company_id": company_id,
        "post_id": post_id,
        created_at: "created_at"
    })
    post = Post(payload)
    table_record = post.get_record()
    # PK = table_record["PK"]
    # SK = table_record["SK"]
    try:
        response = table.put_item(
            ConditionExpression="attribute_exists(PK) and attribute_exists(SK)",
            Item=table_record,
            ReturnConsumedCapacity="TOTAL",
        )

    except ClientError as e:
        if e.response["Error"]["Code"] == "ConditionalCheckFailedException":
            return _response(404, {"status": "ITEM NOT FOUND"})

        print(e.response["Error"]["Message"])
        return _response(500, {"status": "DynamoDB Client Error"})
    else:
        print("PutItem succeeded:")
        print(json.dumps(table_record, indent=4, cls=DecimalEncoder))
    # print("Item update response: ", response)
    # if not response.get("Attributes"):
    #     return _response(404, {"status": "ITEM NOT FOUND"})
    # item = UserMeta(response["Attributes"]).get_item()
    item = post.get_item()
    return _response(200, item)