import time
from workflow.gisaid.gisaid_helpers import gisaid_obj


def run_gisaid(run_id):
    print("\n================================\nGISAID Report Script\n================================\n\n")

    # import relevant data from json file
    data_obj = gisaid_obj()
    data_obj.get_json()

    data_obj.get_priority()
    data_obj.scan_db()
    data_obj.get_gisaid_df()
    data_obj.compile_fasta(run_id)
    data_obj.compile_gisaid()
    data_obj.make_fasta_file()
    data_obj.make_gisaid_file(run_id)
    data_obj.database_push()


    print("\n================================\nSUCCESS - END OF SCRIPT\n================================\n\n")
    time.sleep(2)


import time
from workflow.gisaid.gisaid_helpers import gisaid_obj


def run_gisaid(run_id):
    print("\n================================\nGISAID Report Script\n================================\n\n")

    # import relevant data from json file
    data_obj = gisaid_obj()
    data_obj.get_json()

    data_obj.get_priority()
    data_obj.scan_db()
    data_obj.get_gisaid_df()
    data_obj.compile_fasta(run_id)
    data_obj.compile_gisaid()
    data_obj.make_fasta_file()
    data_obj.make_gisaid_file(run_id)
    data_obj.database_push()


    print("\n================================\nSUCCESS - END OF SCRIPT\n================================\n\n")
    time.sleep(2)


# import relevant data from json file