import datetime
import json
import random
from random import randint

from util import updateTripInfo, getTripInfo, DecimalEncoder

def lambda_handler(event, context):
    
    rate_codes = [1, 2, 3, 4, 5, 6, 99]
    
    #rider_id = "person69257@example.com"
    #trip_info = input("Enter your tripinfo : ") 
    #trip_info = event["key1"]
    
    i_rider_id= int(event['queryStringParameters']['rider_id'])
    trip_info = event['queryStringParameters']['trip_info']
    
    rider_name = "person" + str(i_rider_id)
    rider_id = rider_name + "@example.com"
    
    dropoff_longitude = round(random.uniform(-74,-73),6)
    dropoff_latitude = round(random.uniform(40,41),6) 
    
    tripCompletedInfo = {
        "riderid"  : rider_id,
        "tripinfo" : trip_info,
        "DROPOFF_LATITUDE" : str(dropoff_latitude),
        "RATE_CODE_ID" : rate_codes[randint(0, 6)],
        "TOLLS_AMOUNT" : str(round(random.uniform(0,5),2)),
        "IMPROVEMENT_SURCHARGE" : str(round(random.uniform(0,1),1)),
        "TIP_AMOUNT" : str(round(random.uniform(0,10),2)),
        "DROPOFF_DATETIME" : datetime.datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ"),
        "TRIP_DISTANCE" : randint(1, 50),
        "TOTAL_AMOUNT" : str(round(random.uniform(5,150),2)),
        "MTA_TAX" : str(round(random.uniform(0,1),1)),
        "DROPOFF_LONGITUDE" : str(dropoff_longitude),
        "PAYMENT_TYPET" : randint(1, 7),
        "EXTRA" : str(round(random.uniform(0,1),1)),
        "FARE_AMOUNT" : str(round(random.uniform(5,150),2)), 
        "PASSENGER_COUNT": randint(1, 7),
        "Status" : "Completed"
    }

    print ("Trip completion information = ", json.dumps(tripCompletedInfo, indent=2))

    response = updateTripInfo(tripCompletedInfo, "InProgress")
    print("Trip completion information has been updated to Trips table")
    
    print("Driver trip completion information =" + json.dumps(response['Attributes'], indent = 4, cls=DecimalEncoder))
    
    responseObjects = {}
    responseObjects['statusCode'] = 200
    responseObjects['headers'] = {}
    responseObjects['headers']['Content-Type'] = 'application/json'
    responseObjects['body'] = json.dumps(response['Attributes'], indent = 4, cls=DecimalEncoder)
    
    return responseObjects