indymedia
/
epicyon
mirror of https://gitlab.com/bashrc2/epicyon
7
4
Fork 1
Code Issues 14 Releases Wiki Activity

Also remove moved accounts from followers

main
Bob Mottram 2021-01-09 20:27:17 +00:00
parent e7d0668cd6
commit 6d0678db54
1 changed files with 58 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
migrate.py
View File

@ -19,12 +19,12 @@ from follow import unfollowAccount
def _moveFollowingHandlesForAccount(baseDir: str, nickname: str, domain: str,
session,
httpPrefix: str, cachedWebfingers: {},
followFile: str, debug: bool) -> int:
debug: bool) -> int:
"""Goes through all follows for an account and updates any that have moved
"""
ctr = 0
followingFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + '/' + followFile
baseDir + '/accounts/' + nickname + '@' + domain + '/following.txt'
if not os.path.isfile(followingFilename):
return ctr
with open(followingFilename, "r") as f:
@ -35,14 +35,14 @@ def _moveFollowingHandlesForAccount(baseDir: str, nickname: str, domain: str,
_updateMovedHandle(baseDir, nickname, domain,
followHandle, session,
httpPrefix, cachedWebfingers,
followFile, debug)
debug)
return ctr
def _updateMovedHandle(baseDir: str, nickname: str, domain: str,
handle: str, session,
httpPrefix: str, cachedWebfingers: {},
followFile: str, debug: bool) -> int:
debug: bool) -> int:
"""Check if an account has moved, and if so then alter following.txt
for each account.
Returns 1 if moved, 0 otherwise
@ -112,12 +112,12 @@ def _updateMovedHandle(baseDir: str, nickname: str, domain: str,
# so just unfollow them
unfollowAccount(baseDir, nickname, domain,
movedToNickname, movedToDomainFull,
followFile, debug)
'following.txt', debug)
return ctr
followingFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + '/' + followFile
if not os.path.isfile(followingFilename):
return ctr
baseDir + '/accounts/' + nickname + '@' + domain + '/following.txt'
if os.path.isfile(followingFilename):
with open(followingFilename, "r") as f:
followingHandles = f.readlines()
@ -125,7 +125,8 @@ def _updateMovedHandle(baseDir: str, nickname: str, domain: str,
handleLower = handle.lower()
refollowFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + '/refollow.txt'
baseDir + '/accounts/' + \
nickname + '@' + domain + '/refollow.txt'
# unfollow the old handle
with open(followingFilename, 'w+') as f:
@ -139,7 +140,7 @@ def _updateMovedHandle(baseDir: str, nickname: str, domain: str,
unfollowAccount(baseDir, nickname, domain,
handleNickname,
handleDomain,
followFile, debug)
'following.txt', debug)
ctr += 1
print('Unfollowed ' + handle + ' who has moved to ' +
movedToHandle)
@ -151,6 +152,25 @@ def _updateMovedHandle(baseDir: str, nickname: str, domain: str,
else:
with open(refollowFilename, 'w+') as f:
f.write(movedToHandle + '\n')
followersFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + '/followers.txt'
if os.path.isfile(followersFilename):
with open(followersFilename, "r") as f:
followerHandles = f.readlines()
handleLower = handle.lower()
# remove followers who have moved
with open(followersFilename, 'w+') as f:
for followerHandle in followerHandles:
if followerHandle.strip("\n").strip("\r").lower() != \
handleLower:
f.write(followerHandle)
else:
ctr += 1
print('Removed follower who has moved ' + handle)
return ctr
@ -176,7 +196,6 @@ def migrateAccounts(baseDir: str, session,
ctr += \
_moveFollowingHandlesForAccount(baseDir, nickname, domain,
session, httpPrefix,
cachedWebfingers,
'following.txt', debug)
cachedWebfingers, debug)
break
return ctr