snowlib 0.4.2__tar.gz → 0.4.4__tar.gz

{snowlib-0.4.2 → snowlib-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowlib
-Version: 0.4.2
+Version: 0.4.4
 Summary: Python-Snowflake utilities
 Author: Rashad Aziz
 Maintainer: Rashad Aziz

{snowlib-0.4.2 → snowlib-0.4.4}/src/snowlib/__init__.py

@@ -43,7 +43,7 @@ from snowlib.models import (
 # Session: Context-bound convenience layer
 from snowlib.session import Session, create_session
 
-__version__ = "0.4.2"
+__version__ = "0.4.4"
 __all__ = [
     # Layer 1: Configuration & Connection
     "load_profile", 

{snowlib-0.4.2 → snowlib-0.4.4}/src/snowlib/models/schema.py

@@ -23,6 +23,58 @@ class Schema(Container):
         super().__init__(context)
         self._fqn = FQN.from_parts(database, name)
     
+    @classmethod
+    def from_name(
+        cls,
+        name: str,
+        context: SnowflakeContext,
+        default_database: str | None = None,
+        default_schema: str | None = None,
+        **kwargs: object
+    ) -> 'Schema':
+        """Parse qualified name and resolve missing parts from context.
+        
+        Args:
+            name: Schema name in format 'DATABASE.SCHEMA' or 'SCHEMA'
+            context: SnowflakeContext to use for resolution
+            default_database: Ignored (Schema doesn't use default_database)
+            default_schema: Ignored (Schema doesn't use default_schema)
+            **kwargs: Additional arguments (ignored)
+        
+        Returns:
+            Schema instance
+        
+        Raises:
+            ValueError: If name cannot be resolved
+        
+        Examples:
+            >>> Schema.from_name("MY_DB.MY_SCHEMA", ctx)
+            >>> Schema.from_name("MY_SCHEMA", ctx)  # Uses current database
+        """
+        parts = name.split(".")
+        
+        if len(parts) == 2:
+            # Fully qualified: DATABASE.SCHEMA
+            return cls(parts[0], parts[1], context)
+        
+        elif len(parts) == 1:
+            # Need database from context: SCHEMA
+            db = context.current_database
+            if not db:
+                msg = (
+                    f"Cannot resolve '{name}': no database in context. "
+                    "Use DATABASE.SCHEMA format or USE DATABASE first."
+                )
+                raise ValueError(msg)
+            return cls(db, parts[0], context)
+        
+        else:
+            msg = (
+                f"Invalid schema name '{name}'. "
+                "Expected format: 'DATABASE.SCHEMA' or 'SCHEMA'."
+            )
+            raise ValueError(msg)
+    
     @property
     def database(self) -> 'Database':
         """Get parent database object"""

{snowlib-0.4.2 → snowlib-0.4.4}/src/snowlib/models/stage.py

@@ -127,10 +127,29 @@ class Stage(SchemaChild):
         auto_compress: bool = True,
         overwrite: bool = False,
         show_progress: bool = True,
+        prefix: Optional[str] = None,
     ) -> list[dict[str, Any]]:
-        """Upload local files to the stage with progress bar"""
+        """Upload local files to the stage with progress bar
+        
+        Args:
+            files: List of local file paths to upload
+            auto_compress: Whether to automatically compress files during upload
+            overwrite: Whether to overwrite existing files in the stage
+            show_progress: Whether to show progress bar (requires tqdm)
+            prefix: Optional subdirectory path within the stage (e.g., 'courses', 'data/2024')
+        
+        Returns:
+            List of dicts with upload results for each file
+        """
         results = []
         
+        # Build target path with optional prefix
+        target_path = self.stage_path
+        if prefix:
+            # Normalize prefix (strip leading/trailing slashes)
+            prefix = prefix.strip('/')
+            target_path = f"{self.stage_path}/{prefix}"
+        
         iterator = tqdm(files, desc="Uploading") if (show_progress and HAS_TQDM and tqdm is not None) else files
         
         for file_path in iterator:
@@ -142,7 +161,7 @@ class Stage(SchemaChild):
                 })
                 continue
             
-            sql = f"PUT 'file://{file_path.as_posix()}' {self.stage_path}"
+            sql = f"PUT 'file://{file_path.as_posix()}' {target_path}"
             sql += f" AUTO_COMPRESS = {str(auto_compress).upper()}"
             sql += f" OVERWRITE = {str(overwrite).upper()}"
             

{snowlib-0.4.2 → snowlib-0.4.4}/src/snowlib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowlib
-Version: 0.4.2
+Version: 0.4.4
 Summary: Python-Snowflake utilities
 Author: Rashad Aziz
 Maintainer: Rashad Aziz