pch_xyz/sedi-relay
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

optimize RelayDatabase.get_inbox

Browse source
This commit is contained in:
Izalia Mae 2022-11-18 14:36:30 -05:00
parent 9369b598fa
commit 4ea6a040fb
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
relay/database.py
View file

@ -120,14 +120,14 @@ class RelayDatabase(dict):
if domain.startswith('http'): if domain.startswith('http'):
domain = urlparse(domain).hostname domain = urlparse(domain).hostname
if domain not in self['relay-list']: inbox = self['relay-list'].get(domain)
if inbox:
return inbox
if fail: if fail:
raise KeyError(domain) raise KeyError(domain)
return
return self['relay-list'][domain]
def add_inbox(self, inbox, followid=None, fail=False): def add_inbox(self, inbox, followid=None, fail=False):
assert inbox.startswith('https'), 'Inbox must be a url' assert inbox.startswith('https'), 'Inbox must be a url'