clue-api 1.0.1.dev99__tar.gz → 1.0.1.dev101__tar.gz

{clue_api-1.0.1.dev99 → clue_api-1.0.1.dev101}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clue-api
-Version: 1.0.1.dev99
+Version: 1.0.1.dev101
 Summary: Clue distributed enrichment service
 License: MIT
 License-File: LICENSE

{clue_api-1.0.1.dev99 → clue_api-1.0.1.dev101}/clue/plugin/helpers/trino.py

@@ -5,6 +5,7 @@ from flask import request
 from trino.auth import JWTAuthentication
 from trino.dbapi import Connection, Cursor, connect
 
+from clue.common.exceptions import ClueValueError
 from clue.common.logging import get_logger
 from clue.plugin.helpers.token import get_username
 
@@ -20,7 +21,13 @@ def get_trino_connection(
     access_token: str | None = None,
     host: str | None = None,
 ) -> Connection:
-    "Get a trino connection based on the provided JWT"
+    """Get a trino connection based on the provided JWT"""
+    if not host and "TRINO_HOST" not in os.environ:
+        raise ClueValueError("You must specify a host, or there must be a TRINO_HOST environment variable.")
+
+    if not source and "APP_NAME" not in os.environ:
+        raise ClueValueError("You must specify a source, or there must be an APP_NAME environment variable.")
+
     jwt_token: str = access_token or cast(str, request.headers.get("Authorization", None, type=str)).split(" ")[1]
     if jwt_token not in connections:
         connections[jwt_token] = connect(

{clue_api-1.0.1.dev99 → clue_api-1.0.1.dev101}/pyproject.toml

@@ -146,7 +146,7 @@ log_cli_level = "WARN"
 [tool.poetry]
 package-mode = true
 name = "clue-api"
-version = "1.0.1.dev99"
+version = "1.0.1.dev101"
 description = "Clue distributed enrichment service"
 authors = ["Canadian Centre for Cyber Security <contact@cyber.gc.ca>"]
 license = "MIT"