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

add software kwarg to RelayDatabase.add_inbox

Browse source
This commit is contained in:
Izalia Mae 2022-11-18 16:39:53 -05:00
parent 306b526808
commit 4a8a8da740
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
relay/database.py
View file

@ -129,20 +129,25 @@ class RelayDatabase(dict):
raise KeyError(domain) raise KeyError(domain)
def add_inbox(self, inbox, followid=None, fail=False): def add_inbox(self, inbox, followid=None, software=None):
assert inbox.startswith('https'), 'Inbox must be a url' assert inbox.startswith('https'), 'Inbox must be a url'
domain = urlparse(inbox).hostname domain = urlparse(inbox).hostname
instance = self.get_inbox(domain)
if self.get_inbox(domain): if instance:
if fail: if followid:
raise KeyError(domain) instance['followid'] = followid
return False if software:
instance['software'] = software
return instance
self['relay-list'][domain] = { self['relay-list'][domain] = {
'domain': domain, 'domain': domain,
'inbox': inbox, 'inbox': inbox,
'followid': followid 'followid': followid,
'software': software
} }
logging.verbose(f'Added inbox to database: {inbox}') logging.verbose(f'Added inbox to database: {inbox}')