44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import configparser
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import typer
|
|
|
|
|
|
class Settings:
|
|
APP_NAME: str = "discord-rss-bot"
|
|
|
|
app_dir = typer.get_app_dir(APP_NAME)
|
|
|
|
# Create the data directory if it doesn't exist
|
|
os.makedirs(app_dir, exist_ok=True)
|
|
|
|
# Store the database file in the data directory
|
|
db_file: Path = Path(os.path.join(app_dir, "db.sqlite"))
|
|
|
|
# Store the config in the data directory
|
|
config_location: Path = Path(os.path.join(app_dir, "config.conf"))
|
|
|
|
if not os.path.isfile(config_location):
|
|
# TODO: Add config for db_file and config_location
|
|
print("No config file found, creating one...")
|
|
with open(config_location, "w") as config_file:
|
|
config = configparser.ConfigParser()
|
|
config.add_section("config")
|
|
config.set(
|
|
"config",
|
|
"webhook_url",
|
|
"https://discord.com/api/webhooks/1234/567890/ABCDEFGHIJKLMNOPQRSTUVWXYZ_abcdefghijklmnopqrstuvwxyz",
|
|
)
|
|
|
|
config.write(config_file)
|
|
sys.exit(f"Please edit the config file at {config_location}")
|
|
|
|
# Read the config file
|
|
config = configparser.ConfigParser()
|
|
config.read(config_location)
|
|
|
|
# Get the webhook url from the config file
|
|
webhook_url = config.get("config", "webhook_url")
|