Add files via upload

2_bib_fill_pages.py

@@ -0,0 +1,119 @@
+import json
+from login.login import login
+from login.close_login_popup import close_login_popup
+
+from get_tree.get_layout_url import get_layout_url
+
+from getpass import getpass
+
+from selenium import webdriver
+from selenium.webdriver.firefox.service import Service as FirefoxService
+from webdriver_manager.firefox import GeckoDriverManager
+
+from content.get_content_list import get_content_list
+from content.change_url import change_url
+from content.press_close_button import press_close_button
+from content.get_input_fields import get_input_fields
+from content.press_save_button import press_save_button
+from content.press_sourcecode_button import press_sourcecode_button
+from content.set_textarea import set_textarea
+
+from bib.create_bib_html import create_bib_html
+
+import time
+
+
+def main(page_id: int, bib_filename: str, base_url: str, zfn_password: str):
+    with open("username.json", "r") as file:
+        json_dict = json.load(file)
+    zfn_user: str = json_dict["zfn_user"]
+
+    with open("types_db.json", "r") as file:
+        type_dict = json.load(file)
+
+    with open("user_pages.json", "r") as file:
+        user_dict = json.load(file)
+
+    user_string: str = user_dict[str(page_id)]
+
+    driver = webdriver.Firefox(service=FirefoxService(GeckoDriverManager().install()))
+
+    # Login
+    if login(driver, base_url, zfn_user, zfn_password) is False:
+        print("Login failed")
+        exit(1)
+
+    # Get rid of the popup
+    close_login_popup(driver)
+
+    # Get the protected URL to the pages
+    page_url_base = get_layout_url(driver, base_url)
+
+    # Jump to page
+    print(f"Go to page {int(page_id)}")
+    page_url: str = page_url_base + f"&id={int(page_id)}"
+    driver.get(page_url)
+
+    # Get content from page
+    content_id, content_type, urls = get_content_list(driver)
+
+    interesting_content_ids: list = []
+    for i in range(0, len(content_type)):
+        if content_type[i] == "textmedia":
+            interesting_content_ids.append(content_id[i])
+
+    to_modify: str = str("header")
+
+    for id in interesting_content_ids:
+        idx = content_id.index(id)
+        assert idx is not None
+        change_url(driver, base_url + urls[idx])
+
+        extract_list, results = get_input_fields(driver)
+        idx = extract_list.index(to_modify)
+        value = results[idx]
+        if len(value) > 0:
+            print(value)
+
+            key_value: str | None = None
+            for t_id in type_dict.keys():
+                assert len(type_dict[t_id]) == 3
+                if (value == type_dict[t_id][1]) or (value == type_dict[t_id][2]):
+                    key_value = t_id
+
+            if key_value is not None:
+                type_string: str = key_value
+                print(key_value)
+
+                # Press source button
+                press_sourcecode_button(driver)
+
+                new_string: str = create_bib_html(
+                    user_string, type_string, bib_filename
+                )
+                set_textarea(driver, int(id), new_string)
+
+                # Press save button
+                time.sleep(1)
+                press_save_button(driver)
+
+        # Press close button
+        time.sleep(1)
+        press_close_button(driver)
+        print("Wait for 2 sec")
+        time.sleep(2)
+
+    print("Close shop")
+    driver.close()
+    driver.quit()
+
+
+bib_filename: str = "neuro.bib"
+base_url: str = "https://www.uni-bremen.de"
+zfn_password: str = getpass()
+
+with open("user_pages.json", "r") as file:
+    user_dict = json.load(file)
+
+for page_id in user_dict.keys():
+    main(int(page_id), bib_filename, base_url, zfn_password)