Add config file, dropdown for webhooks, and sends stuff to Discord

discord_rss_bot/settings.py

@@ -1,28 +1,44 @@
+import functools
+import logging
 import os
 from pathlib import Path
 
 from platformdirs import user_data_dir
+from reader import make_reader
+from tomlkit import TOMLDocument, comment, document, parse, table
+
+logging.basicConfig(
+    level=logging.DEBUG,
+    format="[%(asctime)s] [%(funcName)s:%(lineno)d] %(message)s",
+)
+logger = logging.getLogger(__name__)
+
+# For get_data_dir()
+data_directory = user_data_dir(appname="discord_rss_bot", appauthor="TheLovinator", roaming=True)
 
 
-def get_app_dir(app_dir: str = user_data_dir("discord_rss_bot")) -> Path:
+def get_data_dir(data_dir: str = data_directory) -> Path:
     """
-    Get the application directory. This is where the database file is stored.
+    Get the data directory. This is where the database file and config file are stored.
 
     Args:
-        app_dir: The application directory, defaults to user_data_dir().
+        data_dir: The application directory, defaults to user_data_dir().
 
     Returns:
         Path: The application directory.
     """
-    print(f"Data directory: {app_dir}")
+    if data_dir != user_data_dir("discord_rss_bot"):
+        logger.info(f"Using custom data directory: {data_dir}")
 
     # Use the environment variable if it exists instead of the default app dir.
-    app_dir = os.getenv("DATABASE_LOCATION") or app_dir
+    data_dir = os.getenv("DATA_DIR") or data_dir
+
+    logger.debug(f"Data directory: {data_dir}")
 
     # Create the data directory if it doesn't exist
-    os.makedirs(app_dir, exist_ok=True)
+    os.makedirs(data_dir, exist_ok=True)
 
-    return Path(app_dir)
+    return Path(data_dir)
 
 
 def get_db_file(custom_db_name: str = "db.sqlite") -> Path:
@@ -34,13 +50,76 @@ def get_db_file(custom_db_name: str = "db.sqlite") -> Path:
     Returns:
         Path: The database file.
     """
+    if custom_db_name != "db.sqlite":
+        logger.info(f"Using custom database file: {custom_db_name}")
+
     # Store the database file in the data directory
-    app_dir = get_app_dir()
+    data_dir = get_data_dir()
+    db_location: Path = Path(os.path.join(data_dir, custom_db_name))
 
     # Use the environment variable if it exists instead of the default db name.
-    db_name = os.getenv("DATABASE_NAME") or custom_db_name
-
-    db_file: Path = Path(os.path.join(app_dir, db_name))
-    print(f"Database file: {db_file}")
+    db_file = os.getenv("DATABASE_LOCATION") or db_location
+    logger.debug(f"Database file: {db_file}")
 
     return Path(db_file)
+
+
+def _create_settings_file(settings_file) -> None:
+    """Create the settings file if it doesn't exist."""
+    logger.debug(f"Settings file: {settings_file}")
+
+    # [webhooks]
+    # Both options are commented out by default.
+    webhooks = table()
+    webhooks.add(comment('"First webhook" = "https://discord.com/api/webhooks/1234567890/abcdefghijklmnopqrstuvwxyz"'))
+    webhooks.add(comment('"Second webhook" = "https://discord.com/api/webhooks/1234567890/abcdefghijklmnopqrstuvwxyz"'))
+
+    # [database]
+    # Option is commented out by default.
+    database = table()
+    database.add(comment('"location" = "/path/to/database/file"'))
+
+    doc = document()
+    doc.add("webhooks", webhooks)
+    doc.add("database", database)
+
+    logger.debug(f"Settings file: {doc}")
+    logger.debug(f"Settings file as TOML: {doc.as_string()}")
+
+    # Write the settings file
+    with open(settings_file, "w") as f:
+        f.write(doc.as_string())
+
+
+def read_settings_file(custom_settings_name: str = "settings.toml") -> TOMLDocument:
+    """Read the settings file
+
+    Args:
+        custom_settings_name: The name of the settings file, defaults to settings.toml.
+
+    Returns:
+        dict: The settings file as a dict.
+    """
+    if custom_settings_name != "settings.toml":
+        logger.info(f"Using custom name for settings file: {custom_settings_name}")
+
+    # Store the database file in the data directory
+    data_dir = get_data_dir()
+    settings_file_location: Path = Path(os.path.join(data_dir, custom_settings_name))
+
+    # Use the environment variable if it exists instead of the default db name.
+    settings_file = os.getenv("SETTINGS_FILE_LOCATION") or settings_file_location
+    logger.debug(f"Settings file: {settings_file}")
+
+    # Create the settings file if it doesn't exist
+    if not os.path.exists(settings_file):
+        _create_settings_file(settings_file)
+
+    with open(settings_file, encoding="utf-8") as f:
+        data = parse(f.read())
+        logger.debug(f"Contents of settings file: {data}")
+
+        return data
+
+
+reader = make_reader(str(get_db_file()))