danbooru-bot/config.py
mctaylors 35ea45d9c7 fix: set encoding for _c.read
Signed-off-by: mctaylors <cantsendmails@mctaylors.ru>
2025-05-18 10:12:35 +03:00

43 lines
1 KiB
Python

import configparser
from typing import Optional
import requests
_c = configparser.ConfigParser()
_c.read("config.ini", "utf-8")
class _General:
token: str
sourceurl: Optional[str]
def __init__(self):
self.token = _c.get('General', 'Token')
self.sourceurl = _c.get('General', 'SourceUrl')
if not self.sourceurl.startswith("http://") and not self.sourceurl.startswith("https://"):
self.sourceurl = None
class _App:
name: str
host: str
hostname: str
def __init__(self):
self.name = _c.get('App', 'Name')
self.host = _c.get('App', 'Host')
# noinspection PyBroadException
try:
response = requests.get(f"http://{self.host}/status.json")
if response.status_code != 200:
raise
except:
print("Unable validate App.Host in config.ini.")
exit(1)
self.hostname = _c.get('App', 'HostName')
if not self.hostname:
self.hostname = self.host
general = _General()
app = _App()