Skip to content

Commit 68e520f

Browse files
committed
ruff: exceptions naming
1 parent e394ed3 commit 68e520f

File tree

4 files changed

+19
-21
lines changed

4 files changed

+19
-21
lines changed

services/api-server/src/simcore_service_api_server/api/errors/log_handling_error.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,21 +3,21 @@
33
from starlette.responses import JSONResponse
44

55
from ...services.log_streaming import (
6-
LogDistributionBaseException,
7-
LogStreamerNotRegistered,
8-
LogStreamerRegistionConflict,
6+
LogDistributionBaseError,
7+
LogStreamerNotRegisteredError,
8+
LogStreamerRegistionConflictError,
99
)
1010
from .http_error import create_error_json_response
1111

1212

1313
async def log_handling_error_handler(
14-
_: Request, exc: LogDistributionBaseException
14+
_: Request, exc: LogDistributionBaseError
1515
) -> JSONResponse:
1616
msg = f"{exc}"
1717
status_code: int = 500
18-
if isinstance(exc, LogStreamerNotRegistered):
18+
if isinstance(exc, LogStreamerNotRegisteredError):
1919
status_code = status.HTTP_500_INTERNAL_SERVER_ERROR
20-
elif isinstance(exc, LogStreamerRegistionConflict):
20+
elif isinstance(exc, LogStreamerRegistionConflictError):
2121
status_code = status.HTTP_409_CONFLICT
2222

2323
return create_error_json_response(msg, status_code=status_code)

services/api-server/src/simcore_service_api_server/core/application.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,7 @@
1212
from simcore_service_api_server.api.errors.log_handling_error import (
1313
log_handling_error_handler,
1414
)
15-
from simcore_service_api_server.services.log_streaming import (
16-
LogDistributionBaseException,
17-
)
15+
from simcore_service_api_server.services.log_streaming import LogDistributionBaseError
1816
from starlette import status
1917
from starlette.exceptions import HTTPException
2018

@@ -108,7 +106,7 @@ def init_app(settings: ApplicationSettings | None = None) -> FastAPI:
108106
app.add_exception_handler(HTTPException, http_error_handler)
109107
app.add_exception_handler(RequestValidationError, http422_error_handler)
110108
app.add_exception_handler(HTTPStatusError, httpx_client_error_handler)
111-
app.add_exception_handler(LogDistributionBaseException, log_handling_error_handler)
109+
app.add_exception_handler(LogDistributionBaseError, log_handling_error_handler)
112110
app.add_exception_handler(CustomBaseError, custom_error_handler)
113111

114112
# SEE https://docs.python.org/3/library/exceptions.html#exception-hierarchy

services/api-server/src/simcore_service_api_server/services/log_streaming.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -16,15 +16,15 @@
1616
_SLEEP_SECONDS_BEFORE_CHECK_JOB_STATUS: Final[PositiveInt] = 10
1717

1818

19-
class LogDistributionBaseException(Exception):
19+
class LogDistributionBaseError(Exception):
2020
pass
2121

2222

23-
class LogStreamerNotRegistered(LogDistributionBaseException):
23+
class LogStreamerNotRegisteredError(LogDistributionBaseError):
2424
pass
2525

2626

27-
class LogStreamerRegistionConflict(LogDistributionBaseException):
27+
class LogStreamerRegistionConflictError(LogDistributionBaseError):
2828
pass
2929

3030

@@ -63,7 +63,7 @@ async def _distribute_logs(self, data: bytes):
6363
callback = self._log_streamers.get(item.job_id)
6464
if callback is None:
6565
msg = f"Could not forward log because a logstreamer associated with job_id={item.job_id} was not registered"
66-
raise LogStreamerNotRegistered(msg)
66+
raise LogStreamerNotRegisteredError(msg)
6767
await callback(item)
6868
return True
6969

@@ -72,7 +72,7 @@ async def register(
7272
):
7373
if job_id in self._log_streamers:
7474
msg = f"A stream was already connected to {job_id=}. Only a single stream can be connected at the time"
75-
raise LogStreamerRegistionConflict(msg)
75+
raise LogStreamerRegistionConflictError(msg)
7676
self._log_streamers[job_id] = callback
7777
await self._rabbit_client.add_topics(
7878
LoggerRabbitMessage.get_channel_name(), topics=[f"{job_id}.*"]
@@ -81,7 +81,7 @@ async def register(
8181
async def deregister(self, job_id: JobID):
8282
if job_id not in self._log_streamers:
8383
msg = f"No stream was connected to {job_id=}."
84-
raise LogStreamerNotRegistered(msg)
84+
raise LogStreamerNotRegisteredError(msg)
8585
await self._rabbit_client.remove_topics(
8686
LoggerRabbitMessage.get_channel_name(), topics=[f"{job_id}.*"]
8787
)
@@ -128,7 +128,7 @@ async def _project_done(self) -> bool:
128128
async def log_generator(self) -> AsyncIterable[str]:
129129
if not self._is_registered:
130130
msg = f"LogStreamer for job_id={self._job_id} is not correctly registered"
131-
raise LogStreamerNotRegistered(msg)
131+
raise LogStreamerNotRegisteredError(msg)
132132
last_log_time: datetime | None = None
133133
while True:
134134
while self._queue.empty():

services/api-server/tests/unit/test_services_rabbitmq.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@
4141
from simcore_service_api_server.services.log_streaming import (
4242
LogDistributor,
4343
LogStreamer,
44-
LogStreamerNotRegistered,
45-
LogStreamerRegistionConflict,
44+
LogStreamerNotRegisteredError,
45+
LogStreamerRegistionConflictError,
4646
)
4747

4848
pytest_simcore_core_services_selection = [
@@ -217,7 +217,7 @@ async def _(job_log: JobLog):
217217
pass
218218

219219
await log_distributor.register(project_id, _)
220-
with pytest.raises(LogStreamerRegistionConflict):
220+
with pytest.raises(LogStreamerRegistionConflictError):
221221
await log_distributor.register(project_id, _)
222222
await log_distributor.deregister(project_id)
223223

@@ -413,6 +413,6 @@ async def test_log_generator(mocker: MockFixture, faker: Faker):
413413

414414
async def test_log_generator_context(mocker: MockFixture, faker: Faker):
415415
log_streamer = LogStreamer(user_id=3, director2_api=None, job_id=None, log_distributor=None, max_log_check_seconds=1) # type: ignore
416-
with pytest.raises(LogStreamerNotRegistered):
416+
with pytest.raises(LogStreamerNotRegisteredError):
417417
async for log in log_streamer.log_generator():
418418
print(log)

0 commit comments

Comments
 (0)