Return lists

main
Bob Mottram 2020-09-25 15:14:59 +01:00
parent 351cb6ad14
commit b1672fb9d5
3 changed files with 25 additions and 6 deletions

View File

@ -8,7 +8,7 @@ __status__ = "Production"
from pprint import pprint from pprint import pprint
import os import os
from utils import getFollowersOfPerson from utils import getFollowersList
from utils import validNickname from utils import validNickname
from utils import domainPermitted from utils import domainPermitted
from utils import getDomainFromActor from utils import getDomainFromActor
@ -118,9 +118,9 @@ def getMutualsOfPerson(baseDir: str,
i.e. accounts which they follow and which also follow back i.e. accounts which they follow and which also follow back
""" """
followers = \ followers = \
getFollowersOfPerson(baseDir, nickname, domain, 'followers.txt') getFollowersList(baseDir, nickname, domain, 'followers.txt')
following = \ following = \
getFollowersOfPerson(baseDir, nickname, domain, 'following.txt') getFollowersList(baseDir, nickname, domain, 'following.txt')
mutuals = [] mutuals = []
for handle in following: for handle in following:
if handle in followers: if handle in followers:

View File

@ -30,7 +30,7 @@ from session import postJsonString
from session import postImage from session import postImage
from webfinger import webfingerHandle from webfinger import webfingerHandle
from httpsig import createSignedHeader from httpsig import createSignedHeader
from utils import getFollowersOfPerson from utils import getFollowersList
from utils import isEvil from utils import isEvil
from utils import removeIdEnding from utils import removeIdEnding
from utils import siteIsActive from utils import siteIsActive
@ -3338,9 +3338,9 @@ def getNonMutualsOfPerson(baseDir: str,
i.e. accounts which follow you but you don't follow them i.e. accounts which follow you but you don't follow them
""" """
followers = \ followers = \
getFollowersOfPerson(baseDir, nickname, domain, 'followers.txt') getFollowersList(baseDir, nickname, domain, 'followers.txt')
following = \ following = \
getFollowersOfPerson(baseDir, nickname, domain, 'following.txt') getFollowersList(baseDir, nickname, domain, 'following.txt')
nonMutuals = [] nonMutuals = []
for handle in following: for handle in following:
if handle not in followers: if handle not in followers:

View File

@ -19,6 +19,25 @@ from calendar import monthrange
from followingCalendar import addPersonToCalendar from followingCalendar import addPersonToCalendar
def getFollowersList(baseDir: str,
nickname: str, domain: str,
followFile='following.txt') -> []:
"""Returns a list of followers for the given account
"""
filename = \
baseDir + '/accounts/' + nickname + '@' + domain + '/' + followFile
if not os.path.isfile(filename):
return []
with open(filename, "r") as f:
lines = f.readlines()
for i in range(len(lines)):
lines[i] = lines[i].strip()
return lines
return []
def getFollowersOfPerson(baseDir: str, def getFollowersOfPerson(baseDir: str,
nickname: str, domain: str, nickname: str, domain: str,
followFile='following.txt') -> []: followFile='following.txt') -> []: