shaba/openuds
1
0
Fork 1
mirror of https://github.com/dkmstr/openuds.git synced 2025-01-03 01:17:56 +03:00
Code Issues Releases Wiki Activity

chore: Update restrain_server function signature to specify return type

Browse Source
This commit is contained in:
Adolfo Gómez García 2024-08-23 02:00:25 +02:00
parent a568ff9a28
commit bb448d2876
No known key found for this signature in database
GPG Key ID: DD1ABF20724CDA23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/src/uds/core/managers/servers_api/requester.py
View File

@ -55,8 +55,8 @@ AUTH_TOKEN = 'X-TOKEN-AUTH'
# If server is restrained, it will return False
# If server is not restrained, it will execute the function and return it's result
# If exception is raised, it will restrain the server and return False
def restrain_server(func: collections.abc.Callable[..., typing.Any]) -> collections.abc.Callable[..., typing.Any]:
def inner(self: 'ServerApiRequester', *args: typing.Any, **kwargs: typing.Any) -> typing.Any:
def restrain_server(func: collections.abc.Callable[..., bool]) -> collections.abc.Callable[..., bool]:
def inner(self: 'ServerApiRequester', *args: typing.Any, **kwargs: typing.Any) -> bool:
if self.server.is_restrained():
return False
@ -64,7 +64,7 @@ def restrain_server(func: collections.abc.Callable[..., typing.Any]) -> collecti
return func(self, *args, **kwargs)
except Exception as e:
restrained_until = sql_now() + datetime.timedelta(seconds=consts.system.FAILURE_TIMEOUT)
logger.exception('Error executing %s: %s. Server restrained until %s', func.__name__, e, restrained_until)
logger.error('Error executing %s: %s. Server restrained until %s', func.__name__, e, restrained_until)
self.server.set_restrained_until(
restrained_until
) # Block server for a while