Add ability to get # links

This commit is contained in:
iamdoubz
2026-03-25 10:12:15 -05:00
parent 51a1921b12
commit 3bb6d742e6
+9 -2
View File
@@ -98,7 +98,10 @@ def main():
# Function to read platform/letter and write links to file
def get_links(pf, lets):
driver.get(f"https://vimm.net/vault/{pf}/{lets}")
if lets in string.ascii_uppercase:
driver.get(f"https://vimm.net/vault/{pf}/{lets}")
else:
driver.get(f"https://vimm.net/vault/{lets}")
data = []
# Locate the main table
table = driver.find_element(By.CSS_SELECTOR, "table.rounded")
@@ -149,8 +152,12 @@ def main():
if letter == 'ALL':
big_list = []
big_list.append({"platform": platform, "letter": f"?p=list&system={platform}&section=number"})
for l in string.ascii_uppercase:
get_links(platform, l)
big_list.append({"platform": platform, "letter": l})
for a in big_list:
get_links(a['platform'], a['letter'])
else:
get_links(platform, letter)