Skip to content

Commit 95cd9d5

Browse files
author
Andrei Neagu
committed
made modules flat
1 parent 187d78f commit 95cd9d5

File tree

8 files changed

+22
-27
lines changed

8 files changed

+22
-27
lines changed

packages/service-library/src/servicelib/long_running_tasks/_rabbit/lrt_client.py renamed to packages/service-library/src/servicelib/long_running_tasks/_lrt_client.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,14 @@
55
from models_library.rabbitmq_basic_types import RPCMethodName
66
from pydantic import PositiveInt, TypeAdapter
77

8-
from ...logging_errors import create_troubleshootting_log_kwargs
9-
from ...logging_utils import log_decorator
10-
from ...long_running_tasks.task import RegisteredTaskName
11-
from ...rabbitmq._client_rpc import RabbitMQRPCClient
12-
from .._serialization import string_to_object
13-
from ..models import LRTNamespace, TaskBase, TaskContext, TaskId, TaskStatus
8+
from ..logging_errors import create_troubleshootting_log_kwargs
9+
from ..logging_utils import log_decorator
10+
from ..rabbitmq._client_rpc import RabbitMQRPCClient
1411
from ._models import RPCErrorResponse
15-
from .namespace import get_rabbit_namespace
12+
from ._rabbit_namespace import get_rabbit_namespace
13+
from ._serialization import string_to_object
14+
from .models import LRTNamespace, TaskBase, TaskContext, TaskId, TaskStatus
15+
from .task import RegisteredTaskName
1616

1717
_logger = logging.getLogger(__name__)
1818

packages/service-library/src/servicelib/long_running_tasks/_rabbit/lrt_server.py renamed to packages/service-library/src/servicelib/long_running_tasks/_lrt_server.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
import traceback
22
from typing import Any
33

4-
from ...rabbitmq import RPCRouter
5-
from .._serialization import object_to_string
6-
from ..base_long_running_manager import BaseLongRunningManager
7-
from ..errors import BaseLongRunningError, TaskNotCompletedError, TaskNotFoundError
8-
from ..models import TaskBase, TaskContext, TaskId, TaskStatus
9-
from ..task import RegisteredTaskName
4+
from ..rabbitmq import RPCRouter
105
from ._models import RPCErrorResponse
6+
from ._serialization import object_to_string
7+
from .base_long_running_manager import BaseLongRunningManager
8+
from .errors import BaseLongRunningError, TaskNotCompletedError, TaskNotFoundError
9+
from .models import TaskBase, TaskContext, TaskId, TaskStatus
10+
from .task import RegisteredTaskName
1111

1212
router = RPCRouter()
1313

packages/service-library/src/servicelib/long_running_tasks/_rabbit/_models.py renamed to packages/service-library/src/servicelib/long_running_tasks/_models.py

File renamed without changes.

packages/service-library/src/servicelib/long_running_tasks/_rabbit/__init__.py

Whitespace-only changes.

packages/service-library/src/servicelib/long_running_tasks/_rabbit/namespace.py renamed to packages/service-library/src/servicelib/long_running_tasks/_rabbit_namespace.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from models_library.rabbitmq_basic_types import RPCNamespace
22
from pydantic import TypeAdapter
33

4-
from ..models import LRTNamespace
4+
from .models import LRTNamespace
55

66

77
def get_rabbit_namespace(namespace: LRTNamespace) -> RPCNamespace:

packages/service-library/src/servicelib/long_running_tasks/_serialization.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,8 @@
11
import base64
2-
import logging
32
import pickle
43
from abc import ABC, abstractmethod
54
from typing import Any, Final, Generic, TypeVar
65

7-
_logger = logging.getLogger(__name__)
8-
9-
106
T = TypeVar("T")
117

128

packages/service-library/src/servicelib/long_running_tasks/base_long_running_manager.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from settings_library.redis import RedisSettings
55

66
from ..rabbitmq._client_rpc import RabbitMQRPCClient
7-
from ._rabbit.namespace import get_rabbit_namespace
7+
from ._rabbit_namespace import get_rabbit_namespace
88
from .models import LRTNamespace
99
from .task import TasksManager
1010

@@ -61,8 +61,7 @@ async def setup(self) -> None:
6161
client_name=f"lrt-client-{self.lrt_namespace}",
6262
settings=self.rabbit_settings,
6363
)
64-
65-
from ._rabbit.lrt_server import router
64+
from ._lrt_server import router
6665

6766
await self.rpc_server.register_router(
6867
router,

packages/service-library/src/servicelib/long_running_tasks/lrt_api.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from typing import Any
22

33
from ..rabbitmq._client_rpc import RabbitMQRPCClient
4-
from ._rabbit import lrt_client
4+
from . import _lrt_client
55
from .base_long_running_manager import BaseLongRunningManager
66
from .models import TaskBase, TaskContext, TaskId, TaskStatus
77
from .task import RegisteredTaskName
@@ -43,7 +43,7 @@ async def start_task(
4343
TaskId: the task unique identifier
4444
"""
4545

46-
return await lrt_client.start_task(
46+
return await _lrt_client.start_task(
4747
long_running_manager.rpc_client,
4848
long_running_manager.lrt_namespace,
4949
registered_task_name=registered_task_name,
@@ -60,7 +60,7 @@ async def list_tasks(
6060
long_running_manager: BaseLongRunningManager,
6161
task_context: TaskContext,
6262
) -> list[TaskBase]:
63-
return await lrt_client.list_tasks(
63+
return await _lrt_client.list_tasks(
6464
rabbitmq_rpc_client,
6565
long_running_manager.lrt_namespace,
6666
task_context=task_context,
@@ -74,7 +74,7 @@ async def get_task_status(
7474
task_id: TaskId,
7575
) -> TaskStatus:
7676
"""returns the status of a task"""
77-
return await lrt_client.get_task_status(
77+
return await _lrt_client.get_task_status(
7878
rabbitmq_rpc_client,
7979
long_running_manager.lrt_namespace,
8080
task_id=task_id,
@@ -88,7 +88,7 @@ async def get_task_result(
8888
task_context: TaskContext,
8989
task_id: TaskId,
9090
) -> Any:
91-
return await lrt_client.get_task_result(
91+
return await _lrt_client.get_task_result(
9292
rabbitmq_rpc_client,
9393
long_running_manager.lrt_namespace,
9494
task_context=task_context,
@@ -103,7 +103,7 @@ async def remove_task(
103103
task_id: TaskId,
104104
) -> None:
105105
"""cancels and removes the task"""
106-
await lrt_client.remove_task(
106+
await _lrt_client.remove_task(
107107
rabbitmq_rpc_client,
108108
long_running_manager.lrt_namespace,
109109
task_id=task_id,

0 commit comments

Comments
 (0)