Upload scrape_fake_app.py

scrape_fake_app.py

@@ -1,6 +1,7 @@
 from playwright.sync_api import sync_playwright
 import os
 from dotenv import load_dotenv
+import json
 
 def load_credentials()-> dict:
     load_dotenv()
@@ -45,39 +46,46 @@ def extract_timetable_text(page):
 
 # print(URL,USERNAME,PASSWORD)
 def get_homework() -> str:
-    URL,USERNAME,PASSWORD=load_credentials()
-    
-    with sync_playwright() as playwright:
-        browser = playwright.firefox.launch(headless=True)
-        page = browser.new_page()
-        page.goto(URL,wait_until="domcontentloaded")
+    try:
+        URL,USERNAME,PASSWORD=load_credentials()
         
-        page.get_by_role('textbox',name='username').fill(USERNAME)
-        page.get_by_role('textbox',name='password').fill(PASSWORD)
-        page.get_by_role('button',name='login').click()
-        page.wait_for_url("**/dashboard")
-        # page.wait_for_timeout(1000)         
-        homework = extract_homework_text(page)
-        browser.close()        
-    return homework
+        with sync_playwright() as playwright:
+            browser = playwright.firefox.launch(headless=True)
+            page = browser.new_page()
+            page.goto(URL,wait_until="domcontentloaded")
+            
+            page.get_by_role('textbox',name='username').fill(USERNAME)
+            page.get_by_role('textbox',name='password').fill(PASSWORD)
+            page.get_by_role('button',name='login').click()
+            page.wait_for_url("**/dashboard")
+            # page.wait_for_timeout(1000)         
+            homework = extract_homework_text(page)
+            browser.close()        
+        return homework
+    
+    except Exception as e:
+        return json.dumps(e)
 
 def get_timetable() -> str:
-    URL,USERNAME,PASSWORD=load_credentials()
-    
-    with sync_playwright() as playwright:
-        browser = playwright.firefox.launch(headless=True)
-        page = browser.new_page()
-        page.goto(URL,wait_until="domcontentloaded")
+    try:
+        URL,USERNAME,PASSWORD=load_credentials()
         
-        page.get_by_role('textbox',name='username').fill(USERNAME)
-        page.get_by_role('textbox',name='password').fill(PASSWORD)
-        page.get_by_role('button',name='login').click()
-        page.wait_for_url("**/dashboard")
-        # page.wait_for_timeout(1000)        
-        timetable = extract_timetable_text(page)
-        browser.close()
-        
-    return timetable
+        with sync_playwright() as playwright:
+            browser = playwright.firefox.launch(headless=True)
+            page = browser.new_page()
+            page.goto(URL,wait_until="domcontentloaded")
+            
+            page.get_by_role('textbox',name='username').fill(USERNAME)
+            page.get_by_role('textbox',name='password').fill(PASSWORD)
+            page.get_by_role('button',name='login').click()
+            page.wait_for_url("**/dashboard")
+            # page.wait_for_timeout(1000)        
+            timetable = extract_timetable_text(page)
+            browser.close()
+        return timetable
+    
+    except Exception as e:
+        return json.dumps(e)
 
 if __name__=="__main__":
     print(get_homework())