
from _add_use_case_scripts.requestPost  import postLayersToSwagger, postTableToSwagger


def add_table(use_case: str, table_name: str):
    '''
    take the columns and add the mappings at the server
    replace all "/"'s in the internal representation with a "_"
    '''

    columns = [
        "id", # pk
        "balance",
        "numberOfTripsDone",
        "role",
        "sumOfTripsScores"
    ]
    
    columns = { c : c for c in columns }
    columns["UniqueID"] = "id"
    
    table = {
        "name": table_name,
        "mappings": columns
    }

    postTableToSwagger(use_case,table)


def add_layers(use_case:str, table_name: str):
    layers = [
        {
            "use_case": use_case,
            "table": table_name,
            "name": "User_layer",
            "properties": [
                "UniqueID",
                "balance",
                "role",
                "numberOfTripsDone",
                "sumOfTripsScores"
                
            ],
            "cluster_properties": [
                "balance",
                "numberOfTripsDone",
                "sumOfTripsScores"
            ]
        },
        {
            "use_case": use_case,
            "table": table_name,
            "name": "UserBalance_layer",
            "properties": [
                "UniqueID",
                "balance",
                "role",
                "numberOfTripsDone",
                "sumOfTripsScores"
                
            ],
            "cluster_properties": [
                "balance",
                "sumOfTripsScores"
            ]
        },
        {
            "use_case": use_case,
            "table": table_name,
            "name": "UserVeterancy_layer",
            "properties": [
                "UniqueID",
                "balance",
                "role",
                "numberOfTripsDone",
                "sumOfTripsScores"
                
            ],
            "cluster_properties": [
                "numberOfTripsDone",
                "sumOfTripsScores"
            ]
        }
    ]

    postLayersToSwagger(use_case,layers)

def main(use_case: str):
    print("USER")
    table_name = "user"
    add_table(use_case, table_name)
    add_layers(use_case, table_name)