41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
|
from datetime import timedelta
|
||
|
|
|
||
|
|
from django.utils import timezone
|
||
|
|
from loguru import logger
|
||
|
|
|
||
|
|
from common.models import BaseJob, JobManager
|
||
|
|
from mastodon.api import detect_server_info
|
||
|
|
from mastodon.models import MastodonApplication
|
||
|
|
|
||
|
|
|
||
|
|
@JobManager.register
|
||
|
|
class MastodonSiteCheck(BaseJob):
|
||
|
|
max_unreachable_days = 31
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
logger.info("Mastodon Site Check start.")
|
||
|
|
count_checked = 0
|
||
|
|
count_unreachable = 0
|
||
|
|
count_disabled = 0
|
||
|
|
for site in MastodonApplication.objects.exclude(disabled=True):
|
||
|
|
domain = None
|
||
|
|
count_checked += 1
|
||
|
|
try:
|
||
|
|
domain, api_domain, v = detect_server_info(site.domain_name)
|
||
|
|
site.last_reachable_date = timezone.now()
|
||
|
|
except:
|
||
|
|
logger.warning(f"Failed to detect server info for {site.domain_name}")
|
||
|
|
count_unreachable += 1
|
||
|
|
if site.last_reachable_date is None:
|
||
|
|
site.last_reachable_date = timezone.now() - timedelta(days=1)
|
||
|
|
if timezone.now() > site.last_reachable_date + timedelta(
|
||
|
|
days=self.max_unreachable_days
|
||
|
|
):
|
||
|
|
site.disabled = True
|
||
|
|
count_disabled += 1
|
||
|
|
finally:
|
||
|
|
site.save(update_fields=["last_reachable_date", "disabled"])
|
||
|
|
logger.info(
|
||
|
|
f"Mastodon Site Check finished, {count_checked} checked, {count_unreachable} unreachable, {count_disabled} disabled."
|
||
|
|
)
|