Commitaa1613c

authored

feat: expose require_partition_filter for hive_partition (#257)

1 parenta125160 commitaa1613cCopy full SHA for aa1613c

File tree

+19

-0

lines changed

+19

-0

lines changed

Lines changed: 15 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -586,6 +586,21 @@ def source_uri_prefix(self):`
`586`	`586`	`defsource_uri_prefix(self,value):`
`587`	`587`	`self._properties["sourceUriPrefix"]=value`
`588`	`588`
	`589`	`+@property`
	`590`	`+defrequire_partition_filter(self):`
	`591`	`+"""Optional[bool]: If set to true, queries over the partitioned table require a`
	`592`	`+ partition filter that can be used for partition elimination to be`
	`593`	`+ specified.`
	`594`	`+`
	`595`	`+ See`
	`596`	`+ https://cloud.google.com/bigquery/docs/reference/rest/v2/tables#HivePartitioningOptions.FIELDS.mode`
	`597`	`+ """`
	`598`	`+returnself._properties.get("requirePartitionFilter")`
	`599`	`+`
	`600`	`+@require_partition_filter.setter`
	`601`	`+defrequire_partition_filter(self,value):`
	`602`	`+self._properties["requirePartitionFilter"]=value`
	`603`	`+`
`589`	`604`	`defto_api_repr(self):`
`590`	`605`	`"""Build an API representation of this object.`
`591`	`606`

Lines changed: 4 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -181,6 +181,7 @@ def test_from_api_repr_hive_partitioning(self):`
`181`	`181`	`"hivePartitioningOptions": {`
`182`	`182`	`"sourceUriPrefix":"http://foo/bar",`
`183`	`183`	`"mode":"STRINGS",`
	`184`	`+"requirePartitionFilter":True,`
`184`	`185`	`},`
`185`	`186`	`},`
`186`	`187`	`)`
`@@ -194,6 +195,7 @@ def test_from_api_repr_hive_partitioning(self):`
`194`	`195`	`)`
`195`	`196`	`self.assertEqual(ec.hive_partitioning.source_uri_prefix,"http://foo/bar")`
`196`	`197`	`self.assertEqual(ec.hive_partitioning.mode,"STRINGS")`
	`198`	`+self.assertEqual(ec.hive_partitioning.require_partition_filter,True)`
`197`	`199`
`198`	`200`	`# converting back to API representation should yield the same result`
`199`	`201`	`got_resource=ec.to_api_repr()`
`@@ -210,6 +212,7 @@ def test_to_api_repr_hive_partitioning(self):`
`210`	`212`	`hive_partitioning=external_config.HivePartitioningOptions()`
`211`	`213`	`hive_partitioning.source_uri_prefix="http://foo/bar"`
`212`	`214`	`hive_partitioning.mode="STRINGS"`
	`215`	`+hive_partitioning.require_partition_filter=False`
`213`	`216`
`214`	`217`	`ec=external_config.ExternalConfig("FORMAT_FOO")`
`215`	`218`	`ec.hive_partitioning=hive_partitioning`
`@@ -221,6 +224,7 @@ def test_to_api_repr_hive_partitioning(self):`
`221`	`224`	`"hivePartitioningOptions": {`
`222`	`225`	`"sourceUriPrefix":"http://foo/bar",`
`223`	`226`	`"mode":"STRINGS",`
	`227`	`+"requirePartitionFilter":False,`
`224`	`228`	`},`
`225`	`229`	`}`
`226`	`230`	`self.assertEqual(got_resource,expected_resource)`

Comments

(0)