Added option 2 in lantern.py

This commit is contained in:
SovereigntyIsNotFreedom 2025-04-13 13:48:03 +01:00
parent 9158e265fe
commit 3cc79f7392

View file

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