forked from indymedia/epicyon
70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
|
__filename__ = "follow.py"
|
||
|
|
__author__ = "Bob Mottram"
|
||
|
|
__license__ = "AGPL3+"
|
||
|
|
__version__ = "0.0.1"
|
||
|
|
__maintainer__ = "Bob Mottram"
|
||
|
|
__email__ = "bob@freedombone.net"
|
||
|
|
__status__ = "Production"
|
||
|
|
|
||
|
|
import json
|
||
|
|
from pprint import pprint
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def followPerson(username: str, domain: str, followUsername: str, followDomain: str, followFile='following.txt') -> None:
|
||
|
|
"""Adds a person to the follow list
|
||
|
|
"""
|
||
|
|
handle=username.lower()+'@'+domain.lower()
|
||
|
|
handleToFollow=followUsername.lower()+'@'+followDomain.lower()
|
||
|
|
baseDir=os.getcwd()
|
||
|
|
if not os.path.isdir(baseDir+'/accounts/'+handle):
|
||
|
|
os.mkdir(baseDir+'/accounts/'+handle)
|
||
|
|
filename=baseDir+'/accounts/'+handle+'/'+followFile
|
||
|
|
if os.path.isfile(filename):
|
||
|
|
if handleToFollow in open(filename).read():
|
||
|
|
return
|
||
|
|
with open(filename, "a") as followfile:
|
||
|
|
followfile.write(handleToFollow+'\n')
|
||
|
|
return
|
||
|
|
with open(filename, "w") as followfile:
|
||
|
|
followfile.write(handleToFollow+'\n')
|
||
|
|
|
||
|
|
def followerOfPerson(username: str, domain: str, followerUsername: str, followerDomain: str) -> None:
|
||
|
|
followPerson(username, domain, followerUsername, followerDomain,'followers.txt')
|
||
|
|
|
||
|
|
def unfollowPerson(username: str, domain: str, followUsername: str, followDomain: str,followFile='following.txt') -> None:
|
||
|
|
"""Removes a person to the follow list
|
||
|
|
"""
|
||
|
|
handle=username.lower()+'@'+domain.lower()
|
||
|
|
handleToUnfollow=followUsername.lower()+'@'+followDomain.lower()
|
||
|
|
baseDir=os.getcwd()
|
||
|
|
if not os.path.isdir(baseDir+'/accounts/'+handle):
|
||
|
|
os.mkdir(baseDir+'/accounts/'+handle)
|
||
|
|
filename=baseDir+'/accounts/'+handle+'/'+followFile
|
||
|
|
if os.path.isfile(filename):
|
||
|
|
if handleToUnfollow not in open(filename).read():
|
||
|
|
return
|
||
|
|
with open(filename, "r") as f:
|
||
|
|
lines = f.readlines()
|
||
|
|
with open(filename, "w") as f:
|
||
|
|
for line in lines:
|
||
|
|
if line.strip("\n") != handleToUnfollow:
|
||
|
|
f.write(line)
|
||
|
|
|
||
|
|
def unfollowerOfPerson(username: str, domain: str, followerUsername: str, followerDomain: str) -> None:
|
||
|
|
unfollowPerson(username, domain, followerUsername, followerDomain,'followers.txt')
|
||
|
|
|
||
|
|
def clearFollows(username: str, domain: str,followFile='following.txt') -> None:
|
||
|
|
"""Removes all follows
|
||
|
|
"""
|
||
|
|
handle=username.lower()+'@'+domain.lower()
|
||
|
|
baseDir=os.getcwd()
|
||
|
|
if not os.path.isdir(baseDir+'/accounts/'+handle):
|
||
|
|
os.mkdir(baseDir+'/accounts/'+handle)
|
||
|
|
filename=baseDir+'/accounts/'+handle+'/'+followFile
|
||
|
|
if os.path.isfile(filename):
|
||
|
|
os.remove(filename)
|
||
|
|
|
||
|
|
def clearFollowers(username: str, domain: str) -> None:
|
||
|
|
clearFollows(username, domain,'followers.txt')
|