salesforce-data-customcode 6.0.4.dev2__py3-none-any.whl → 6.0.4.dev3__py3-none-any.whl

datacustomcode/client.py

@@ -289,8 +289,38 @@ class Client:
         return self._writer.write_to_dmo(name, dataframe, write_mode, **kwargs)  # type: ignore[no-any-return]
 
     def find_file_path(self, file_name: str) -> Path:
-        """Return a file path"""
+        """Resolve a bundled file shipped in the package to an absolute path.
+
+        Resolution order (first existing path wins):
+
+        1. ``$LIBRARY_PATH/<file_folder>/<file_name>`` then
+           ``$LIBRARY_PATH/<file_name>`` — when the ``LIBRARY_PATH`` environment
+           variable is set. The Data Cloud runtime sets this to the directory
+           containing the extracted package.
+        2. ``<code_package>/<file_folder>/<file_name>`` relative to the current
+           working directory — the default ``payload/files/<file_name>`` layout
+           used by ``datacustomcode run`` from a project root.
+        3. ``<config_dir>/<file_folder>/<file_name>`` where ``<config_dir>`` is
+           the directory containing the nearest ``config.json`` discoverable
+           by walking the cwd subtree.
+
+        ``LIBRARY_PATH`` must point to the directory that *contains*
+        ``files/`` — i.e., the package root, the same directory that holds
+        ``config.json`` and ``entrypoint.py``. See ``Bundled file resolution``
+        in the ``readme.md`` for more details.
 
+        Args:
+            file_name: A file under the package's ``files/`` folder. Relative
+                subpaths (e.g., ``"file/data2.csv"``) are supported.
+
+        Returns:
+            A ``pathlib.Path`` that exists.
+
+        Raises:
+            FileNotFoundError: If the file does not exist at any of the
+                resolution-order locations. The message lists every candidate
+                path that was tried.
+        """
         return self._file.find_file_path(file_name)  # type: ignore[no-any-return]
 
     def llm_gateway_generate_text(

datacustomcode/file/path/default.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 
 import os
 from pathlib import Path
-from typing import Optional
+from typing import Iterator, Optional
 
 from datacustomcode.file.base import BaseDataAccessLayer
 
@@ -66,7 +66,7 @@ class DefaultFindFilePath(BaseDataAccessLayer):
             file_name: The name of the file to open
 
         Returns:
-            A file path
+            A file path that exists
 
         Raises:
             FileNotFoundError: If the file cannot be found
@@ -74,46 +74,40 @@ class DefaultFindFilePath(BaseDataAccessLayer):
         if not file_name:
             raise ValueError("file_name cannot be empty")
 
-        file_path = self._resolve_file_path(file_name)
+        tried: list[Path] = []
+        for candidate in self._candidate_paths(file_name):
+            tried.append(candidate)
+            if candidate.exists():
+                return candidate
 
-        if not file_path.exists():
-            raise FileNotFoundError(
-                f"File '{file_name}' not found in any search location"
-            )
+        raise FileNotFoundError(
+            f"File '{file_name}' not found in any search location. "
+            f"Tried: {[str(p) for p in tried]}"
+        )
 
-        return file_path
-
-    def _resolve_file_path(self, file_name: str) -> Path:
-        """Resolve the full path to a file.
+    def _candidate_paths(self, file_name: str) -> Iterator[Path]:
+        """Yield candidate paths for ``file_name`` in resolution order.
 
         Args:
             file_name: The name of the file to resolve
 
         Returns:
-            The full path to the file
+            An iterator of candidate paths
         """
-        # First check if environment variable is set
+        # 1. $LIBRARY_PATH/<file_folder>/<file_name>, then $LIBRARY_PATH/<file_name>
         env_path = os.getenv(self.DEFAULT_ENV_VAR)
         if env_path:
-            file_path = Path(env_path) / file_name
-            if file_path.exists():
-                return file_path
+            yield Path(env_path) / self.file_folder / file_name
+            yield Path(env_path) / file_name
 
-        # First try the default code package location
+        # 2. <code_package>/<file_folder>/<file_name> relative to cwd
         if self._code_package_exists():
-            file_path = self._get_code_package_file_path(file_name)
-            if file_path.exists():
-                return file_path
+            yield self._get_code_package_file_path(file_name)
 
-        # Fall back to config.json-based location
+        # 3. <config_dir>/<file_folder>/<file_name> via config.json discovery
         config_path = self._find_config_file()
-        if config_path:
-            file_path = self._get_config_based_file_path(file_name, config_path)
-            if file_path.exists():
-                return file_path
-
-        # Return the file name as a Path if not found in any location
-        return Path(file_name)
+        if config_path is not None:
+            yield self._get_config_based_file_path(file_name, config_path)
 
     def _code_package_exists(self) -> bool:
         """Check if the default code package directory exists.
@@ -146,6 +140,10 @@ class DefaultFindFilePath(BaseDataAccessLayer):
     def _get_config_based_file_path(self, file_name: str, config_path: Path) -> Path:
         """Get the file path relative to the config file location.
 
+        Anchors on the directory containing the discovered ``config.json`` so a
+        package found by walking up from cwd resolves files relative to its own
+        root, not the caller's cwd.
+
         Args:
             file_name: The name of the file
             config_path: The path to the config file
@@ -153,8 +151,7 @@ class DefaultFindFilePath(BaseDataAccessLayer):
         Returns:
             The full path to the file
         """
-        relative_path = f"{self.file_folder}/{file_name}"
-        return Path(relative_path)
+        return config_path.parent / self.file_folder / file_name
 
     def _find_file_in_tree(self, filename: str, search_path: Path) -> Optional[Path]:
         """Find a file within a directory tree.

{salesforce_data_customcode-6.0.4.dev2.dist-info → salesforce_data_customcode-6.0.4.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: salesforce-data-customcode
-Version: 6.0.4.dev2
+Version: 6.0.4.dev3
 Summary: Data Cloud Custom Code SDK
 License-Expression: Apache-2.0
 License-File: LICENSE.txt
@@ -15,7 +15,6 @@ Requires-Dist: click (>=8.1.8,<9.0.0)
 Requires-Dist: loguru (>=0.7.3,<0.8.0)
 Requires-Dist: numpy
 Requires-Dist: pandas
-Requires-Dist: pipreqs
 Requires-Dist: pydantic (==2.13.1)
 Requires-Dist: pyspark (==3.5.1)
 Requires-Dist: pyyaml (>=6.0,<7.0)
@@ -150,8 +149,8 @@ The SDK automatically handles all dependency packaging for Data Cloud deployment
 ├── payload
 │   ├── config.json
 │   ├── entrypoint.py
-├── files
-│   ├── data.csv
+│   ├── files
+│   │   ├── data.csv
 ```
 
 ## py-files directory
@@ -163,10 +162,10 @@ Your Python dependencies can be packaged as .py files, .zip archives (containing
 ├── payload
 │   ├── config.json
 │   ├── entrypoint.py
-├── py-files
-│   ├── moduleA
-│   │   ├── __init__.py
-│   │   ├── moduleA.py
+│   ├── py-files
+│   │   ├── moduleA
+│   │   │   ├── __init__.py
+│   │   │   ├── moduleA.py
 ```
 
 ## API
@@ -174,7 +173,7 @@ Your Python dependencies can be packaged as .py files, .zip archives (containing
 Your entry point script will define logic using the `Client` object which wraps data access layers.
 
 You should only need the following methods:
-* `find_file_path(file_name)` - Returns a file path
+* `find_file_path(file_name)` – Resolve a bundled file (placed under `payload/files/`) to a `pathlib.Path` that exists. Works the same locally and inside Data Cloud — see [Bundled file resolution](#bundled-file-resolution) below for the full lookup order. Raises `FileNotFoundError` if the file isn't found.
 * `read_dlo(name)` – Read from a Data Lake Object by name
 * `read_dmo(name)` – Read from a Data Model Object by name
 * `write_to_dlo(name, spark_dataframe, write_mode)` – Write to a Data Model Object by name with a Spark dataframe
@@ -195,6 +194,24 @@ client.write_to_dlo('output_DLO')
 > [!WARNING]
 > Currently we only support reading from DMOs and writing to DMOs or reading from DLOs and writing to DLOs, but they cannot mix.
 
+### Bundled file resolution
+
+Place bundled files (CSVs, prompt files, etc.) under `payload/files/`. The same `client.find_file_path("data.csv")` call resolves consistently across all three runtimes:
+
+- `datacustomcode run` (local) → `<cwd>/payload/files/data.csv`
+- Data Cloud script package → `$LIBRARY_PATH/files/data.csv`
+- Data Cloud function package → `$LIBRARY_PATH/files/data.csv`
+
+Resolution order (first existing path wins):
+
+1. `$LIBRARY_PATH/files/<file_name>`, then `$LIBRARY_PATH/<file_name>` — when `LIBRARY_PATH` is set. Data Cloud sets this for you to the package root.
+2. `payload/files/<file_name>` relative to the current working directory.
+3. `<config_dir>/files/<file_name>` where `<config_dir>` is the directory of the nearest `config.json` discoverable by walking down from cwd.
+
+If none of these exist, `find_file_path` raises `FileNotFoundError` with the list of paths it tried.
+
+`$LIBRARY_PATH` is set automatically to the root of the package at runtime inside Data Cloud.
+
 
 ## CLI
 

{salesforce_data_customcode-6.0.4.dev2.dist-info → salesforce_data_customcode-6.0.4.dev3.dist-info}/RECORD

@@ -1,7 +1,7 @@
 datacustomcode/__init__.py,sha256=qx-X3U9lU_791ZbAxhGqQeyDw6eqTi9-E9AbYnwJfxU,2071
 datacustomcode/auth.py,sha256=fpSjhIBdv9trC8yq2vuljAix_Euu-4Ah7HDCGhYjOxI,8309
 datacustomcode/cli.py,sha256=i7hPoQqJskTmx3Odz2wVgkxBnf_PFaIawI12FXhaRUE,13264
-datacustomcode/client.py,sha256=eV_Bv-YfxXcayjgtAbu9tXb_UcHgcE2f3-L_3vvc0cI,13616
+datacustomcode/client.py,sha256=fc3JmDbAFcTT-yP_k_0jNiGqNp_KwZaIpEsI-NPxNcw,15115
 datacustomcode/cmd.py,sha256=ZMs46aydJw2EaU26JgCtZmnqESQFHvvaJz10hnjZTBk,3537
 datacustomcode/common_config.py,sha256=SAUnxj3kqmOeWwPmFoYq4tuxokMgURVm4QwcGi-avL4,1928
 datacustomcode/config.py,sha256=2Pk61ieQsEWSxKDxlS66_rliKE8iX0j-9LweEm0aaGo,4072
@@ -19,7 +19,7 @@ datacustomcode/einstein_predictions_config.py,sha256=oxKgp2u7t5SdHVsasBBxsvd3KAf
 datacustomcode/file/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
 datacustomcode/file/base.py,sha256=2IUlQufPvpHccgyoeZlKhcktO4pS4NKdha03Whq33Zk,745
 datacustomcode/file/path/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
-datacustomcode/file/path/default.py,sha256=jyDv61JUlD5H6lUPAp55_B2toMryKItA8Hu2kKr3k_o,5529
+datacustomcode/file/path/default.py,sha256=oG5S3JMnJ_c1ewUx5kWfEmJnMVXsu6AnwsBxlriw6Kk,5705
 datacustomcode/function/__init__.py,sha256=oag8FevDxbx1vR0tFhtxPd5SvmUEYDj8esRo4v2qu3Q,749
 datacustomcode/function/base.py,sha256=H_i80GcsFScvHJJ6bYLry5phTpsEU1KubpR8D3qnsjo,691
 datacustomcode/function/feature_types/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
@@ -92,8 +92,8 @@ datacustomcode/templates/script/requirements-dev.txt,sha256=OWwuy1awesqOZOS3Zizm
 datacustomcode/templates/script/requirements.txt,sha256=qJbqzs5z0Hrx590U3T7dHq_m8UQEx2TdhgrJSz-QHIQ,40
 datacustomcode/token_provider.py,sha256=kGPxdxGZEr6VFknW7jFb9P6wOvRaRo-sfdBN2cUjEZY,6792
 datacustomcode/version.py,sha256=9LlbVrzwBvut1L308QbwNBgxdQk5CrghH39JARVSxms,989
-salesforce_data_customcode-6.0.4.dev2.dist-info/METADATA,sha256=y3E6K4mTg2QEdSDQhYB1uN1Vmob74w_OFnWUhv9hkLw,21953
-salesforce_data_customcode-6.0.4.dev2.dist-info/WHEEL,sha256=eY7nduwzv-ldUxpzbRlxwvC693Hg6PX8bWDjEHjZ_dk,88
-salesforce_data_customcode-6.0.4.dev2.dist-info/entry_points.txt,sha256=WpQ94UB7UuRCYGOLtJV3vAgm1tl7iVf43VYRWIuNu5Y,74
-salesforce_data_customcode-6.0.4.dev2.dist-info/licenses/LICENSE.txt,sha256=iOi8EmQpfkFhMENi7VYtkl2EqS14pEOccoXHiW2dyPU,11443
-salesforce_data_customcode-6.0.4.dev2.dist-info/RECORD,,
+salesforce_data_customcode-6.0.4.dev3.dist-info/METADATA,sha256=Tg_qVJWkMhhj1pz1hSHwVpLh00hppX1cDMkmts7ycJM,23234
+salesforce_data_customcode-6.0.4.dev3.dist-info/WHEEL,sha256=eY7nduwzv-ldUxpzbRlxwvC693Hg6PX8bWDjEHjZ_dk,88
+salesforce_data_customcode-6.0.4.dev3.dist-info/entry_points.txt,sha256=WpQ94UB7UuRCYGOLtJV3vAgm1tl7iVf43VYRWIuNu5Y,74
+salesforce_data_customcode-6.0.4.dev3.dist-info/licenses/LICENSE.txt,sha256=iOi8EmQpfkFhMENi7VYtkl2EqS14pEOccoXHiW2dyPU,11443
+salesforce_data_customcode-6.0.4.dev3.dist-info/RECORD,,