LFLeaveGroup(if needed).py

# This file is designed for the lambda function to delete groups
# in the project CUThen, the final project for COMSE6998_010_2023_3,
# Topics in Computer Science: Cloud Computing and Big Data.


import json
from opensearchpy import OpenSearch, RequestsHttpConnection
from requests_aws4auth import AWS4Auth
import boto3


REGION = 'us-east-1'
HOST = 'search-cuthen-temp-5fyo5fvs7x7t2myle4ztwa7swa.us-east-1.es.amazonaws.com'
INDEX1 = 'user_to_group'
INDEX2 = 'group_to_user'
INDEX3 = 'user_to_inv'
INDEX0 = 'max_group_id'

def get_awsauth(region, service):
    cred = boto3.Session().get_credentials()
    return AWS4Auth(cred.access_key,
                    cred.secret_key,
                    region,
                    service,
                    session_token=cred.token)

def opensearch_init():
    opensearch_client = OpenSearch(hosts=[{
        'host': HOST,
        'port': 443
    }],
                        http_auth=get_awsauth(REGION, 'es'),
                        use_ssl=True,
                        verify_certs=True,
                        connection_class=RequestsHttpConnection)
    return opensearch_client
def del_by_index(INDEX):
    opensearch_client=opensearch_init()
    delete_query = {
        "query": {
            "match_all": {}
        }
    }
    opensearch_client.delete_by_query(index=INDEX, body=delete_query, refresh=True)

def ins_by_index(INDEX,document,id):
    opensearch_client=opensearch_init()
    # document = {"user_id": 1,"group_id": [2]}
    rm_res = opensearch_client.index(index=INDEX, id = id, body=document, refresh=True)

def search_by_index(INDEX,field,user_id):
    opensearch_client=opensearch_init()
    # field = "user_id"
    q3 = {
        "query": {
            "query_string": {
              "query": user_id,
                "fields": [field]
                }
            }
        }
    res3 = opensearch_client.search(index=INDEX, body=q3)
    
    hits = res3['hits']['hits']
    results = []
    for hit in hits:
        results.append(hit['_source'])

    print(f"{results =}")
    return results


def lambda_handler(event, context):
    print(event)

    # mock input data
    # input_data = {
    #     "user_id":1,
    #     "group_id":2,
    # }

    # TODO implement
    print(f"{event = }")
    print(f"{context = }")

    input_data = event['body']
    input_data = json.loads(input_data)

    user_id = input_data["user_id"]
    group_id = input_data["group_id"]
    opensearch_client = opensearch_init()

    print("1. opening opensearch")
    opensearch_client = opensearch_init()

    # adding it to user-group, group-user
    orginal_data={}
    updated_data={}
    print("2. adding it to user-group, group-user")
    result = search_by_index(INDEX1,"user_id",user_id)
    results1=result[0]['group_id']
    print(f"\tBefore user - Group id: {results1}")
    orginal_data["orginal_master_user_group_id"]=results1

    results1.remove(group_id)
    user_document = {"user_id": user_id, "group_id": results1}
    ins_by_index(INDEX1,user_document,user_id)

    result = search_by_index(INDEX1,"user_id",user_id)
    check_group_id=result[0]['group_id']
    print(f"\tAfter user -Group id: {check_group_id}")
    updated_data["user_master_group_id"]=check_group_id

    result = search_by_index(INDEX2,"group_id",group_id)
    results2=result[0]['user_id']
    print(f"\tBefore Group to user_id: {results2}")
    orginal_data["orginal_group_user_id"]=results2
    
    results2.remove(user_id)
    group_document = {"group_id":group_id, "user_id": results2}
    ins_by_index(INDEX2,group_document,group_id)

    result = search_by_index(INDEX2,"group_id",group_id)
    check_group_id=result[0]['user_id']
    print(f"\tAfter Group to user_id: {check_group_id}")
    updated_data["group_user_id"]=check_group_id
    
    response = {"input":input_data, "orginal_data":orginal_data, "updated_data":updated_data}
    return {
        'statusCode': 200,
        "headers": {"Access-Control-Allow-Origin": "*"},
        'body': json.dumps(response)
    }




# def lambda_handler(event, context):
#     print(event)

#     # FOR TESTING ONLY
#     dummy_response = "TEST SUCCESS"
#     return {
#         'statusCode': 200,
#         'body': json.dumps(dummy_response)
#     }