Fix typing information to please pyright

flathunter/crawler/wggesucht.py

@@ -170,7 +170,7 @@ def __init__(self, config):
         self.config = config
 
     # pylint: disable=too-many-locals
-    def extract_data(self, soup: BeautifulSoup):
+    def extract_data(self, soup: BeautifulSoup) -> List[Dict]:
         """Extracts all exposes from a provided Soup object"""
         entries = []
 

test/crawler/test_crawl_wggesucht.py

@@ -1,5 +1,6 @@
 import os
 import unittest
+from typing import Dict
 from functools import reduce
 from bs4 import BeautifulSoup
 from flathunter.crawler.wggesucht import WgGesucht
@@ -25,9 +26,10 @@ def test(self):
         self.assertTrue(entries[0]['url'].startswith("https://www.wg-gesucht.de/wohnungen"), u"URL should be an apartment link")
         for attr in [ 'title', 'price', 'size', 'rooms', 'address', 'image', 'from' ]:
             self.assertIsNotNone(entries[0][attr], attr + " should be set")
-        for attr in [ 'to' ]:
-            found = reduce(lambda i, e: attr in e or i, entries, False)
-            self.assertTrue(found, "Expected " + attr + " to sometimes be set")
+        def shrink(i: bool, e: Dict) -> bool:
+            return 'to' in e or i
+        found = reduce(shrink, entries, False)
+        self.assertTrue(found, "Expected 'to' to sometimes be set")
 
     def test_filter_spotahome_ads(self):
         with open(os.path.join(os.path.dirname(os.path.realpath(__file__)), "fixtures", "wg-gesucht-spotahome.html")) as fixture: