Opraven skript.

This commit is contained in:
archos 2024-07-06 23:10:18 +02:00
parent 0f255b5138
commit 6b97f87f61

View File

@ -1,7 +1,10 @@
import argparse import argparse
from mastodon import Mastodon from mastodon import Mastodon
import os import os
import sqlite3 import sqlite3
import logging
import time
import tomllib import tomllib
ACCOUNT_FETCH_LIMIT = 100000 ACCOUNT_FETCH_LIMIT = 100000
@ -27,18 +30,52 @@ def set_user_welcomed(cursor: sqlite3.Cursor, userid: int):
def create_user(cursor: sqlite3.Cursor, userid: int, username: str): def create_user(cursor: sqlite3.Cursor, userid: int, username: str):
cursor.execute("INSERT INTO welcome_log (userdb_id, username) VALUES(?, ?)", (userid, username)) cursor.execute("INSERT INTO welcome_log (userdb_id, username) VALUES(?, ?)", (userid, username))
def check_and_welcome_new_users():
logging.debug("Checking if database exists")
fresh_database = not check_db_exists(cursor)
if fresh_database:
logging.info("Database was freshly created - we'll set all pre-existing users as of now to 'welcomed' to avoid spamming everyone on the server.")
logging.info("Starting the welcome process")
all_accounts = mastodon.admin_accounts(remote=False, status='active', limit=ACCOUNT_FETCH_LIMIT)
for account in all_accounts:
logging.debug(f"Checking if user {account['id']} exists")
if not user_exists(cursor, account['id']):
logging.debug(f"Creating user {account['username']} with ID {account['id']}")
create_user(cursor, account['id'], account['username'])
connection.commit()
logging.debug(f"Checking if user {account['id']} was welcomed")
if fresh_database:
set_user_welcomed(cursor, account['id'])
connection.commit()
elif not user_welcomed(cursor, account['id']):
result_id = None
for message in config['messages']:
content_warning = message['content_warning'] if 'content_warning' in message else None
result = mastodon.status_post(status=f"@{account['username']}, {message['content']}", in_reply_to_id=result_id, visibility='public', spoiler_text=content_warning)
result_id = result.id
set_user_welcomed(cursor, account['id'])
connection.commit()
if __name__ == '__main__': if __name__ == '__main__':
try:
import tomllib
except ModuleNotFoundError:
import tomli as tomllib
logging.basicConfig(filename='/home/archos/welcome_bot/welcome_bot.log', level=logging.DEBUG, format='%(asctime)s %(levelname)s:%(message)s')
arg_parser = argparse.ArgumentParser( arg_parser = argparse.ArgumentParser(
description = "Welcomes users to the Mastodon instance", description="Welcomes users to the Mastodon instance",
epilog = "Note: the user this bot logs in as must have admin:read:accounts access" epilog="Note: the user this bot logs in as must have admin:read:accounts access"
) )
arg_parser.add_argument('--email', required=False, help="Only required for first execution") arg_parser.add_argument('--email', required=False, help="Only required for first execution")
arg_parser.add_argument('--password', required=False, help="Only required for first execution") arg_parser.add_argument('--password', required=False, help="Only required for first execution")
arg_parser.add_argument('--config', default="config.toml") arg_parser.add_argument('--config', default="config.toml")
args = arg_parser.parse_args() args = arg_parser.parse_args()
config = None # not strictly required, but I like explicit scope
with open(args.config, "rb") as toml_file: with open(args.config, "rb") as toml_file:
config = tomllib.load(toml_file) config = tomllib.load(toml_file)
@ -46,34 +83,34 @@ if __name__ == '__main__':
if not os.path.exists(config['mastodon']['credential_storage']): if not os.path.exists(config['mastodon']['credential_storage']):
if not (args.email and args.password): if not (args.email and args.password):
print("Initial login has not yet occured - this is required to generate the credential file. Please supply login credentials (--email, --password)") logging.error("Initial login has not yet occurred - this is required to generate the credential file. Please supply login credentials (--email, --password)")
exit(-1) exit(-1)
else: else:
print("Registering app") logging.info("Registering app")
Mastodon.create_app( Mastodon.create_app(
client_name = "Welcome Bot", client_name="Welcome Bot",
scopes = ["write:statuses", "admin:read:accounts"], scopes=["write:statuses", "admin:read:accounts"],
to_file = config['mastodon']['secret_storage'], to_file=config['mastodon']['secret_storage'],
api_base_url = config['mastodon']['base_url'] api_base_url=config['mastodon']['base_url']
) )
print("Initializing client") logging.info("Initializing client")
mastodon = Mastodon( mastodon = Mastodon(
client_id = config['mastodon']['secret_storage'], client_id=config['mastodon']['secret_storage'],
api_base_url = config['mastodon']['base_url'] api_base_url=config['mastodon']['base_url']
) )
print("Performing login") logging.info("Performing login")
mastodon.log_in( mastodon.log_in(
username = args.email, username=args.email,
password = args.password, password=args.password,
to_file = config['mastodon']['credential_storage'], to_file=config['mastodon']['credential_storage'],
scopes = ["write:statuses", "admin:read:accounts"] scopes=["write:statuses", "admin:read:accounts"]
) )
else: else:
mastodon = Mastodon( mastodon = Mastodon(
access_token = config['mastodon']['credential_storage'], access_token=config['mastodon']['credential_storage'],
api_base_url = config['mastodon']['base_url'] api_base_url=config['mastodon']['base_url']
) )
assert mastodon is not None, "Mastodon client not initialized" assert mastodon is not None, "Mastodon client not initialized"
@ -81,33 +118,7 @@ if __name__ == '__main__':
connection = sqlite3.connect(config['database']['sqlite_path']) connection = sqlite3.connect(config['database']['sqlite_path'])
cursor = connection.cursor() cursor = connection.cursor()
# are our tables defined? while True:
fresh_database = not check_db_exists(cursor) check_and_welcome_new_users()
if fresh_database: logging.info("Sleeping for 10 minutes")
print("Database was freshly created - we'll set all pre-existing users as of now to 'welcomed' to avoid spamming everyone on the server.") time.sleep(600)
all_accounts = mastodon.admin_accounts(remote=False, status='active', limit=ACCOUNT_FETCH_LIMIT)
for account in all_accounts:
# despite status='active', we still get zombie users from the API
if not (account.confirmed and account.approved) or account.disabled or account.suspended or account.silenced:
continue
# does our welcome bot know about this user?
if not user_exists(cursor, account.id):
create_user(cursor, account.id, account.username)
connection.commit()
# have we welcomed them yet?
if fresh_database:
set_user_welcomed(cursor, account.id)
connection.commit()
elif not user_welcomed(cursor, account.id):
result_id = None
for message in config['messages']:
content_warning = message['content_warning'] if 'content_warning' in message else None
result = mastodon.status_post(status=f"@{account.username}, {message['content']}", visibility='public', spoiler_text=content_warning)
result_id = result.id
set_user_welcomed(cursor, account.id)
connection.commit()