add new class that respects fully qualified URLs but validates Hosts are Anvil hosts

manggit · manggit · commit ac3bc467dca5 · 2025-04-28T21:45:16.000-07:00
diff --git a/python_anvil/api_resources/requests.py b/python_anvil/api_resources/requests.py
@@ -161,3 +161,32 @@ class PlainRequest(BaseAnvilHttpRequest):
 
     def get_url(self):
         return f"{self.API_HOST}/{self.API_BASE}"
+
+
+class FullyQualifiedRequest(BaseAnvilHttpRequest):
+    """A request class that validates URLs are fully qualified and point to Anvil domains."""
+
+    VALID_HOSTS = [
+        "https://app.useanvil.com",
+        # Future Anvil specific URLs
+    ]
+
+    def __init__(self, client, options=None):
+        super().__init__(client, options)
+
+    def get_url(self):
+        return ""  # Not used since we expect full URLs
+
+    def _validate_url(self, url):
+        if not any(url.startswith(host) for host in self.VALID_HOSTS):
+            raise ValueError(
+                f"URL must start with one of: {', '.join(self.VALID_HOSTS)}"
+            )
+
+    def get(self, url, params=None, **kwargs):
+        self._validate_url(url)
+        return super().get(url, params, **kwargs)
+
+    def post(self, url, data=None, **kwargs):
+        self._validate_url(url)
+        return super().post(url, data, **kwargs)
