Added option 4 and did some refactoring in lantern.py

This commit is contained in:
SovereigntyIsNotFreedom 2025-04-13 14:03:51 +01:00
parent bc18f76568
commit 4837034ec3

View file

@ -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: