Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

_log_and_render_error

Browse source
This commit is contained in:
Jack 2016-11-22 15:11:25 -05:00
parent ba7f326e72
commit e08d0eb8f9
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
lbrynet/lbrynet_daemon/auth/server.py
View file

@ -95,17 +95,14 @@ class AuthJSONRPCServer(AuthorizedBase):
def __init__(self, use_authentication=settings.use_auth_http):
AuthorizedBase.__init__(self)
self._use_authentication = use_authentication
self.announced_startup = False
self.allowed_during_startup = []
self.sessions = {}
def setup(self):
return NotImplementedError()
def _render_error(self, failure, request, version=jsonrpclib.VERSION_1, response_code=FAILURE,
log_failure=False, log_msg=None):
if log_failure:
msg = log_msg or "API Failure: %s"
log_support.failure(Failure(failure), log, msg)
def _render_error(self, failure, request, version=jsonrpclib.VERSION_1, response_code=FAILURE):
err = JSONRPCException(Failure(failure), response_code)
fault = jsonrpclib.dumps(err, version=version)
self._set_headers(request, fault)
@ -114,6 +111,11 @@ class AuthJSONRPCServer(AuthorizedBase):
request.write(fault)
request.finish()
def _log_and_render_error(self, failure, request, message=None, **kwargs):
msg = message or "API Failure: %s"
log_support.failure(Failure(failure), log, msg)
self._render_error(failure, request, **kwargs)
def render(self, request):
notify_finish = request.notifyFinish()
assert self._check_headers(request), InvalidHeaderError
@ -163,8 +165,7 @@ class AuthJSONRPCServer(AuthorizedBase):
self._verify_token(session_id, parsed, token)
except InvalidAuthenticationToken as err:
log.warning("API validation failed")
self._render_error(err, request, version,
response_code=AuthJSONRPCServer.UNAUTHORIZED)
self._render_error(err, request, version=version, response_code=AuthJSONRPCServer.UNAUTHORIZED)
return server.NOT_DONE_YET
self._update_session_secret(session_id)
reply_with_next_secret = True
@ -184,7 +185,7 @@ class AuthJSONRPCServer(AuthorizedBase):
# cancel the response if the connection is broken
notify_finish.addErrback(self._response_failed, d)
d.addCallback(self._callback_render, request, version, reply_with_next_secret)
d.addErrback(self._render_error, request, version, log_failure=True)
d.addErrback(self._log_and_render_error, request, version=version)
return server.NOT_DONE_YET
def _register_user_session(self, session_id):
@ -285,8 +286,7 @@ class AuthJSONRPCServer(AuthorizedBase):
self._render_message(request, encoded_message)
except Exception as err:
msg = "Failed to render API response: %s"
self._render_error(err, request, response_code=self.FAILURE, version=version,
log_failure=True, log_msg=msg)
self._log_and_render_error(err, request, message=msg, version=version)
def _render_response(self, result, code):
return defer.succeed({'result': result, 'code': code})