a !ff2@sdZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z Gd d d e ZGd d d e ZdS)zDefine API Queries.N)_TypedProperty)_rows_from_json)Dataset)QueryJob)UDFResourcesProperty)_build_udf_resources)_parse_schema_resourcec@s,eZdZdZdZdZdZdZdZdZ dZ dS)_SyncQueryConfigurationzwUser-settable configuration options for synchronous query jobs. Values which are ``None`` -> server defaults. N) __name__ __module__ __qualname____doc__Z_default_dataset_dry_runZ _max_resultsZ _timeout_msZ_preserve_nullsZ_use_query_cacheZ_use_legacy_sqlrrV/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/bigquery/query.pyr sr c@seZdZdZdZd.ddZeddZdd Zed d Z ed d Z eddZ eddZ eddZ eddZeddZeddZeddZeddZedeZedeZed ejZed!eZed"ejZeZed#eZed$eZ d%d&Z!d'd(Z"d/d*d+Z#d0d,d-Z$d)S)1 QueryResultsaSynchronous job: query tables. :type query: string :param query: SQL query string :type client: :class:`gcloud.bigquery.client.Client` :param client: A client which holds credentials and project configuration for the dataset (which requires a project). :type udf_resources: tuple :param udf_resources: An iterable of :class:`gcloud.bigquery.job.UDFResource` (empty by default) ZuserDefinedFunctionResourcesrcCs*||_i|_||_t|_||_d|_dS)N)_client _propertiesqueryr Z_configuration udf_resources_job)selfrclientrrrr__init__<s zQueryResults.__init__cCs|jjS)zsProject bound to the job. :rtype: string :returns: the project (derived from the client). )rprojectrrrrrDszQueryResults.projectcCs|dur|j}|S)aCheck client or verify over-ride. :type client: :class:`gcloud.bigquery.client.Client` or ``NoneType`` :param client: the client to use. If not passed, falls back to the ``client`` stored on the current dataset. :rtype: :class:`gcloud.bigquery.client.Client` :returns: The client passed in or the currently bound client. N)r)rrrrr_require_clientMs zQueryResults._require_clientcCs |jdS)a%Query results served from cache. See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#cacheHit :rtype: boolean or ``NoneType`` :returns: True if the query results were served from cache (None until set by the server). ZcacheHitrgetrrrr cache_hit[s zQueryResults.cache_hitcCs |jdS)aServer completed query. See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#jobComplete :rtype: boolean or ``NoneType`` :returns: True if the query completed on the server (None until set by the server). Z jobCompleterrrrrcompletehs zQueryResults.completecCs |jdS)a,Errors generated by the query. See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#errors :rtype: list of mapping, or ``NoneType`` :returns: Mappings describing errors generated on the server (None until set by the server). errorsrrrrrr!us zQueryResults.errorscCs|jdidS)a8Job name, generated by the back-end. See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#jobReference :rtype: list of mapping, or ``NoneType`` :returns: Mappings describing errors generated on the server (None until set by the server). jobReferencejobIdrrrrrnames zQueryResults.namecCs:|jdur4|jd}|dur4t|d|j|j|_|jS)zJob instance used to run the query. :rtype: :class:`gcloud.bigquery.job.QueryJob`, or ``NoneType`` :returns: Job instance used to run the query (None until ``jobReference`` property is set by the server). Nr"r#)rrrrrr)rZjob_refrrrjobs   zQueryResults.jobcCs |jdS)a Token for fetching next bach of results. See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#pageToken :rtype: string, or ``NoneType`` :returns: Token generated on the server (None until set by the server). pageTokenrrrrr page_tokens zQueryResults.page_tokencCs |jdS)a Total number of rows returned by the query See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#totalRows :rtype: integer, or ``NoneType`` :returns: Count generated on the server (None until set by the server). totalRowsrrrrr total_rowss zQueryResults.total_rowscCs |jdS)aTotal number of bytes processed by the query See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#totalBytesProcessed :rtype: integer, or ``NoneType`` :returns: Count generated on the server (None until set by the server). ZtotalBytesProcessedrrrrrtotal_bytes_processeds z"QueryResults.total_bytes_processedcCst|jdd|jS)zQuery results. See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#rows :rtype: list of tuples of row values, or ``NoneType`` :returns: fields describing the schema (None until set by the server). rowsr)rrrschemarrrrr+s zQueryResults.rowscCst|jdiS)a Schema for query results. See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query#schema :rtype: list of :class:`SchemaField`, or ``NoneType`` :returns: fields describing the schema (None until set by the server). r,)rrrrrrrr,s zQueryResults.schemadefault_datasetdry_run max_resultspreserve_nulls timeout_msuse_query_cacheuse_legacy_sqlcCs|j|j|dS)zUpdate properties from resource in body of ``api_response`` :type api_response: httplib2.Response :param api_response: response returned from an API call N)rclearupdate)r api_responserrr_set_propertiess zQueryResults._set_propertiescCsd|ji}|jdur(|j|jjd|d<|jdur<|j|d<|jdurP|j|d<|jdurd|j|d<|jdurx|j|d<|jdur|j|d <|j dur|j |d <t |j d krt |j ||j <|S) z&Generate a resource for :meth:`begin`.rN)Z projectIdZ datasetIdZdefaultDataset maxResultsZ preserveNulls timeoutMsZ useQueryCacheZ useLegacySqlZdryRunr)rr-rr$r/r0r1r2r3r.lenZ_udf_resourcesr_UDF_KEY)rresourcerrr_build_resource s(               zQueryResults._build_resourceNcCs:||}d|jf}|jjd||d}||dS)adAPI call: run the query via a POST request See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/query :type client: :class:`gcloud.bigquery.client.Client` or ``NoneType`` :param client: the client to use. If not passed, falls back to the ``client`` stored on the current dataset. z/projects/%s/queriesPOST)methodpathdataN)rr connection api_requestr=r7)rrr@r6rrrrun*s   zQueryResults.runc Cs|jdurtd||}i}|dur0||d<|dur@||d<|durP||d<|dur`||d<d|j|jf}|jjd||d }|||d } | durt| } |d}t |d d |j } | | |fS) aAPI call: fetch a page of query result data via a GET request See: https://cloud.google.com/bigquery/docs/reference/v2/jobs/getQueryResults :type max_results: integer or ``NoneType`` :param max_results: maximum number of rows to return. :type page_token: string or ``NoneType`` :param page_token: token representing a cursor into the table's rows. :type start_index: integer or ``NoneType`` :param start_index: zero-based index of starting row :type timeout_ms: integer or ``NoneType`` :param timeout_ms: timeout, in milliseconds, to wait for query to complete :type client: :class:`gcloud.bigquery.client.Client` or ``NoneType`` :param client: the client to use. If not passed, falls back to the ``client`` stored on the current dataset. :rtype: tuple :returns: ``(row_data, total_rows, page_token)``, where ``row_data`` is a list of tuples, one per result row, containing only the values; ``total_rows`` is a count of the total number of rows in the table; and ``page_token`` is an opaque string which can be used to fetch the next batch of rows (``None`` if no further batches can be fetched). :raises: ValueError if the query has not yet been executed. Nz%Query not yet executed: call 'run()'r8r&Z startIndexr9z/projects/%s/queries/%sGET)r?r@Z query_paramsr(r+r) r$ ValueErrorrrrBrCr7rintrr,) rr/r' start_indexr1rparamsr@responser)Z rows_datarrr fetch_data:s0!     zQueryResults.fetch_data)r)N)NNNNN)%r r r r r;rpropertyrrrr r!r$r%r'r)r*r+r,rrr-boolr.six integer_typesr/r0r1rrr2r3r7r=rDrKrrrrr*sP                   ! r)r rNZgcloud.bigquery._helpersrrZgcloud.bigquery.datasetrZgcloud.bigquery.jobrrrZgcloud.bigquery.tablerobjectr rrrrrs