mirror of https://gitlab.com/bashrc2/epicyon
146 lines
6.3 KiB
Python
146 lines
6.3 KiB
Python
__filename__ = "manualapprove.py"
|
|
__author__ = "Bob Mottram"
|
|
__license__ = "AGPL3+"
|
|
__version__ = "1.0.0"
|
|
__maintainer__ = "Bob Mottram"
|
|
__email__ = "bob@freedombone.net"
|
|
__status__ = "Production"
|
|
|
|
import os
|
|
import json
|
|
import commentjson
|
|
|
|
from follow import followedAccountAccepts
|
|
from follow import followedAccountRejects
|
|
from follow import removeFromFollowRequests
|
|
|
|
def manualDenyFollowRequest(session,baseDir: str, \
|
|
httpPrefix: str,
|
|
nickname: str,domain: str,port: int, \
|
|
denyHandle: str, \
|
|
federationList: [], \
|
|
sendThreads: [],postLog: [], \
|
|
cachedWebfingers: {},personCache: {}, \
|
|
debug: bool, \
|
|
projectVersion: str) -> None:
|
|
"""Manually deny a follow request
|
|
"""
|
|
handle=nickname+'@'+domain
|
|
accountsDir=baseDir+'/accounts/'+handle
|
|
|
|
# has this handle already been rejected?
|
|
rejectedFollowsFilename=accountsDir+'/followrejects.txt'
|
|
if os.path.isfile(rejectedFollowsFilename):
|
|
if denyHandle in open(rejectedFollowsFilename).read():
|
|
removeFromFollowRequests(baseDir,nickname,domain,denyHandle)
|
|
print(denyHandle+' has already been rejected as a follower of '+nickname)
|
|
return
|
|
|
|
removeFromFollowRequests(baseDir,nickname,domain,denyHandle)
|
|
|
|
# Store rejected follows
|
|
rejectsFile=open(rejectedFollowsFilename, "a+")
|
|
rejectsFile.write(denyHandle+'\n')
|
|
rejectsFile.close()
|
|
|
|
denyNickname=denyHandle.split('@')[0]
|
|
approveDomain=denyHandle.split('@')[1].replace('\n','')
|
|
approvePort=port
|
|
if ':' in denyDomain:
|
|
denyPort=denyDomain.split(':')[1]
|
|
denyDomain=denyDomain.split(':')[0]
|
|
followedAccountRejects(session,baseDir,httpPrefix, \
|
|
nickname,domain,port, \
|
|
denyNickname,denyDomain,denyPort, \
|
|
followJson['actor'],federationList, \
|
|
followJson, \
|
|
sendThreads,postLog, \
|
|
cachedWebfingers,personCache, \
|
|
debug,projectVersion)
|
|
|
|
print('Follow request from '+denyHandle+' was denied.')
|
|
|
|
def manualApproveFollowRequest(session,baseDir: str, \
|
|
httpPrefix: str,
|
|
nickname: str,domain: str,port: int, \
|
|
approveHandle: str, \
|
|
federationList: [], \
|
|
sendThreads: [],postLog: [], \
|
|
cachedWebfingers: {},personCache: {}, \
|
|
acceptedCaps: [], \
|
|
debug: bool, \
|
|
projectVersion: str) -> None:
|
|
"""Manually approve a follow request
|
|
"""
|
|
print('Manually approving follow request from '+approveHandle)
|
|
handle=nickname+'@'+domain
|
|
accountsDir=baseDir+'/accounts/'+handle
|
|
approveFollowsFilename=accountsDir+'/followrequests.txt'
|
|
if not os.path.isfile(approveFollowsFilename):
|
|
if debug:
|
|
print('WARN: Follow requests file '+approveFollowsFilename+' not found')
|
|
return
|
|
if approveHandle not in open(approveFollowsFilename).read():
|
|
if debug:
|
|
print(handle+' not in '+approveFollowsFilename)
|
|
return
|
|
|
|
approvefilenew = open(approveFollowsFilename+'.new', 'w+')
|
|
updateApprovedFollowers=False
|
|
with open(approveFollowsFilename, 'r') as approvefile:
|
|
for handle in approvefile:
|
|
if handle.startswith(approveHandle):
|
|
handle=handle.replace('\n','')
|
|
port2=port
|
|
if ':' in handle:
|
|
port2=int(handle.split(':')[1])
|
|
requestsDir=accountsDir+'/requests'
|
|
followActivityfilename=requestsDir+'/'+handle+'.follow'
|
|
if os.path.isfile(followActivityfilename):
|
|
followJson=None
|
|
tries=0
|
|
while tries<5:
|
|
try:
|
|
with open(followActivityfilename, 'r') as fp:
|
|
followJson=commentjson.load(fp)
|
|
break
|
|
except Exception as e:
|
|
print(e)
|
|
time.sleep(1)
|
|
tries+=1
|
|
if followJson:
|
|
approveNickname=approveHandle.split('@')[0]
|
|
approveDomain=approveHandle.split('@')[1].replace('\n','')
|
|
approvePort=port2
|
|
if ':' in approveDomain:
|
|
approvePort=approveDomain.split(':')[1]
|
|
approveDomain=approveDomain.split(':')[0]
|
|
print('Sending Accept for '+handle+' follow request from '+approveHandle)
|
|
followedAccountAccepts(session,baseDir,httpPrefix, \
|
|
nickname,domain,port, \
|
|
approveNickname,approveDomain,approvePort, \
|
|
followJson['actor'],federationList, \
|
|
followJson,acceptedCaps, \
|
|
sendThreads,postLog, \
|
|
cachedWebfingers,personCache, \
|
|
debug,projectVersion)
|
|
os.remove(followActivityfilename)
|
|
updateApprovedFollowers=True
|
|
else:
|
|
approvefilenew.write(handle)
|
|
approvefilenew.close()
|
|
os.rename(approveFollowsFilename+'.new',approveFollowsFilename)
|
|
|
|
if updateApprovedFollowers:
|
|
# update the followers
|
|
followersFilename=accountsDir+'/followers.txt'
|
|
if os.path.isfile(followersFilename):
|
|
if approveHandle not in open(followersFilename).read():
|
|
followersFile=open(followersFilename, "a+")
|
|
followersFile.write(approveHandle+'\n')
|
|
followersFile.close()
|
|
else:
|
|
followersFile=open(followersFilename, "w+")
|
|
followersFile.write(approveHandle+'\n')
|
|
followersFile.close()
|