flake8 format

main
Bob Mottram 2020-04-03 17:01:25 +00:00
parent 295814cd9b
commit 08a1b05a76
1 changed files with 24 additions and 21 deletions

View File

@ -1,50 +1,53 @@
__filename__="migrate.py"
__author__="Bob Mottram"
__license__="AGPL3+"
__version__="1.1.0"
__maintainer__="Bob Mottram"
__email__="bob@freedombone.net"
__status__="Production"
__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:
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
followData = None
with open(followFilename, 'r') as followFile:
followData=followFile.read()
followData = followFile.read()
if not followData:
return
newFollowData=followData.replace(oldHandle,newHandle)
if followData==newFollowData:
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:
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:]
oldHandle = oldHandle[1:]
if '@' not in oldHandle:
return
if newHandle.startswith('@'):
ewHandle=newHandle[1:]
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 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)
accountDir = baseDir + '/accounts/' + handle
followFilename = accountDir + '/following.txt'
migrateFollows(followFilename, oldHandle, newHandle)
followFilename = accountDir + '/followers.txt'
migrateFollows(followFilename, oldHandle, newHandle)