forked from indymedia/epicyon
55 lines
1.8 KiB
Python
55 lines
1.8 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)
|
|
break
|