Merge branch 'main' into feat-374142081-add-source-column-match

Author: chalmerlowe

google/cloud/bigquery/_job_helpers.py

@@ -560,6 +560,7 @@ def do_query():
             num_dml_affected_rows=query_results.num_dml_affected_rows,
             query=query,
             total_bytes_processed=query_results.total_bytes_processed,
+            slot_millis=query_results.slot_millis,
         )
 
     if job_retry is not None:

google/cloud/bigquery/client.py

@@ -4144,6 +4144,7 @@ def _list_rows_from_query_results(
         num_dml_affected_rows: Optional[int] = None,
         query: Optional[str] = None,
         total_bytes_processed: Optional[int] = None,
+        slot_millis: Optional[int] = None,
     ) -> RowIterator:
         """List the rows of a completed query.
         See
@@ -4195,6 +4196,8 @@ def _list_rows_from_query_results(
                 The query text used.
             total_bytes_processed (Optional[int]):
                 total bytes processed from job statistics, if present.
+            slot_millis (Optional[int]):
+                Number of slot ms the user is actually billed for.
 
         Returns:
             google.cloud.bigquery.table.RowIterator:
@@ -4234,6 +4237,7 @@ def _list_rows_from_query_results(
             num_dml_affected_rows=num_dml_affected_rows,
             query=query,
             total_bytes_processed=total_bytes_processed,
+            slot_millis=slot_millis,
         )
         return row_iterator
 

google/cloud/bigquery/external_config.py

@@ -882,6 +882,37 @@ def schema(self, value):
             prop = {"fields": [field.to_api_repr() for field in value]}
         self._properties["schema"] = prop
 
+    @property
+    def date_format(self) -> Optional[str]:
+        """Optional[str]: Format used to parse DATE values. Supports C-style and SQL-style values.
+
+        See:
+        https://cloud.google.com/bigquery/docs/reference/rest/v2/tables#ExternalDataConfiguration.FIELDS.date_format
+        """
+        result = self._properties.get("dateFormat")
+        return typing.cast(str, result)
+
+    @date_format.setter
+    def date_format(self, value: Optional[str]):
+        self._properties["dateFormat"] = value
+
+    @property
+    def time_zone(self) -> Optional[str]:
+        """Optional[str]: Time zone used when parsing timestamp values that do not
+        have specific time zone information (e.g. 2024-04-20 12:34:56). The expected
+        format is an IANA timezone string (e.g. America/Los_Angeles).
+
+        See:
+        https://cloud.google.com/bigquery/docs/reference/rest/v2/tables#ExternalDataConfiguration.FIELDS.time_zone
+        """
+
+        result = self._properties.get("timeZone")
+        return typing.cast(str, result)
+
+    @time_zone.setter
+    def time_zone(self, value: Optional[str]):
+        self._properties["timeZone"] = value
+
     @property
     def connection_id(self):
         """Optional[str]: [Experimental] ID of a BigQuery Connection API

google/cloud/bigquery/job/load.py

@@ -582,6 +582,33 @@ def source_column_match(self, value: Optional[SourceColumnMatch]):
             )
         self._set_sub_prop("sourceColumnMatch", value.value if value else None)
 
+    @property
+    def date_format(self) -> Optional[str]:
+        """Optional[str]: Date format used for parsing DATE values.
+
+        See:
+        https://cloud.google.com/bigquery/docs/reference/rest/v2/Job#JobConfigurationLoad.FIELDS.date_format
+        """
+        return self._get_sub_prop("dateFormat")
+
+    @date_format.setter
+    def date_format(self, value: Optional[str]):
+        self._set_sub_prop("dateFormat", value)
+
+    @property
+    def time_zone(self) -> Optional[str]:
+        """Optional[str]: Default time zone that will apply when parsing timestamp
+        values that have no specific time zone.
+
+        See:
+        https://cloud.google.com/bigquery/docs/reference/rest/v2/Job#JobConfigurationLoad.FIELDS.time_zone
+        """
+        return self._get_sub_prop("timeZone")
+
+    @time_zone.setter
+    def time_zone(self, value: Optional[str]):
+        self._set_sub_prop("timeZone", value)
+
     @property
     def time_partitioning(self):
         """Optional[google.cloud.bigquery.table.TimePartitioning]: Specifies time-based
@@ -930,6 +957,20 @@ def source_column_match(self):
         """
         return self.configuration.source_column_match
 
