Add verrification_url parameter to verifier

examples/basic/verify-token.py

@@ -9,7 +9,8 @@ def your_function_or_handler():
     token = str(uuid4())  # This is an example UUIDv4
 
     # This handles the connections with a session and can be reused
-    # for better performance
+    # for better performance. The `verification_url` keyword arguement
+    # can be set to your url or leave blank to use a default one
     verifier = TokenVerifier()
 
     try:

examples/flask/app.py

@@ -12,7 +12,9 @@ def get_token_verifier() -> TokenVerifier:
     It's a good practice in order to reuse connections.
     """
     if "VQueueTokenVerifier" not in g:
-        g.VQueueTokenVerifier = TokenVerifier()
+        # This is an example, the `verification_url` keyword arguement can be set to
+        # your verification url or leave blank to use a default url
+        g.VQueueTokenVerifier = TokenVerifier(verification_url="http://localhost:8000")
 
     return g.VQueueTokenVerifier
 

src/vqueue/queues.py

@@ -14,9 +14,14 @@
 class TokenVerifier:
     """Client to verify Virtual Queue tokens"""
 
-    def __init__(self):
-        """Initialize the TokenVerifier with a network session"""
+    def __init__(self, *, verification_url: str | None = None):
+        """Initialize the TokenVerifier with a network session
+
+        Args:
+            verification_url: URL of the verification service. If none is given, a default value will be used
+        """
         self.session = requests.Session()
+        self._verification_token = verification_url or VERIFY_API_URL
 
     def close(self):
         self.session.close()
@@ -49,7 +54,7 @@ def verify_token(self, token: str) -> VerificationResult:
         uuid_token = validate_uuidv4(token)
 
         try:
-            response = self.session.get(f"{VERIFY_API_URL}?token={uuid_token}")
+            response = self.session.get(f"{self._verification_token}?token={uuid_token}")
             response_data = response.json()
         except requests.JSONDecodeError as e:
             raise VQueueError("Invalid JSON response") from e