diff --git a/scripts/lantern.py b/scripts/lantern.py index 39abe3b..8bd43da 100644 --- a/scripts/lantern.py +++ b/scripts/lantern.py @@ -1208,6 +1208,7 @@ Maintenance: # can't use instance variable because it will go inside www submission_csv_file = "../submissions/submission.csv" verified_csv_df = pd.read_csv(verifiedcsvfile) + unverified_csv_df = pd.read_csv(unverifiedcsvfile) match option: case 1: submission_df = pd.read_csv(submission_csv_file) @@ -1229,6 +1230,28 @@ Maintenance: print_colors("Try again", is_error=True) break + case 2: + submission_df = pd.read_csv(submission_csv_file) + print_colors(submission_df) + try: + indx = int(input("Which index do you want to add? ")) + link = submission_df.iloc[indx]['link'] + name = submission_df.iloc[indx]['name'] + desc = submission_df.iloc[indx]['desc'] + + # new_row = [instance,category,name,link,sensitive,desc,'YES','100'] + new_row = pd.DataFrame({'Instance': [instance], 'Category': [category], 'Name':[name],'URL':[link],'Sensitive':[sensitive],'Description':[desc],'Status':['YES'],'Score':['100']}) + + df = pd.concat([unverified_csv_df, new_row], ignore_index=True) + df.to_csv(unverifiedcsvfile, index=False) + + except Exception as e: + print(e) + print_colors("Try again", is_error=True) + break + + + except Exception as e: print(e)