From 4837034ec3503152235757f8fb03d4b66d5e3695 Mon Sep 17 00:00:00 2001 From: SovereigntyIsNotFreedom Date: Sun, 13 Apr 2025 14:03:51 +0100 Subject: [PATCH] Added option 4 and did some refactoring in lantern.py --- scripts/lantern.py | 63 ++++++++++++++++++++-------------------------- 1 file changed, 27 insertions(+), 36 deletions(-) diff --git a/scripts/lantern.py b/scripts/lantern.py index 6754f87..0dc93d2 100644 --- a/scripts/lantern.py +++ b/scripts/lantern.py @@ -1202,9 +1202,8 @@ Maintenance: category = "EXTERNAL USER" sensitive = "YES" try: - print_colors("[+] LISTING ALL LINK INSIDE SUBMISSION") - print_colors("1) MOVE TO VERIFIED 2) MOVE TO UNVERIFIED 3) DELETE FROM LIST 0) EXIT") - option = int(input("ENTER AN OPTION: ")) + print_colors("1) Move entries to verified 2) Move entries from submission to unverified 3) Delete from submission file 0) exit") + option = int(input("Enter an option: ")) # can't use instance variable because it will go inside www submission_csv_file = "../submissions/submission.csv" submission_df = pd.read_csv(submission_csv_file) @@ -1213,54 +1212,46 @@ Maintenance: match option: case 1: 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'] + + 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([verified_csv_df, new_row], ignore_index=True) - df.to_csv(verifiedcsvfile, index=False) - except Exception as e: - print(e) - print_colors("Try again", is_error=True) - break + new_row = pd.DataFrame({'Instance': [instance], 'Category': [category], 'Name':[name],'URL':[link],'Sensitive':[sensitive],'Description':[desc],'Status':['YES'],'Score':['100']}) + + df = pd.concat([verified_csv_df, new_row], ignore_index=True) + df.to_csv(verifiedcsvfile, index=False) + case 2: 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'] + + 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']}) + 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) + 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 case 3: - print(submission_df) + print_colors(submission_df) choose = int(input("Enter which row should be removed: ")) - submission_df.drop(index=choose, inplace=True) submission_df.to_csv(submission_csv_file) - print(submission_df) + print_colors(submission_df) - - except Exception as e: - print(e) + case 0: + break + + except: + print_colors('Try again',is_error=True) break break case 0: