Merge pull request #731 from lupko/flight-fixes

RELATED: CQ-555 - Tweaks & fixes in the new package

Author: no23reason

gooddata-flight-server/gooddata_flight_server/__init__.py

@@ -1 +1,32 @@
 # (C) 2024 GoodData Corporation
+
+from gooddata_flight_server.config.config import (
+    AuthenticationMethod,
+    OtelConfig,
+    OtelExporterType,
+    ServerConfig,
+)
+from gooddata_flight_server.errors.error_code import ErrorCode
+from gooddata_flight_server.errors.error_info import ErrorInfo, RetryInfo
+from gooddata_flight_server.flexfun.flex_fun import FlexFun
+from gooddata_flight_server.health.server_health_monitor import ModuleHealthStatus, ServerHealthMonitor
+from gooddata_flight_server.server.auth.auth_middleware import TokenAuthMiddleware
+from gooddata_flight_server.server.auth.token_verifier import TokenVerificationStrategy
+from gooddata_flight_server.server.base import FlightServerMethodsFactory, ServerContext
+from gooddata_flight_server.server.flight_rpc.flight_middleware import (
+    CallFinalizer,
+    CallInfo,
+)
+from gooddata_flight_server.server.flight_rpc.server_methods import FlightServerMethods
+from gooddata_flight_server.tasks.base import TaskWaitTimeoutError
+from gooddata_flight_server.tasks.task import Task
+from gooddata_flight_server.tasks.task_error import TaskError
+from gooddata_flight_server.tasks.task_executor import (
+    TaskExecutor,
+)
+from gooddata_flight_server.tasks.task_result import (
+    FlightDataTaskResult,
+    ListFlightsTaskResult,
+    TaskExecutionResult,
+    TaskResult,
+)

gooddata-flight-server/gooddata_flight_server/py.typed

@@ -0,0 +1 @@
+Mark package as supporting typing. See https://www.python.org/dev/peps/pep-0561/ for details.

gooddata-flight-server/gooddata_flight_server/tasks/task.py

@@ -3,7 +3,7 @@
 import threading
 import uuid
 from concurrent.futures import CancelledError
-from typing import Dict, List, Literal, Optional, Union, final
+from typing import Optional, Union, final
 
 from gooddata_flight_server.tasks.task_error import TaskError
 from gooddata_flight_server.tasks.task_result import TaskResult
@@ -191,25 +191,3 @@ def run(self) -> Union[TaskResult, TaskError]:
         :raise CancelledError: when the task's run was cancelled
         """
         raise NotImplementedError
-
-
-class TaskFactory(abc.ABC):
-    """
-    Interface for factories that create concrete implementations of tasks.
-    """
-
-    def create_task(
-        self,
-        cmd: bytes,
-        headers: Dict[str, List[str]],
-        method: Union[Literal["get-flight-info", "list-flights"], str] = "get-flight-info",
-    ) -> Task:
-        """
-        Creates a new task to execute a particular command.
-
-        :param cmd: command included in the FlightDescriptor
-        :param headers: headers provided with the request
-        :param method: flight RPC method / action through which the request to create this task was invoked
-        :return: new instance of task
-        """
-        raise NotImplementedError

gooddata-flight-server/setup.py

@@ -35,8 +35,11 @@
         "Documentation": "https://gooddata-flight-server.readthedocs.io/en/v1.21.0",
         "Source": "https://github.com/gooddata/gooddata-python-sdk",
     },
+    scripts=[
+        "bin/gooddata-flight-server",
+    ],
     classifiers=[
-        "Development Status :: 5 - Production/Stable",
+        "Development Status :: 4 - Beta",
         "Environment :: Console",
         "License :: OSI Approved :: MIT License",
         "Programming Language :: Python :: 3.8",

pyproject.toml

@@ -34,6 +34,7 @@ lint.ignore = [
 "gooddata-sdk/gooddata_sdk/__init__.py" = ["F401"]
 "gooddata-dbt/gooddata_dbt/__init__.py" = ["F401"]
 "gooddata-fdw/gooddata_fdw/__init__.py" = ["F401"]
+"gooddata-flight-server/gooddata_flight_server/__init__.py" = ["F401"]
 
 [tool.ruff.format]
 exclude = ['(gooddata-api-client|.*\.snapshot\..*)']