import os from functools import lru_cache from platformdirs import user_data_dir from reader import Reader, make_reader # type: ignore data_dir: str = user_data_dir(appname="discord_rss_bot", appauthor="TheLovinator", roaming=True) os.makedirs(data_dir, exist_ok=True) print(f"Data is stored in '{data_dir}'.") # TODO: Add default things to the database and make the edible. default_custom_message: str = "{{entry_title}}\n{{entry_link}}" default_custom_embed: dict[str, str] = { "title": "{{entry_title}}", "description": "{{entry_text}}", "image_url": "{{image_1}}", "color": "#469ad9", } @lru_cache() def get_db_location(custom_location: str = "") -> str: """Where we store the database file. Args: custom_location: Where the database file should be stored. This should be with the file name. Returns: The database location. """ # Use the custom location if it is provided. db_loc: str = custom_location or os.path.join(data_dir, "db.sqlite") return db_loc @lru_cache() def get_reader(custom_location: str = "") -> Reader: """Get the reader. Args: custom_location: The location of the database file. """ db_location: str = get_db_location(custom_location) return make_reader(url=db_location) def list_webhooks(reader: Reader) -> list[dict[str, str]]: """ Get current webhooks from the database if they exist otherwise use an empty list. Args: reader: The reader to use. Returns: list[dict[str, str]]: The webhooks. """ webhooks: list[dict[str, str]] = [] # Get global tags if reader.get_tags(()) is not None: for tag in reader.get_tag_keys(()): # Check if the tag is named webhooks if tag == "webhooks": webhooks = reader.get_tag((), "webhooks") # type: ignore return webhooks