jinql
/
webssh
mirror of https://github.com/huashengdun/webssh
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Use class attribute formater

pull/26/head
Sheng 2018-08-26 19:31:45 +08:00
parent b69bdefcc9
commit 9b38d111f4
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
webssh/handler.py
View File

@ -44,6 +44,8 @@ class InvalidException(Exception):
class MixinHandler(object):
formater = 'Missing value {}'
def write_error(self, status_code, **kwargs):
exc_info = kwargs.get('exc_info')
if exc_info and len(exc_info) > 1:
@ -52,10 +54,10 @@ class MixinHandler(object):
self._reason = info.split(':', 1)[-1].strip()
super(MixinHandler, self).write_error(status_code, **kwargs)
def get_value(self, name, formater='Missing value {}'):
def get_value(self, name):
value = self.get_argument(name)
if not value:
raise InvalidException(formater.format(name))
raise InvalidException(self.formater.format(name))
return value
def get_real_client_addr(self):
@ -262,7 +264,7 @@ class WsockHandler(MixinHandler, tornado.websocket.WebSocketHandler):
self.src_addr = self.get_client_addr()
logging.info('Connected from {}:{}'.format(*self.src_addr))
try:
worker_id = self.get_value('id', formater=self.formater)
worker_id = self.get_value('id')
except (tornado.web.MissingArgumentError, InvalidException) as exc:
self.close(reason=str(exc).split(':', 1)[-1].strip())
else: