fediseer/updater.py

from dotenv import load_dotenv
import os
import logging

load_dotenv()

from loguru import logger
from fediseer.flask import OVERSEER, db 
import fediseer.database.functions as database
from fediseer.register import ensure_instance_registered
from concurrent.futures import ThreadPoolExecutor

def refresh_info(domain):
    logger.info(f"Refreshing domain '{domain}")
    with OVERSEER.app_context():
        try:
            ensure_instance_registered(
                domain,
                # We  don't want to set allow_unreachable = True here
                # As InstanceInfo() won't raise an exception when failing
                # Which will cause the poll_failures not not increment
                allow_unreachable=False,
                record_unreachable=True,
                allowed_timeout=20
            )
        except Exception as err:
            pass

if __name__ == "__main__":
    # Only setting this for the WSGI logs
    logging.basicConfig(format='%(asctime)s - %(levelname)s - %(module)s:%(lineno)d - %(message)s', level=logging.WARNING)
    logger.init("Updater", status="Starting")
    logger.info(f"Skipping instances with poll fails > {int(os.getenv('FEDISEER_IGNORE_POLL_FAILS', 0))}")
    # try: # Debug
    #     with OVERSEER.app_context():
    #         ensure_instance_registered(
    #             "firefish.social",
    #             allow_unreachable=False,
    #             record_unreachable=True,
    #             allowed_timeout=20
    #         )         
    # except Exception as err:
    #     logger.error(err)
    futures = []
    with ThreadPoolExecutor(max_workers=int(os.getenv('FEDISEER_UPDATE_THREADS', 25))) as executor:
        with OVERSEER.app_context():
            for instance in database.get_all_instances(0,0):
                if instance.software == 'wildcard':
                    continue
                # -1 doesn't skip anything
                if int(os.getenv('FEDISEER_IGNORE_POLL_FAILS', -1)) >= 0 and instance.poll_failures > int(os.getenv('FEDISEER_IGNORE_POLL_FAILS', 0)):
                    logger.debug(f"Skipped {instance.domain} due to too many poll fails.")
                    continue
                futures.append(executor.submit(refresh_info, instance.domain))
                if len(futures) >= 500:
                    for future in futures:
                        future.result()
                    futures = []
            for future in futures:
                future.result()

    logger.init("Updater", status="Ended")
[]