comfygit-core 0.2.0__py3-none-any.whl

comfygit_core/models/workspace_config.py

@@ -0,0 +1,71 @@
+from dataclasses import dataclass
+
+@dataclass(repr=False)
+class APICredentials:
+    """Secure storage for external API credentials."""
+    civitai_token: str | None = None
+
+    @classmethod
+    def from_dict(cls, data):
+        if not data:
+            return None
+        return cls(civitai_token=data.get("civitai_token"))
+
+    def to_dict(self):
+        return {"civitai_token": self.civitai_token} if self.civitai_token else {}
+
+    def __repr__(self):
+        """Obfuscate token in logs."""
+        if self.civitai_token:
+            return f"APICredentials(civitai_token='***{self.civitai_token[-4:]}')"
+        return "APICredentials(civitai_token=None)"
+
+@dataclass
+class ModelDirectory:
+    path: str
+    added_at: str
+    last_sync: str
+
+    @classmethod
+    def from_dict(cls, data):
+        return cls(
+            path=data["path"],
+            added_at=data["added_at"],
+            last_sync=data["last_sync"],
+        )
+
+    @classmethod
+    def to_dict(cls, instance):
+        return instance.__dict__
+
+@dataclass
+class WorkspaceConfig:
+    version: int
+    active_environment: str
+    created_at: str
+    global_model_directory: ModelDirectory | None
+    api_credentials: APICredentials | None = None
+    prefer_registry_cache: bool = True  # Use local node mappings cache instead of API
+
+    @classmethod
+    def from_dict(cls, data):
+        return cls(
+            version=data["version"],
+            active_environment=data["active_environment"],
+            created_at=data["created_at"],
+            global_model_directory=ModelDirectory.from_dict(data["global_model_directory"]) if data.get("global_model_directory") else None,
+            api_credentials=APICredentials.from_dict(data.get("api_credentials")) if data.get("api_credentials") else None,
+            prefer_registry_cache=data.get("prefer_registry_cache", True),
+        )
+
+    @classmethod
+    def to_dict(cls, instance):
+        result = {
+            "version": instance.version,
+            "active_environment": instance.active_environment,
+            "created_at": instance.created_at,
+            "global_model_directory": ModelDirectory.to_dict(instance.global_model_directory) if instance.global_model_directory else None,
+            "api_credentials": instance.api_credentials.to_dict() if instance.api_credentials else None,
+            "prefer_registry_cache": instance.prefer_registry_cache,
+        }
+        return result

comfygit_core/repositories/migrate_paths.py

@@ -0,0 +1,49 @@
+"""One-time migration to normalize path separators in existing databases."""
+from pathlib import Path
+from .model_repository import ModelRepository
+
+
+def migrate_path_separators(db_path: Path) -> int:
+    """Normalize all path separators in model_locations to forward slashes.
+
+    Args:
+        db_path: Path to models.db database
+
+    Returns:
+        Number of paths updated
+    """
+    repo = ModelRepository(db_path)
+
+    # Get all locations - we'll filter in Python to avoid SQL escaping issues
+    query = """
+    SELECT model_hash, base_directory, relative_path, filename, mtime
+    FROM model_locations
+    """
+
+    all_results = repo.sqlite.execute_query(query)
+
+    # Filter for paths with backslashes
+    results = [r for r in all_results if '\\' in r['relative_path']]
+
+    if not results:
+        return 0
+
+    # Update each path
+    update_query = """
+    UPDATE model_locations
+    SET relative_path = ?
+    WHERE model_hash = ? AND base_directory = ? AND relative_path = ?
+    """
+
+    count = 0
+    for row in results:
+        old_path = row['relative_path']
+        new_path = old_path.replace('\\', '/')
+
+        repo.sqlite.execute_write(
+            update_query,
+            (new_path, row['model_hash'], row['base_directory'], old_path)
+        )
+        count += 1
+
+    return count