data-sourcerer 0.1.0__py3-none-any.whl

sourcerer/settings.py

@@ -0,0 +1,72 @@
+"""
+Application settings module.
+
+This module contains configuration settings for the application,
+including encryption keys, UI constants, and file type definitions.
+It centralizes configuration values to make them easily accessible
+and modifiable throughout the application.
+"""
+
+import os
+from pathlib import Path
+
+from platformdirs import user_state_dir
+
+from sourcerer.utils import get_encryption_key
+
+APP_NAME = "sourcerer"
+
+APP_DIR = Path(user_state_dir(APP_NAME))
+os.makedirs(APP_DIR, exist_ok=True)
+
+DB_NAME = "sourcerer.db"
+
+ENCRYPTION_KEY = get_encryption_key(APP_DIR)
+
+# Maximum number of parallel download operations
+MAX_PARALLEL_DOWNLOADS = 8
+
+PATH_DELIMITER = "/"
+
+# UI icons for different file system elements
+DIRECTORY_ICON = "📁"
+FILE_ICON = "📄"
+
+# Action icons
+DOWNLOAD_ICON = "📥"
+UPLOAD_ICON = "📤"
+PREVIEW_ICON = "✨"
+
+# Time threshold for detecting double-clicks (in seconds)
+DOUBLE_CLICK_THRESHOLD = 1.5
+
+# Set of file extensions that are considered text files
+TEXT_EXTENSIONS = {
+    ".txt",
+    ".md",
+    ".json",
+    ".yaml",
+    ".yml",
+    ".csv",
+    ".tsv",
+    ".log",
+    ".ini",
+    ".py",
+    ".js",
+    ".ts",
+    ".html",
+    ".css",
+    ".xml",
+    ".toml",
+    ".cfg",
+    ".sh",
+    ".bat",
+    ".java",
+    ".c",
+    ".cpp",
+    ".go",
+    ".rs",
+}
+
+
+PAGE_SIZE = 100

sourcerer/utils.py

@@ -0,0 +1,32 @@
+# Generate and store encryption key in a file
+import os
+import uuid
+from pathlib import Path
+
+
+def get_encryption_key(path: Path):
+    """
+    Get the encryption key from a file or generate a new one if the file doesn't exist.
+
+    path (Path): The path to the file where the encryption key is stored.
+    Returns:
+        str: The encryption key
+    """
+
+    key_file_path = path / "encryption_key"
+
+    # If key file exists, read the key from it
+    if os.path.exists(key_file_path):
+        with open(key_file_path, "r") as f:
+            return f.read().strip()
+
+    # Otherwise, generate a new key and store it
+
+    # Create directory if it doesn't exist
+    os.makedirs(os.path.dirname(key_file_path), exist_ok=True)
+
+    new_key = str(uuid.uuid4())
+    with open(key_file_path, "w") as f:
+        f.write(new_key)
+
+    return new_key