add schema to StagePathParts

src/snowflake/cli/_plugins/stage/manager.py

@@ -41,7 +41,7 @@
 from snowflake.cli.api.console import cli_console
 from snowflake.cli.api.constants import PYTHON_3_12
 from snowflake.cli.api.identifiers import FQN
-from snowflake.cli.api.project.util import to_string_literal
+from snowflake.cli.api.project.util import extract_schema, to_string_literal
 from snowflake.cli.api.secure_path import SecurePath
 from snowflake.cli.api.sql_execution import SqlExecutionMixin
 from snowflake.cli.api.stage_path import StagePath
@@ -86,6 +86,10 @@ def path(self) -> str:
     def full_path(self) -> str:
         raise NotImplementedError
 
+    @property
+    def schema(self) -> str | None:
+        raise NotImplementedError
+
     def replace_stage_prefix(self, file_path: str) -> str:
         raise NotImplementedError
 
@@ -145,6 +149,10 @@ def path(self) -> str:
     def full_path(self) -> str:
         return f"{self.stage.rstrip('/')}/{self.directory}"
 
+    @property
+    def schema(self) -> str | None:
+        return extract_schema(self.stage)
+
     def replace_stage_prefix(self, file_path: str) -> str:
         stage = Path(self.stage).parts[0]
         file_path_without_prefix = Path(file_path).parts[OMIT_FIRST]