+    @property
+    def date_format(self):
+        """See
+        :attr:`google.cloud.bigquery.job.LoadJobConfig.date_format`.
+        """
+        return self.configuration.date_format
+
+    @property
+    def time_zone(self):
+        """See
+        :attr:`google.cloud.bigquery.job.LoadJobConfig.time_zone`.
+        """
+        return self.configuration.time_zone
+
     @property
     def schema_update_options(self):
         """See

google/cloud/bigquery/job/query.py

@@ -1766,6 +1766,7 @@ def is_job_done():
             num_dml_affected_rows=self._query_results.num_dml_affected_rows,
             query=self.query,
             total_bytes_processed=self.total_bytes_processed,
+            slot_millis=self.slot_millis,
             **list_rows_kwargs,
         )
         rows._preserve_order = _contains_order_by(self.query)

google/cloud/bigquery/query.py

@@ -1282,6 +1282,20 @@ def total_bytes_processed(self):
         if total_bytes_processed is not None:
             return int(total_bytes_processed)
 
+    @property
+    def slot_millis(self):
+        """Total number of slot ms the user is actually billed for.
+
+        See:
+        https://cloud.google.com/bigquery/docs/reference/rest/v2/jobs/query#body.QueryResponse.FIELDS.slot_millis
+
+        Returns:
+            Optional[int]: Count generated on the server (None until set by the server).
+        """
+        slot_millis = self._properties.get("totalSlotMs")
+        if slot_millis is not None:
+            return int(slot_millis)
+
     @property
     def num_dml_affected_rows(self):
         """Total number of rows affected by a DML query.

google/cloud/bigquery/table.py

@@ -1812,6 +1812,7 @@ def __init__(
         num_dml_affected_rows: Optional[int] = None,
         query: Optional[str] = None,
         total_bytes_processed: Optional[int] = None,
+        slot_millis: Optional[int] = None,
     ):
         super(RowIterator, self).__init__(
             client,
@@ -1841,6 +1842,7 @@ def __init__(
         self._num_dml_affected_rows = num_dml_affected_rows
         self._query = query
         self._total_bytes_processed = total_bytes_processed
+        self._slot_millis = slot_millis
 
     @property
     def _billing_project(self) -> Optional[str]:
@@ -1898,6 +1900,11 @@ def total_bytes_processed(self) -> Optional[int]:
         """total bytes processed from job statistics, if present."""
         return self._total_bytes_processed
 
+    @property
+    def slot_millis(self) -> Optional[int]:
+        """Number of slot ms the user is actually billed for."""
+        return self._slot_millis
+
     def _is_almost_completely_cached(self):
         """Check if all results are completely cached.
 

samples/desktopapp/requirements-test.txt

@@ -1,4 +1,4 @@
 google-cloud-testutils==1.6.4
 pytest==8.4.1
 mock==5.2.0
-pytest-xdist==3.7.0
+pytest-xdist==3.8.0

samples/geography/requirements-test.txt

@@ -1,3 +1,3 @@
 pytest==8.4.1
 mock==5.2.0
-pytest-xdist==3.7.0
+pytest-xdist==3.8.0

samples/geography/requirements.txt

@@ -1,5 +1,5 @@
 attrs==25.3.0
-certifi==2025.6.15
+certifi==2025.7.9
 cffi==1.17.1
 charset-normalizer==3.4.2
 click===8.1.8; python_version == '3.9'
@@ -24,7 +24,7 @@ idna==3.10
 munch==4.0.0
 mypy-extensions==1.1.0
 packaging==25.0
-pandas==2.3.0
+pandas==2.3.1
 proto-plus==1.26.1
 pyarrow==20.0.0
 pyasn1==0.6.1
@@ -39,6 +39,6 @@ rsa==4.9.1
 Shapely===2.0.7; python_version == '3.9'
 Shapely==2.1.1; python_version >= '3.10'
 six==1.17.0
-typing-extensions==4.14.0
+typing-extensions==4.14.1
 typing-inspect==0.9.0
 urllib3==2.5.0