feat: fix cafenero

freitagsrunde · Dec 7, 2024 · 2fcdf42 · 2fcdf42
1 parent 0db079a
commit 2fcdf42
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/canteens/cafenero.py b/canteens/cafenero.py
@@ -46,8 +46,8 @@ def extract_nextcloud_link(html):
         The link as string.
     """
     soup = bs4.BeautifulSoup(html, 'html.parser')
-    link = [a for a in soup.find_all('a', href=True) if 'Speisekarte als' in a.text and 'nextcloud' in a.get('href') and 'download' in a.get('href')][0].get('href')
-    return link
+    link = [a for a in soup.find_all('a', href=True) if 'speisekarte als' in a.text.lower()][0].get('href')
+    return 'https://cafenero.net/' + link
 
 
 def get_pdf(link):

diff --git a/tests/test_cafenero.py b/tests/test_cafenero.py
@@ -32,9 +32,7 @@ def test_download_website__with_connect_timeout():
 def test_extract_nextcloud_link():
     html = cafenero.download_website()
     link = cafenero.extract_nextcloud_link(html)
-    assert 'nextcloud' in link
-    assert 'download' in link
-
+    assert link
 
 @flaky
 def test_get_pdf(pdf_file):