forked from sass/tipibot
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
BOT_PROFILE = os.getenv("BOT_PROFILE", "dev").strip().lower() or "dev"
|
|
if BOT_PROFILE not in {"dev", "economy"}:
|
|
raise SystemExit("BOT_PROFILE must be either 'dev' or 'economy'.")
|
|
|
|
|
|
def _env_int(name: str, default: int) -> int:
|
|
raw = os.getenv(name)
|
|
if raw is None or not raw.strip():
|
|
return default
|
|
return int(raw)
|
|
|
|
|
|
_LEGACY_DISCORD_TOKEN = os.getenv("DISCORD_TOKEN", "")
|
|
DISCORD_TOKEN_DEV = os.getenv("DISCORD_TOKEN_DEV", "")
|
|
DISCORD_TOKEN_ECONOMY = os.getenv("DISCORD_TOKEN_ECONOMY", "")
|
|
DISCORD_TOKEN = (
|
|
DISCORD_TOKEN_ECONOMY if BOT_PROFILE == "economy" else DISCORD_TOKEN_DEV
|
|
) or _LEGACY_DISCORD_TOKEN
|
|
|
|
SHEET_ID = os.getenv("SHEET_ID")
|
|
GOOGLE_CREDS_PATH = os.getenv("GOOGLE_CREDS_PATH", "credentials.json")
|
|
|
|
_LEGACY_GUILD_ID = _env_int("GUILD_ID", 0)
|
|
GUILD_ID_DEV = _env_int("GUILD_ID_DEV", _LEGACY_GUILD_ID)
|
|
GUILD_ID_ECONOMY = _env_int("GUILD_ID_ECONOMY", _LEGACY_GUILD_ID)
|
|
GUILD_ID = GUILD_ID_ECONOMY if BOT_PROFILE == "economy" else GUILD_ID_DEV
|
|
|
|
_LEGACY_BIRTHDAY_CHANNEL_ID = _env_int("BIRTHDAY_CHANNEL_ID", 0)
|
|
BIRTHDAY_CHANNEL_ID_DEV = _env_int("BIRTHDAY_CHANNEL_ID_DEV", _LEGACY_BIRTHDAY_CHANNEL_ID)
|
|
BIRTHDAY_CHANNEL_ID_ECONOMY = _env_int("BIRTHDAY_CHANNEL_ID_ECONOMY", 0)
|
|
BIRTHDAY_CHANNEL_ID = (
|
|
BIRTHDAY_CHANNEL_ID_ECONOMY
|
|
if BOT_PROFILE == "economy"
|
|
else BIRTHDAY_CHANNEL_ID_DEV
|
|
)
|
|
|
|
BIRTHDAY_WINDOW_DAYS = int(os.getenv("BIRTHDAY_WINDOW_DAYS", "7"))
|
|
BASE_ROLE_IDS: list[int] = [1478304631930228779, 1478302278862766190]
|
|
|
|
PB_URL = os.getenv("PB_URL", "http://127.0.0.1:8090")
|
|
PB_ADMIN_EMAIL = os.getenv("PB_ADMIN_EMAIL", "")
|
|
PB_ADMIN_PASSWORD = os.getenv("PB_ADMIN_PASSWORD", "")
|
|
|
|
_LEGACY_PB_COLLECTION = os.getenv("PB_ECONOMY_COLLECTION", "").strip()
|
|
PB_ECONOMY_COLLECTION_DEV = (
|
|
os.getenv("PB_ECONOMY_COLLECTION_DEV", "").strip()
|
|
or (_LEGACY_PB_COLLECTION if _LEGACY_PB_COLLECTION else "economy_users_dev")
|
|
)
|
|
PB_ECONOMY_COLLECTION_ECONOMY = (
|
|
os.getenv("PB_ECONOMY_COLLECTION_ECONOMY", "").strip()
|
|
or (_LEGACY_PB_COLLECTION if _LEGACY_PB_COLLECTION else "economy_users_prod")
|
|
)
|
|
PB_ECONOMY_COLLECTION = (
|
|
PB_ECONOMY_COLLECTION_ECONOMY if BOT_PROFILE == "economy" else PB_ECONOMY_COLLECTION_DEV
|
|
)
|
|
|
|
|
|
def _parse_admin_roles() -> dict[int, int]:
|
|
"""Parse BOT_ADMIN_ROLES env var (format: guild_id:role_id,guild_id:role_id)."""
|
|
raw = os.getenv("BOT_ADMIN_ROLES", "").strip()
|
|
if not raw:
|
|
return {}
|
|
result: dict[int, int] = {}
|
|
for pair in raw.split(","):
|
|
pair = pair.strip()
|
|
if not pair:
|
|
continue
|
|
parts = pair.split(":")
|
|
if len(parts) != 2:
|
|
continue
|
|
try:
|
|
result[int(parts[0].strip())] = int(parts[1].strip())
|
|
except ValueError:
|
|
continue
|
|
return result
|
|
|
|
|
|
BOT_ADMIN_ROLES: dict[int, int] = _parse_admin_roles()
|