mirror of https://gitlab.com/bashrc2/epicyon
240 lines
9.2 KiB
Python
240 lines
9.2 KiB
Python
__filename__ = "importFollowing.py"
|
|
__author__ = "Bob Mottram"
|
|
__license__ = "AGPL3+"
|
|
__version__ = "1.3.0"
|
|
__maintainer__ = "Bob Mottram"
|
|
__email__ = "bob@libreserver.org"
|
|
__status__ = "Production"
|
|
__module_group__ = "Core"
|
|
|
|
import os
|
|
import time
|
|
import random
|
|
from utils import acct_dir
|
|
from utils import is_account_dir
|
|
from utils import get_nickname_from_actor
|
|
from utils import get_domain_from_actor
|
|
from follow import is_following_actor
|
|
from follow import send_follow_request
|
|
from session import create_session
|
|
from session import set_session_for_sender
|
|
from threads import begin_thread
|
|
|
|
|
|
def _establish_import_session(httpd,
|
|
calling_function: str,
|
|
curr_session,
|
|
proxy_type: str):
|
|
"""Recreates session if needed
|
|
"""
|
|
if curr_session:
|
|
return curr_session
|
|
print('DEBUG: creating new import session during ' + calling_function)
|
|
curr_session = create_session(proxy_type)
|
|
if curr_session:
|
|
set_session_for_sender(httpd, proxy_type, curr_session)
|
|
return curr_session
|
|
print('ERROR: failed to create import session during ' +
|
|
calling_function)
|
|
return None
|
|
|
|
|
|
def _update_import_following(base_dir: str,
|
|
handle: str, httpd,
|
|
import_filename: str) -> bool:
|
|
"""Send out follow requests from the import csv file
|
|
"""
|
|
following_str = ''
|
|
try:
|
|
with open(import_filename, 'r', encoding='utf-8') as fp_import:
|
|
following_str = fp_import.read()
|
|
except OSError:
|
|
print('Ex: failed to load import file ' + import_filename)
|
|
return False
|
|
if following_str:
|
|
main_session = None
|
|
lines = following_str.split('\n')
|
|
random.shuffle(lines)
|
|
nickname = handle.split('@')[0]
|
|
domain = handle.split('@')[1]
|
|
for line in lines:
|
|
orig_line = line
|
|
notes = None
|
|
line = line.strip()
|
|
if ',' in line:
|
|
fields = line.split(',')
|
|
line = fields[0].strip()
|
|
if len(fields) >= 3:
|
|
notes = fields[2]
|
|
if line.startswith('#'):
|
|
continue
|
|
following_nickname = get_nickname_from_actor(line)
|
|
if not following_nickname:
|
|
continue
|
|
following_domain, following_port = get_domain_from_actor(line)
|
|
if not following_domain:
|
|
continue
|
|
if following_nickname == nickname and \
|
|
following_domain == domain:
|
|
# don't follow yourself
|
|
continue
|
|
following_handle = following_nickname + '@' + following_domain
|
|
if notes:
|
|
notes = notes.replace('<br>', '\n')
|
|
person_notes_filename = \
|
|
acct_dir(base_dir, nickname, domain) + \
|
|
'/notes/' + following_handle + '.txt'
|
|
try:
|
|
with open(person_notes_filename, 'w+',
|
|
encoding='utf-8') as fp_notes:
|
|
fp_notes.write(notes)
|
|
except OSError:
|
|
print('EX: Unable to import notes for ' +
|
|
following_handle)
|
|
if is_following_actor(base_dir,
|
|
nickname, domain, following_handle):
|
|
# remove the followed handle from the import list
|
|
following_str = following_str.replace(orig_line + '\n', '')
|
|
try:
|
|
with open(import_filename, 'w+',
|
|
encoding='utf-8') as fp_import:
|
|
fp_import.write(following_str)
|
|
except OSError:
|
|
print('EX: unable to remove import 1 ' + line +
|
|
' from ' + import_filename)
|
|
continue
|
|
|
|
# send follow request
|
|
curr_domain = domain
|
|
curr_port = httpd.port
|
|
curr_http_prefix = httpd.http_prefix
|
|
following_actor = following_handle
|
|
if ':' in following_domain:
|
|
following_port = following_handle.split(':')[1]
|
|
following_domain = following_domain.split(':')[0]
|
|
|
|
# get the appropriate session
|
|
curr_session = main_session
|
|
curr_proxy_type = httpd.proxy_type
|
|
use_onion_session = False
|
|
use_i2p_session = False
|
|
if '.onion' not in domain and \
|
|
httpd.onion_domain and '.onion' in following_domain:
|
|
curr_session = httpd.session_onion
|
|
curr_domain = httpd.onion_domain
|
|
curr_port = 80
|
|
following_port = 80
|
|
curr_http_prefix = 'http'
|
|
curr_proxy_type = 'tor'
|
|
use_onion_session = True
|
|
if '.i2p' not in domain and \
|
|
httpd.i2p_domain and '.i2p' in following_domain:
|
|
curr_session = httpd.session_i2p
|
|
curr_domain = httpd.i2p_domain
|
|
curr_port = 80
|
|
following_port = 80
|
|
curr_http_prefix = 'http'
|
|
curr_proxy_type = 'i2p'
|
|
use_i2p_session = True
|
|
|
|
curr_session = \
|
|
_establish_import_session(httpd, "import follow",
|
|
curr_session, curr_proxy_type)
|
|
if curr_session:
|
|
if use_onion_session:
|
|
httpd.session_onion = curr_session
|
|
elif use_i2p_session:
|
|
httpd.session_i2p = curr_session
|
|
else:
|
|
main_session = curr_session
|
|
|
|
send_follow_request(curr_session,
|
|
base_dir, nickname,
|
|
domain, curr_domain, curr_port,
|
|
curr_http_prefix,
|
|
following_nickname,
|
|
following_domain,
|
|
following_actor,
|
|
following_port, curr_http_prefix,
|
|
False, httpd.federation_list,
|
|
httpd.send_threads,
|
|
httpd.postLog,
|
|
httpd.cached_webfingers,
|
|
httpd.person_cache, httpd.debug,
|
|
httpd.project_version,
|
|
httpd.signing_priv_key_pem,
|
|
httpd.domain,
|
|
httpd.onion_domain,
|
|
httpd.i2p_domain)
|
|
|
|
# remove the followed handle from the import list
|
|
following_str = following_str.replace(orig_line + '\n', '')
|
|
try:
|
|
with open(import_filename, 'w+',
|
|
encoding='utf-8') as fp_import:
|
|
fp_import.write(following_str)
|
|
except OSError:
|
|
print('EX: unable to remove import 2 ' + line +
|
|
' from ' + import_filename)
|
|
return True
|
|
return False
|
|
|
|
|
|
def run_import_following(base_dir: str, httpd):
|
|
"""Sends out follow requests for imported following csv files
|
|
"""
|
|
while True:
|
|
time.sleep(20)
|
|
|
|
# get a list of accounts on the instance, in random sequence
|
|
accounts_list = []
|
|
for _, dirs, _ in os.walk(base_dir + '/accounts'):
|
|
for account in dirs:
|
|
if '@' not in account:
|
|
continue
|
|
if not is_account_dir(account):
|
|
continue
|
|
accounts_list.append(account)
|
|
break
|
|
if not accounts_list:
|
|
continue
|
|
|
|
# check if each accounts has an import csv
|
|
random.shuffle(accounts_list)
|
|
for account in accounts_list:
|
|
account_dir = base_dir + '/accounts/' + account
|
|
import_filename = account_dir + '/import_following.csv'
|
|
|
|
if not os.path.isfile(import_filename):
|
|
continue
|
|
if not _update_import_following(base_dir, account, httpd,
|
|
import_filename):
|
|
try:
|
|
os.remove(import_filename)
|
|
except OSError:
|
|
print('EX: unable to remove import file ' +
|
|
import_filename)
|
|
else:
|
|
break
|
|
|
|
|
|
def run_import_following_watchdog(project_version: str, httpd) -> None:
|
|
"""Imports following lists from csv for every account on the instance
|
|
"""
|
|
print('THREAD: Starting import following watchdog ' + project_version)
|
|
import_following_original = \
|
|
httpd.thrImportFollowing.clone(run_import_following)
|
|
begin_thread(httpd.thrImportFollowing,
|
|
'run_import_following_watchdog')
|
|
while True:
|
|
time.sleep(50)
|
|
if httpd.thrImportFollowing.is_alive():
|
|
continue
|
|
httpd.thrImportFollowing.kill()
|
|
print('THREAD: restarting import following watchdog')
|
|
httpd.thrImportFollowing = \
|
|
import_following_original.clone(run_import_following)
|
|
begin_thread(httpd.thrImportFollowing,
|
|
'run_import_following_watchdog 2')
|
|
print('Restarting import following...')
|