diff --git a/examples/basic/verify-token.py b/examples/basic/verify-token.py index 3ba77fb..cdf9034 100644 --- a/examples/basic/verify-token.py +++ b/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: diff --git a/examples/flask/app.py b/examples/flask/app.py index 627dc09..c7cc4bf 100644 --- a/examples/flask/app.py +++ b/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 diff --git a/src/vqueue/py.typed b/src/vqueue/py.typed new file mode 100644 index 0000000..e69de29 diff --git a/src/vqueue/queues.py b/src/vqueue/queues.py index 459ddcc..0bb6369 100644 --- a/src/vqueue/queues.py +++ b/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