epicyon/migrate.py

54 lines
1.7 KiB
Python

__filename__ = "migrate.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.1.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@freedombone.net"
__status__ = "Production"
import os
def migrateFollows(followFilename: str, oldHandle: str,
newHandle: str) -> None:
"""Changes a handle within following or followers list
"""
if not os.path.isfile(followFilename):
return
if oldHandle not in open(followFilename).read():
return
followData = None
with open(followFilename, 'r') as followFile:
followData = followFile.read()
if not followData:
return
newFollowData = followData.replace(oldHandle, newHandle)
if followData == newFollowData:
return
with open(followFilename, 'w') as followFile:
followFile.write(newFollowData)
def migrateAccount(baseDir: str, oldHandle: str, newHandle: str) -> None:
"""If a followed account changes then this modifies the
following and followers lists for each account accordingly
"""
if oldHandle.startswith('@'):
oldHandle = oldHandle[1:]
if '@' not in oldHandle:
return
if newHandle.startswith('@'):
newHandle = newHandle[1:]
if '@' not in newHandle:
return
# update followers and following lists for each account
for subdir, dirs, files in os.walk(baseDir + '/accounts'):
for handle in dirs:
if '@' in handle:
accountDir = baseDir + '/accounts/' + handle
followFilename = accountDir + '/following.txt'
migrateFollows(followFilename, oldHandle, newHandle)
followFilename = accountDir + '/followers.txt'
migrateFollows(followFilename, oldHandle, newHandle)