Include both following and follower domains within broch

merge-requests/30/head
Bob Mottram 2021-02-15 22:42:16 +00:00
parent 6b55ceab12
commit f4fdedba87
1 changed files with 14 additions and 12 deletions

View File

@ -380,6 +380,7 @@ def setBrochMode(baseDir: str, domainFull: str, enabled: bool) -> None:
else:
# generate instance allow list
allowedDomains = [domainFull]
followFiles = ('following.txt', 'followers.txt')
for subdir, dirs, files in os.walk(baseDir + '/accounts'):
for acct in dirs:
if '@' not in acct:
@ -387,7 +388,8 @@ def setBrochMode(baseDir: str, domainFull: str, enabled: bool) -> None:
if 'inbox@' in acct or 'news@' in acct:
continue
accountDir = os.path.join(baseDir + '/accounts', acct)
followingFilename = accountDir + '/following.txt'
for followFileType in followFiles:
followingFilename = accountDir + '/' + followFileType
if not os.path.isfile(followingFilename):
continue
with open(followingFilename, "r") as f: