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

remove a few lines by using defaultdict()

Browse Source
This commit is contained in:
Brian Warner 2018-02-12 17:34:11 -08:00
parent 73d233a7be
commit 098e315624
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/wormhole_transit_relay/transit_server.py
View File

@ -1,5 +1,6 @@
from __future__ import print_function, unicode_literals from __future__ import print_function, unicode_literals
import re, time, json import re, time, json
from collections import defaultdict
from twisted.python import log from twisted.python import log
from twisted.internet import protocol from twisted.internet import protocol
from .database import get_db from .database import get_db
@ -265,12 +266,10 @@ class Transit(protocol.ServerFactory):
self._db = get_db(usage_db) self._db = get_db(usage_db)
self._rebooted = time.time() self._rebooted = time.time()
# we don't track TransitConnections until they submit a token # we don't track TransitConnections until they submit a token
self._pending_requests = {} # token -> set((side, TransitConnection)) self._pending_requests = defaultdict(set) # token -> set((side, TransitConnection))
self._active_connections = set() # TransitConnection self._active_connections = set() # TransitConnection
def connection_got_token(self, token, new_side, new_tc): def connection_got_token(self, token, new_side, new_tc):
if token not in self._pending_requests:
self._pending_requests[token] = set()
potentials = self._pending_requests[token] potentials = self._pending_requests[token]
for old in potentials: for old in potentials:
(old_side, old_tc) = old (old_side, old_tc) = old