allow to update query string using mod operator

aio-libs · Apr 4, 2020 · f57d0eb · f57d0eb
1 parent 78f49dc
commit f57d0eb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/tests/test_update_query.py b/tests/test_update_query.py
@@ -219,3 +219,12 @@ def test_update_query_multiple_keys():
     u2 = url.update_query([("a", "3"), ("a", "4")])
 
     assert str(u2) == "http://example.com/path?a=3&a=4"
+
+
+# mod
+
+def test_update_query_with_mod_operator():
+    url = URL("http://example.com/")
+    assert str(url % {"a": "1"}) == "http://example.com/?a=1"
+    assert str(url % {"a": "1"} % {"b": "2"}) == "http://example.com/?a=1&b=2"
+    assert str(url % {"a": "1"} % {"a": "3", "b": "2"}) == "http://example.com/?a=3&b=2"
diff --git a/yarl/__init__.py b/yarl/__init__.py
@@ -307,6 +307,9 @@ def __truediv__(self, name):
         return URL(
             self._val._replace(path=new_path, query="", fragment=""), encoded=True
         )
+
+    def __mod__(self, query):
+        return self.update_query(query)
 
     def __bool__(self) -> bool:
         return bool(