surface a profile variable that can be set in the customer's entrypoint

Author: joroscoSF

src/datacustomcode/config.yaml

@@ -1,8 +1,12 @@
 reader_config:
   type_config_name: QueryAPIDataCloudReader
+  options:
+    credentials_profile: default
 
 writer_config:
   type_config_name: PrintDataCloudWriter
+  options:
+    credentials_profile: default
 
 spark_config:
   app_name: DC Custom Code Python SDK Testing

src/datacustomcode/credentials.py

@@ -65,11 +65,11 @@ def from_env(cls) -> Credentials:
             ) from exc
 
     @classmethod
-    def from_available(cls) -> Credentials:
+    def from_available(cls, profile: str = "default") -> Credentials:
         if os.environ.get("SFDC_USERNAME"):
             return cls.from_env()
         if os.path.exists(INI_FILE):
-            return cls.from_ini()
+            return cls.from_ini(profile=profile)
         raise ValueError(
             "Credentials not found in env or ini file. "
             "Run `datacustomcode configure` to create a credentials file."

src/datacustomcode/io/reader/query_api.py

@@ -75,9 +75,11 @@ class QueryAPIDataCloudReader(BaseDataCloudReader):
 
     CONFIG_NAME = "QueryAPIDataCloudReader"
 
-    def __init__(self, spark: SparkSession) -> None:
+    def __init__(
+        self, spark: SparkSession, credentials_profile: str = "default"
+    ) -> None:
         self.spark = spark
-        credentials = Credentials.from_available()
+        credentials = Credentials.from_available(profile=credentials_profile)
 
         self._conn = SalesforceCDPConnection(
             credentials.login_url,

src/datacustomcode/io/writer/print.py

@@ -26,10 +26,17 @@ class PrintDataCloudWriter(BaseDataCloudWriter):
     CONFIG_NAME = "PrintDataCloudWriter"
 
     def __init__(
-        self, spark: SparkSession, reader: Optional[QueryAPIDataCloudReader] = None
+        self,
+        spark: SparkSession,
+        reader: Optional[QueryAPIDataCloudReader] = None,
+        credentials_profile: str = "default",
     ) -> None:
         super().__init__(spark)
-        self.reader = QueryAPIDataCloudReader(self.spark) if reader is None else reader
+        self.reader = (
+            QueryAPIDataCloudReader(self.spark, credentials_profile)
+            if reader is None
+            else reader
+        )
 
     def validate_dataframe_columns_against_dlo(
         self,