import boto3
import json
import time
import uuid
from datetime import datetime
import os
from os import environ

# create a function to add numbers
starttime = time.time()
if 'SQS_QUEUE_URL' in os.environ:
    queue_url = os.environ['SQS_QUEUE_URL']
    region = os.environ['AWS_REGION']
    print (f'SQS URL : {queue_url} - region : {region} ')
else:
    print ('SQS URL Missing!!!!!')

if 'DYNAMODB_TABLE' not in os.environ:
    print ('DYNAMODB_TABLE details Missing!!!!!')
else:
    dynamodb_table = os.environ['DYNAMODB_TABLE']
    print (f'DYNAMODB_TABLE : {dynamodb_table}')


_id = "a3b01bb8-da7c-11ed-aae2-5227b566f9a7"
def receive_message():
    print(f'queue_url {queue_url}- region : {region} - Dynamo : {dynamodb_table} ')
    try:
        print("Start fn receive message")
        sqs_client = boto3.client("sqs", region_name=region)
        response = sqs_client.receive_message(
            QueueUrl= queue_url,
            AttributeNames=[
            'SentTimestamp'
            ],
            MaxNumberOfMessages=1,
            MessageAttributeNames=[
            'All'
            ],
            WaitTimeSeconds=0,
            VisibilityTimeout=60
        )
        print(f"Number of messages received: {len(response.get('Messages', []))}")
        
        #for message in response.get("Messages", []):
        if len(response.get('Messages', [])) != 0:
            message = response['Messages'][0]
            message_body = message["Body"]
            print(f"message_body : {message_body}")

            receipt_handle = message['ReceiptHandle']
            
            save_data(message_body)

            print(f"Receipt Handle: {message['ReceiptHandle']}")
            print(f"Deleting Message : {message_body}")
            # Delete received message from queue
            sqs_client.delete_message(
                QueueUrl=queue_url,
                ReceiptHandle=receipt_handle
            )

            
        print("End fn receive message")
    except Exception as ex:
        print(f"Error happened in receive_message : {ex} ")
    

def save_data(_message):
    try:
        print(f'save data src msg :{_message}')
        jsonMessage = json.loads(_message)
        print(f'Src Message :{jsonMessage["msg"]},{jsonMessage["srcStamp"]}')
        #current_dateTime = json.dumps(datetime.now(),default= str)
        date_format = '%Y-%m-%d %H:%M:%S.%f'
        current_dateTime = datetime.utcnow().strftime(date_format)
        
        print(f"id:{_id}")
        dynamodb = boto3.resource('dynamodb', region_name=region)
        table = dynamodb.Table(dynamodb_table)
        
        
        messageProcessingTime = datetime.utcnow() - datetime.strptime(jsonMessage["srcStamp"],date_format) 
        print(f'messageProcessingTime: {messageProcessingTime.total_seconds()}')

        response = table.put_item(
            Item={
            'id': _id,
            'data': jsonMessage["msg"],
            'srcStamp':jsonMessage["srcStamp"],
            'destStamp':current_dateTime,
            'messageProcessingTime':str(messageProcessingTime.total_seconds())
            }
        )
        status_code = response['ResponseMetadata']['HTTPStatusCode']
        print(f"Data Save Status : {status_code}")
    except Exception as error:
        print(f"Error has happened : {error}")

    
      

while True:
    t = time.localtime()
    time.sleep(1.0 - ((time.time() - starttime) % 1.0)) #sleep for 1 sec
    currenttime = time.strftime("%H:%M:%S", t)
    print(f"Start SQS Call : {currenttime}")

    receive_message()
    #save_data("hi there")

    ## Date format working
    #date_format = '%Y-%m-%d %H:%M:%S.%f'
    #currentDateAndTime = datetime.now().strftime(date_format)#"2023-03-23 17:49:25.651555"
    #currentDateAndTime = '2023-03-23 18:38:42.536417'
    #print("The current date and time is", currentDateAndTime)
    #currentDate = datetime.strptime(currentDateAndTime, date_format)
    #x = datetime.now() - currentDate
    #print("diff",x)
    #print("The current time is", currentDate)
    '''i = 0
    while i < 20:
        i = i+1'''
    currenttime = time.strftime("%H:%M:%S", t)
    print(f"End SQS Call {currenttime}")