open.source/magic-wormhole
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

throw KeyFormatError when given a code with spaces

Browse Source
This commit is contained in:
Brian Warner 2017-03-07 12:09:25 +01:00
parent aebee61816
commit e518f2b799
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/wormhole/_boss.py
View File

@ -17,7 +17,7 @@ from ._rendezvous import RendezvousConnector
from ._lister import Lister
from ._code import Code
from ._terminator import Terminator
from .errors import ServerError, LonelyError, WrongPasswordError
from .errors import ServerError, LonelyError, WrongPasswordError, KeyFormatError
from .util import bytes_to_dict
@attrs
@ -116,6 +116,8 @@ class Boss(object):
def allocate_code(self, code_length):
self._C.allocate_code(code_length)
def set_code(self, code):
if ' ' in code:
raise KeyFormatError("code (%s) contains spaces." % code)
self._C.set_code(code)
@m.input()