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

defensive if's -> assert

Browse Source
This commit is contained in:
meejah 2021-04-18 21:25:25 -06:00
parent dfd3bdd1a1
commit 0ce08b66cf
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/wormhole_transit_relay/transit_server.py
View File

@ -52,11 +52,11 @@ class TransitConnection(LineReceiver):
""" """
ITransitClient API ITransitClient API
""" """
if self._buddy is not None: assert self._buddy is not None, "internal error: no buddy"
if self.factory.log_requests: if self.factory.log_requests:
log.msg("buddy_disconnected {}".format(self._buddy.get_token())) log.msg("buddy_disconnected {}".format(self._buddy.get_token()))
self._buddy._client.disconnect() self._buddy._client.disconnect()
self._buddy = None self._buddy = None
def connectionMade(self): def connectionMade(self):
# ideally more like self._reactor.seconds() ... but Twisted # ideally more like self._reactor.seconds() ... but Twisted
@ -203,11 +203,11 @@ class WebSocketTransitConnection(WebSocketServerProtocol):
""" """
ITransitClient API ITransitClient API
""" """
if self._buddy is not None: assert self._buddy is not None, "internal error: no buddy"
if self.factory.log_requests: if self.factory.log_requests:
log.msg("buddy_disconnected {}".format(self._buddy.get_token())) log.msg("buddy_disconnected {}".format(self._buddy.get_token()))
self._buddy._client.disconnect() self._buddy._client.disconnect()
self._buddy = None self._buddy = None
def connectionMade(self): def connectionMade(self):
""" """