import math

def retrieve_time(jobname,walltime,gpu):
    if walltime == "":
        return f"02:00:00"
    else:
        times=walltime.split(':')
        total_hours=int(times[0])
        if gpu == "":
            if total_hours > (24*7):
                drona_add_warning("Requested  walltime "+walltime+" (hh:mm). Reducing to max of 7 days in cpu queue")
                return f"168:00:00"
            else:
                return f""+walltime+":00"
        else:
            if total_hours > (24*2):
                drona_add_warning("Requested  walltime "+walltime+" (hh:mm). Reducing to max of 48 hours in GPU/PVC queues")
                return f"48:00:00"
            else:
                return f""+walltime+":00"


def retrieve_nodes(nodes,cores):
   if nodes == "0":
       result= ((int(cores)+47) // 48) 
       return f""+str(result)
   else:
       return f""+nodes

def retrieve_loaded_modules(modules=""):
    if modules == "":
        return f""
    elif modules == "modules":
        return f"module load foss/2023a " + modules
    else:
        return f"" + modules  


def retrieve_account(account=""):
    if account == "":
        return f""
    else:
        return f"#SBATCH --account="+account

def retrieve_memory(memory,cores):
    if memory=="":
        return f"--mem-per-cpu=5000M"
    else:
        mem=int(memory[:-1])
        mem=mem*1024
        mem_per_cpu=int(mem/int(cores))
        return f"--mem-per-cpu="+str(mem_per_cpu)+"M" 
    
def retrieve_extra(extras, gpu):
    extra_string=extras+ " " + gpu
    if extra_string == " ":
        return f""
    else:
        return f"#SBATCH " + extra_string