real extension config added

Author: misterhackerman

config.py

@@ -1,6 +1,6 @@
 FOLDER = '/dox/med'
-
 HEADERS = headers = {
             "User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (HTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
         }
 DECOR = ' \033[34;1m::\033[0m '
+EXTENSIONS = ["pdf", "ppt", "m4a"]

scraper.py

@@ -144,7 +144,9 @@ def make_course_folder(self, courses, index, folder):
 
 
     def find_files_paths_and_links(self, navigation_dict, soup):
-        file_tags = soup.find_all('a', string=lambda text: text and '.pdf' in text) + soup.find_all('a', string=lambda text: text and '.ppt' in text) + soup.find_all('a', string=lambda text: text and '.m4a' in text)
+        file_tags = []
+        for extension in EXTENSIONS:
+            file_tags += soup.find_all('a', string=lambda text: text and f'.{extension}' in text)
         files_list = []
         path = []
         associated_nav_link_id = ''