add tld i18n

Zepolimer · Zepolimer · commit 8fd0c1f3a164 · 2024-05-22T12:34:11.000+02:00
diff --git a/python_advanced_search/models/query/__init__.py b/python_advanced_search/models/query/__init__.py
@@ -27,6 +27,7 @@
     RelatedCommand,
     DefineCommand,
 )
+from python_advanced_search.models.location import Location
 from python_advanced_search.services.crawler import GoogleRequest, BingRequest
 
 
@@ -125,11 +126,11 @@ def exclude(self, **_operators):
         self._add_commands(exclude=True, **_operators)
         return self
 
-    def to(self, cls):
-        return cls(self)
+    def to(self, cls, location=Location.WORLDWIDE):
+        return cls(self, tld=location)
 
-    def to_google(self):
-        return self.to(GoogleRequest)
+    def to_google(self, location=Location.WORLDWIDE):
+        return self.to(GoogleRequest, location)
 
-    def to_bing(self):
-        return self.to(BingRequest)
+    def to_bing(self, location=Location.WORLDWIDE):
+        return self.to(BingRequest, location)
diff --git a/tests/services/crawler/__init__.py b/tests/services/crawler/__init__.py
@@ -1,6 +1,7 @@
 import unittest
 from unittest.mock import Mock, patch
 
+from python_advanced_search.models.location import Location
 from python_advanced_search.services.crawler import GoogleRequest
 from python_advanced_search.models.query import Query
 
@@ -21,10 +22,14 @@ def test_request(self):
         with patch('python_advanced_search.services.crawler.sync_playwright', return_value=mock_playwright):
             request = Query().include(
                 expression='python unittest'
-            ).to(GoogleRequest)
+            ).to(GoogleRequest, Location.FRANCE)
 
-            response = request.get()
+            self.assertEqual(
+                request.url,
+                'https://google.fr/search?q=python+unittest'
+            )
 
+            response = request.get()
             self.assertEqual(
                 response.html,
                 '<html></html>'
