Distance Module

# DISTANCE  MODULE


from dataset import load_users_transactions

trans = load_users_transactions()


def calculate_user_distance():
    print("This function calculates the distance between any user's two transactions")

    user_id = input("Input user ID: ")

    if user_id in trans.keys():
        user_transactions = trans[user_id]["transactions"]
        transaction_1 = input("Input the first Transaction Id: ")
        transaction_2 = input("Input the second Transaction Id: ")
        if transaction_1 and transaction_2 in list(user_transactions.keys()):
            x1 = user_transactions[transaction_1]["x"]
            y1 = user_transactions[transaction_1]["y"]
            x2 = user_transactions[transaction_2]["x"]
            y2 = user_transactions[transaction_2]["y"]
            distance = ((x2 - x1) ** 2 + (y2 - y1) ** 2) ** 0.5
            print(f"Distance between the transactions is {distance}")
    else:
        print("User ID doesn't exist!")


def calculate_any_two_users_distance():
    print("This function calculate distance between any two user's transactions")

    user_id_1 = input("Input first user ID: ")
    user_id_2 = input("Input second user ID: ")

    if user_id_1 in trans.keys() and user_id_2 in trans.keys():
        user_1_transactions = trans[user_id_1]["transactions"]
        user_2_transactions = trans[user_id_2]["transactions"]

        transaction_id_1 = input("Input the Transaction Id of user 1: ")
        transaction_id_2 = input("Input the Transaction Id of user 2: ")
        if transaction_id_1 in list(
            user_1_transactions.keys()
        ) and transaction_id_2 in list(user_2_transactions.keys()):
            x1 = user_1_transactions[transaction_id_1]["x"]
            y1 = user_1_transactions[transaction_id_1]["y"]
            x2 = user_2_transactions[transaction_id_2]["x"]
            y2 = user_2_transactions[transaction_id_2]["y"]
            distance = ((x2 - x1) ** 2 + (y2 - y1) ** 2) ** 0.5
            print(f"Distance between the transactions is {distance}")

        else:
            print("One the Transactio IDs doesn't exist")

    else:
        print("One the Users doesn't exist")