Commita2d3979

committed

feat: expose query job on dbapi cursor

1 parentcbbfeb5 commita2d3979Copy full SHA for a2d3979

File tree

+33

-0

lines changed

+33

-0

lines changed

Lines changed: 10 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -79,6 +79,16 @@ def __init__(self, connection):`
`79`	`79`	`self._query_job=None`
`80`	`80`	`self._closed=False`
`81`	`81`
	`82`	`+@property`
	`83`	`+defquery_job(self):`
	`84`	`+"""google.cloud.bigquery.job.query.QueryJob: The query job created by`
	`85`	+ the last ``execute*()`` call.
	`86`	`+`
	`87`	`+ .. note::`
	`88`	+ If the last ``execute*()`` call was ``executemany()``, this is the
	`89`	+ last job created by ``executemany()``."""
	`90`	`+returnself._query_job`
	`91`	`+`
`82`	`92`	`defclose(self):`
`83`	`93`	`"""Mark the cursor as closed, preventing its further use."""`
`84`	`94`	`self._closed=True`

Lines changed: 23 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -662,6 +662,29 @@ def test_is_iterable(self):`
`662`	`662`	`"Iterating again over the same results should produce no rows.",`
`663`	`663`	`)`
`664`	`664`
	`665`	`+deftest_query_job_wo_execute(self):`
	`666`	`+fromgoogle.cloud.bigqueryimportdbapi`
	`667`	`+`
	`668`	`+connection=dbapi.connect(self._mock_client())`
	`669`	`+cursor=connection.cursor()`
	`670`	`+self.assertIsNone(cursor.query_job)`
	`671`	`+`
	`672`	`+deftest_query_job_w_execute(self):`
	`673`	`+fromgoogle.cloud.bigqueryimportdbapi,QueryJob`
	`674`	`+`
	`675`	`+connection=dbapi.connect(self._mock_client())`
	`676`	`+cursor=connection.cursor()`
	`677`	`+cursor.execute("SELECT 1;")`
	`678`	`+self.assertIsInstance(cursor.query_job,QueryJob)`
	`679`	`+`
	`680`	`+deftest_query_job_w_executemany(self):`
	`681`	`+fromgoogle.cloud.bigqueryimportdbapi,QueryJob`
	`682`	`+`
	`683`	`+connection=dbapi.connect(self._mock_client())`
	`684`	`+cursor=connection.cursor()`
	`685`	`+cursor.executemany("SELECT %s;", (("1",), ("2",)))`
	`686`	`+self.assertIsInstance(cursor.query_job,QueryJob)`
	`687`	`+`
`665`	`688`	`deftest__format_operation_w_dict(self):`
`666`	`689`	`fromgoogle.cloud.bigquery.dbapiimportcursor`
`667`	`690`

Comments

(0)