optimize RelayDatabase.get_inbox

This commit is contained in:
Izalia Mae 2022-11-18 14:36:30 -05:00
parent 9369b598fa
commit 4ea6a040fb

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'