mirror of
https://codeberg.org/Freedium-cfd/web.git
synced 2026-03-11 09:04:37 +00:00
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
import sentry_sdk
|
|
from server import config
|
|
|
|
if config.SENTRY_SDK_DSN:
|
|
sentry_sdk.init(dsn=config.SENTRY_SDK_DSN, traces_sample_rate=config.SENTRY_TRACES_SAMPLE_RATE, profiles_sample_rate=config.SENTRY_PROFILES_SAMPLE_RATE)
|
|
|
|
|
|
from contextvars import ContextVar
|
|
from multiprocessing import Value
|
|
from typing import Optional
|
|
|
|
import pickledb
|
|
import redis.asyncio as redis
|
|
import time
|
|
from psycopg2 import OperationalError, connect
|
|
from database_lib import PostgreSQLCacheBackend, migrate_to_postgres, execute_migrate_to_postgres_in_thread
|
|
from loguru import logger
|
|
from xkcdpass import xkcd_password as xp
|
|
|
|
from server.utils.logger import configure_logger
|
|
from medium_parser.core import MediumParser
|
|
from server.utils.loguru_handler import InterceptHandler
|
|
|
|
|
|
def wait_for_postgres(max_retries=5, retry_interval=5):
|
|
retries = 0
|
|
while retries < max_retries:
|
|
try:
|
|
conn = connect("postgresql://postgres:postgres@postgres:5432/postgres")
|
|
conn.close()
|
|
logger.info("Successfully connected to PostgreSQL")
|
|
return
|
|
except OperationalError as e:
|
|
logger.warning(f"PostgreSQL is not ready. Retrying in {retry_interval} seconds... (Attempt {retries + 1}/{max_retries})")
|
|
time.sleep(retry_interval)
|
|
retries += 1
|
|
|
|
raise Exception("Failed to connect to PostgreSQL after multiple attempts")
|
|
|
|
|
|
wait_for_postgres()
|
|
|
|
# logging.basicConfig(handlers=[InterceptHandler()], level=0, force=True)
|
|
configure_logger()
|
|
|
|
medium_cache = PostgreSQLCacheBackend("postgresql://postgres:postgres@postgres:5432/postgres")
|
|
medium_cache.init_db()
|
|
# migrate_to_postgres_thread = execute_migrate_to_postgres_in_thread("medium_db_cache.sqlite", "postgresql://postgres:postgres@postgres:5432/postgres")
|
|
logger.debug(f"Database length: {medium_cache.all_length()}")
|
|
|
|
medium_parser = MediumParser(cache=medium_cache, timeout=3, host_address=config.HOST_ADDRESS, auth_cookies=config.MEDIUM_AUTH_COOKIES, template_folder="server/templates")
|
|
|
|
redis_storage = redis.Redis(
|
|
host=config.REDIS_HOST,
|
|
port=config.REDIS_PORT,
|
|
db=0,
|
|
socket_timeout=config.REDIS_TIMEOUT,
|
|
socket_connect_timeout=config.REDIS_TIMEOUT,
|
|
# decode_responses=True
|
|
)
|
|
|
|
url_correlation: ContextVar[Optional[str]] = ContextVar("url_correlation", default="UNKNOWN_URL")
|
|
transponder_code_correlation: ContextVar[Optional[str]] = ContextVar("transponder_code_correlation", default="unknown transponder location... Beep!")
|
|
|
|
ban_db = pickledb.load("ban_post_list.db", True)
|
|
|
|
WORDS_LIST_FILE = "xkcdpass/static/legac"
|
|
|
|
xkcd_passwd = xp.generate_wordlist(wordfile=WORDS_LIST_FILE, min_length=5, max_length=8)
|
|
|
|
maintenance_mode = Value("b", False)
|