diff options
Diffstat (limited to '')
| -rw-r--r-- | cogs/utils/settings.py | 67 |
1 files changed, 44 insertions, 23 deletions
diff --git a/cogs/utils/settings.py b/cogs/utils/settings.py index d66ce7f..b1a585c 100644 --- a/cogs/utils/settings.py +++ b/cogs/utils/settings.py | |||
| @@ -3,16 +3,18 @@ | |||
| 3 | # Copyright (C) 2026 Salmonik @ salmonik@tuta.io | 3 | # Copyright (C) 2026 Salmonik @ salmonik@tuta.io |
| 4 | # https://codeberg.org/salmonik/senator-sperma | 4 | # https://codeberg.org/salmonik/senator-sperma |
| 5 | 5 | ||
| 6 | import json | ||
| 7 | import logging | 6 | import logging |
| 8 | import aiofiles | 7 | import sqlite3 |
| 9 | import os | 8 | import os |
| 10 | from dataclasses import dataclass, field | 9 | from dataclasses import dataclass, field |
| 11 | 10 | ||
| 11 | from cogs.utils.database import get_connection, init_database | ||
| 12 | |||
| 12 | logger = logging.getLogger(__name__) | 13 | logger = logging.getLogger(__name__) |
| 13 | 14 | ||
| 14 | settings_file = "data/settings.json" | 15 | DEFAULT_BASEURL = "http://localhost:3939" |
| 15 | DEFAULT_BASEURL = "http://localhost:8080" | 16 | |
| 17 | init_database() | ||
| 16 | 18 | ||
| 17 | 19 | ||
| 18 | @dataclass | 20 | @dataclass |
| @@ -28,24 +30,21 @@ class Settings: | |||
| 28 | } | 30 | } |
| 29 | 31 | ||
| 30 | try: | 32 | try: |
| 31 | if os.path.exists(settings_file): | 33 | with get_connection() as conn: |
| 32 | async with aiofiles.open(settings_file, "r") as f: | 34 | c = conn.cursor() |
| 33 | content = await f.read() | 35 | c.execute("SELECT key, value FROM settings") |
| 34 | if content.strip(): | 36 | rows = c.fetchall() |
| 35 | self._data = json.loads(content) | 37 | |
| 36 | self._sync_from_data() | 38 | if rows: |
| 37 | logger.info("Settings loaded into memory") | 39 | self._data = {row[0]: row[1] for row in rows} |
| 38 | else: | 40 | self._sync_from_data() |
| 39 | self._data = default_settings.copy() | 41 | logger.info("Settings loaded from database") |
| 40 | self._sync_from_data() | ||
| 41 | await self.save() | ||
| 42 | logger.warning("Empty settings.json, reset to defaults") | ||
| 43 | else: | 42 | else: |
| 44 | self._data = default_settings.copy() | 43 | self._data = default_settings.copy() |
| 45 | self._sync_from_data() | 44 | self._sync_from_data() |
| 46 | await self.save() | 45 | await self.save() |
| 47 | logger.info("Created default settings.json") | 46 | logger.info("Created default settings in database") |
| 48 | except json.JSONDecodeError as e: | 47 | except Exception as e: |
| 49 | logger.error(f"Error loading settings: {e}") | 48 | logger.error(f"Error loading settings: {e}") |
| 50 | self._data = default_settings.copy() | 49 | self._data = default_settings.copy() |
| 51 | self._sync_from_data() | 50 | self._sync_from_data() |
| @@ -53,17 +52,27 @@ class Settings: | |||
| 53 | return self | 52 | return self |
| 54 | 53 | ||
| 55 | def _sync_from_data(self): | 54 | def _sync_from_data(self): |
| 56 | self.volume = self._data.get("volume", 1.0) | 55 | try: |
| 56 | self.volume = float(self._data.get("volume", 1.0)) | ||
| 57 | except (ValueError, TypeError): | ||
| 58 | self.volume = 1.0 | ||
| 57 | self.baseurl = self._data.get("baseurl", DEFAULT_BASEURL) | 59 | self.baseurl = self._data.get("baseurl", DEFAULT_BASEURL) |
| 58 | 60 | ||
| 59 | def _sync_to_data(self): | 61 | def _sync_to_data(self): |
| 60 | self._data["volume"] = self.volume | 62 | self._data["volume"] = str(self.volume) |
| 61 | self._data["baseurl"] = self.baseurl | 63 | self._data["baseurl"] = self.baseurl |
| 62 | 64 | ||
| 63 | async def save(self): | 65 | async def save(self): |
| 64 | self._sync_to_data() | 66 | self._sync_to_data() |
| 65 | async with aiofiles.open(settings_file, "w") as f: | 67 | with get_connection() as conn: |
| 66 | await f.write(json.dumps(self._data, indent=2)) | 68 | c = conn.cursor() |
| 69 | c.execute("PRAGMA foreign_keys = ON") | ||
| 70 | for key, value in self._data.items(): | ||
| 71 | c.execute( | ||
| 72 | "INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)", | ||
| 73 | (key, value), | ||
| 74 | ) | ||
| 75 | conn.commit() | ||
| 67 | 76 | ||
| 68 | 77 | ||
| 69 | _settings_instance: Settings | None = None | 78 | _settings_instance: Settings | None = None |
| @@ -90,4 +99,16 @@ def get_volume() -> float: | |||
| 90 | 99 | ||
| 91 | 100 | ||
| 92 | def get_baseurl() -> str: | 101 | def get_baseurl() -> str: |
| 93 | return _settings_instance.baseurl.rstrip("/") if _settings_instance else DEFAULT_BASEURL | 102 | url = os.getenv("COPYPARTY_URL") |
| 103 | if not url: | ||
| 104 | logger.warning("COPYPARTY_URL not set! Using default http://localhost:3939") | ||
| 105 | return DEFAULT_BASEURL | ||
| 106 | return url.rstrip("/") | ||
| 107 | |||
| 108 | |||
| 109 | def get_uploadurl() -> str: | ||
| 110 | url = os.getenv("COPYPARTY_URL_UPLOAD") | ||
| 111 | if not url: | ||
| 112 | logger.warning("COPYPARTY_URL_UPLOAD not set! Using default http://localhost:3939") | ||
| 113 | return DEFAULT_BASEURL | ||
| 114 | return url.rstrip("/") | ||
