a !fQ@sTdZddlZddlmZddlmZddlmZGdddeZ Gdd d eZ dS) zDefine API Datasets.N)_datetime_from_microseconds)NotFoundTablec@s4eZdZdZegdZddZddZddZd S) AccessGranta]Represent grant of an access role to an entity. Every entry in the access list will have exactly one of ``userByEmail``, ``groupByEmail``, ``domain``, ``specialGroup`` or ``view`` set. And if anything but ``view`` is set, it'll also have a ``role`` specified. ``role`` is omitted for a ``view``, since ``view`` s are always read-only. See https://cloud.google.com/bigquery/docs/reference/v2/datasets. :type role: string :param role: Role granted to the entity. One of * ``'OWNER'`` * ``'WRITER'`` * ``'READER'`` May also be ``None`` if the ``entity_type`` is ``view``. :type entity_type: string :param entity_type: Type of entity being granted the role. One of :attr:`ENTITY_TYPES`. :type entity_id: string :param entity_id: ID of entity being granted the role. :raises: :class:`ValueError` if the ``entity_type`` is not among :attr:`ENTITY_TYPES`, or if a ``view`` has ``role`` set or a non ``view`` **does not** have a ``role`` set. )Z userByEmailZ groupByEmaildomainZ specialGroupviewcCsr||jvr&d|d|jf}t||dkrF|dur\td|fn|dur\td|f||_||_||_dS)NzEntity type %r not among: %sz, rz/Role must be None for a view. Received role: %rz#Role must be set for entity type %r) ENTITY_TYPESjoin ValueErrorrole entity_type entity_id)selfr r rmessagerX/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/bigquery/dataset.py__init__;s"  zAccessGrant.__init__cCs$|j|jko"|j|jko"|j|jkSNr r r)rotherrrr__eq__Ms    zAccessGrant.__eq__cCsd|j|j|jfS)Nzrrrrr__repr__Ss zAccessGrant.__repr__N) __name__ __module__ __qualname____doc__ frozensetr rrrrrrrrs  rc@s|eZdZdZdZd?ddZeddZedd Zed d Z e j d d Z ed dZ eddZ eddZ eddZeddZeddZej ddZeddZej ddZeddZej ddZed d!Zej d"d!Zed#d$Zd%d&Zed'd(Zd)d*Zd+d,Zd-d.Zd@d/d0ZdAd1d2ZdBd3d4ZdCd5d6ZdDd7d8Z dEd9d:Z!dFd;d<Z"dGd=d>Z#dS)HDatasetaDatasets are containers for tables. See: https://cloud.google.com/bigquery/docs/reference/v2/datasets :type name: string :param name: the name of the dataset :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 access_grants: list of :class:`AccessGrant` :param access_grants: roles granted to entities for this dataset NrcCs||_||_i|_||_dSr)name_client _properties access_grants)rr clientr#rrrrkszDataset.__init__cCs|jjS)zwProject bound to the dataset. :rtype: string :returns: the project (derived from the client). )r!projectrrrrr%rszDataset.projectcCsd|j|jfS)zURL path for the dataset's APIs. :rtype: string :returns: the path based on project and dataste name. z/projects/%s/datasets/%s)r%r rrrrpath{sz Dataset.pathcCs t|jS)zDataset's access grants. :rtype: list of :class:`AccessGrant` :returns: roles granted to entities for this dataset )list_access_grantsrrrrr#szDataset.access_grantscCs(tdd|Dstdt||_dS)a&Update dataset's access grants :type value: list of :class:`AccessGrant` :param value: roles granted to entities for this dataset :raises: TypeError if 'value' is not a sequence, or ValueError if any item in the sequence is not an AccessGrant css|]}t|tVqdSr) isinstancer).0fieldrrr z(Dataset.access_grants..z$Values must be AccessGrant instancesN)allr tupler(rvaluerrrr#s cCs$|jd}|dur td|SdS)zDatetime at which the dataset was created. :rtype: ``datetime.datetime``, or ``NoneType`` :returns: the creation time (None until set from the server). creationTimeN@@r"getr)rZ creation_timerrrcreateds zDataset.createdcCs |jdS)zID for the dataset resource. :rtype: string, or ``NoneType`` :returns: the ID (None until set from the server). idr"r5rrrr dataset_idszDataset.dataset_idcCs |jdS)zETag for the dataset resource. :rtype: string, or ``NoneType`` :returns: the ETag (None until set from the server). etagr8rrrrr:sz Dataset.etagcCs$|jd}|dur td|SdS)zDatetime at which the dataset was last modified. :rtype: ``datetime.datetime``, or ``NoneType`` :returns: the modification time (None until set from the server). lastModifiedTimeNr3r4)rZ modified_timerrrmodifieds zDataset.modifiedcCs |jdS)zURL for the dataset resource. :rtype: string, or ``NoneType`` :returns: the URL (None until set from the server). ZselfLinkr8rrrr self_linkszDataset.self_linkcCs |jdS)zDefault expiration time for tables in the dataset. :rtype: integer, or ``NoneType`` :returns: The time in milliseconds, or None (the default). defaultTableExpirationMsr8rrrrdefault_table_expiration_mssz#Dataset.default_table_expiration_mscCs*t|tjs|durtd||jd<dS)zUpdate default expiration time for tables in the dataset. :type value: integer, or ``NoneType`` :param value: new default time, in milliseconds :raises: ValueError for invalid value types. NPass an integer, or Noner>)r)six integer_typesr r"r0rrrr?s cCs |jdS)zDescription of the dataset. :rtype: string, or ``NoneType`` :returns: The description as set by the user, or None (the default). descriptionr8rrrrrCszDataset.descriptioncCs*t|tjs|durtd||jd<dS)zUpdate description of the dataset. :type value: string, or ``NoneType`` :param value: new description :raises: ValueError for invalid value types. NPass a string, or NonerCr)rA string_typesr r"r0rrrrCs cCs |jdS)zTitle of the dataset. :rtype: string, or ``NoneType`` :returns: The name as set by the user, or None (the default). friendlyNamer8rrrr friendly_nameszDataset.friendly_namecCs*t|tjs|durtd||jd<dS)zUpdate title of the dataset. :type value: string, or ``NoneType`` :param value: new title :raises: ValueError for invalid value types. NrDrGrEr0rrrrHs cCs |jdS)zLocation in which the dataset is hosted. :rtype: string, or ``NoneType`` :returns: The location as set by the user, or None (the default). locationr8rrrrrIszDataset.locationcCs*t|tjs|durtd||jd<dS)zUpdate location in which the dataset is hosted. :type value: string, or ``NoneType`` :param value: new location :raises: ValueError for invalid value types. NrDrIrEr0rrrrIs cCsBd|vsd|dvrtd|dd}|||d}|||S)aFactory: construct a dataset given its API representation :type resource: dict :param resource: dataset resource representation returned from the API :type client: :class:`gcloud.bigquery.client.Client` :param client: Client which holds credentials and project configuration for the dataset. :rtype: :class:`gcloud.bigquery.dataset.Dataset` :returns: Dataset parsed from ``resource``. datasetReference datasetIdzNResource lacks required identity information:["datasetReference"]["datasetId"])r$)KeyError_set_properties)clsresourcer$r datasetrrr from_api_repr&s    zDataset.from_api_reprcCs|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!rr$rrr_require_client=s zDataset._require_clientcCsZg}|D]L}|}|dd}|\}}t|dkrBtd||t|||q|S)aParse a resource fragment into a set of access grants. ``role`` augments the entity type and present **unless** the entity type is ``view``. :type access: list of mappings :param access: each mapping represents a single access grant :rtype: list of :class:`AccessGrant` :returns: a list of parsed grants :raises: :class:`ValueError` if a grant in ``access`` has more keys than ``role`` and one additional key. r Nrz$Grant has unexpected keys remaining.)copypoppopitemlenr appendr)accessresultgrantr r rrrr_parse_access_grantsKs     zDataset._parse_access_grantscCs|j|}|dd}|||_d|vrBt|d|d<d|vrZt|d|d<d|vrrt|d|d<|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 rYrr2r;r>N) r"clearrTrUr\r#floatintupdate)r api_responsecleanedrYrrrrMes   zDataset._set_propertiescCs>g}|jD].}|j|ji}|jdur.|j|d<||q |S)z9Generate a resource fragment for dataset's access grants.Nr )r#r rr rX)rrZr[inforrr_build_access_resourcexs     zDataset._build_access_resourcecCsd|j|jdi}|jdur*|j}||d<|jdur>|j|d<|jdurR|j|d<|jdurf|j|d<t|jdkr||d <|S) z1Generate a resource for ``create`` or ``update``.rJ)Z projectIdrKNr>rCrGrIrrY) r%r r?rCrHrIrWr#rd)rrOr1rrr_build_resources         zDataset._build_resourcecCs:||}d|jf}|jjd||d}||dS)akAPI call: create the dataset via a PUT request See: https://cloud.google.com/bigquery/docs/reference/v2/tables/insert :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/datasetsPOSTmethodr&dataN)rSr% connection api_requestrerM)rr$r&rarrrcreates   zDataset.createcCsD||}z|jjd|jddidWnty:YdS0dSdS)aAPI call: test for the existence of the dataset via a GET request See https://cloud.google.com/bigquery/docs/reference/v2/datasets/get :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: bool :returns: Boolean indicating existence of the dataset. GETfieldsr7rhr&Z query_paramsFTN)rSrjrkr&rrRrrrexistss    zDataset.existscCs*||}|jjd|jd}||dS)aqAPI call: refresh dataset properties via a GET request See https://cloud.google.com/bigquery/docs/reference/v2/datasets/get :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. rmrhr&N)rSrjrkr&rMrr$rarrrreloads zDataset.reloadcKs||}i}d|vrB|d}t|tjs:|dur:td||d<d|vrV|d|d<d|vrj|d|d<d|vr~|d|d<|jjd |j|d }||dS) aAPI call: update individual dataset properties via a PATCH request See https://cloud.google.com/bigquery/docs/reference/v2/datasets/patch :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. :type kw: ``dict`` :param kw: properties to be patched. :raises: ValueError for invalid value types. r?Nr@r>rCrHrGrIPATCHrg) rSr)rArBr rjrkr&rM)rr$kwpartialr1rarrrpatchs"    z Dataset.patchcCs0||}|jjd|j|d}||dS)asAPI call: update dataset properties via a PUT request See https://cloud.google.com/bigquery/docs/reference/v2/datasets/update :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. PUTrgN)rSrjrkr&rerMrrrrrr`s  zDataset.updatecCs ||}|jjd|jddS)anAPI call: delete the dataset via a DELETE request See: https://cloud.google.com/bigquery/docs/reference/v2/tables/delete :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. DELETErqN)rSrjrkr&rRrrrdeletes zDataset.deletecsti}|dur||d<|dur$||d<djjf}jj}|jd||d}fdd|d d D}||d fS) aWList tables for the project associated with this client. See: https://cloud.google.com/bigquery/docs/reference/v2/tables/list :type max_results: int :param max_results: maximum number of tables to return, If not passed, defaults to a value set by the API. :type page_token: string :param page_token: opaque marker for the next "page" of datasets. If not passed, the API will return the first page of datasets. :rtype: tuple, (list, str) :returns: list of :class:`gcloud.bigquery.table.Table`, plus a "next page token" string: if not ``None``, indicates that more tables can be retrieved with another call (pass that value as ``page_token``). NZ maxResultsZ pageTokenz/projects/%s/datasets/%s/tablesrmrocsg|]}t|qSr)rrQ)r*rOrrr 4sz'Dataset.list_tables..tablesrZ nextPageToken)r%r r!rjrkr5)rZ max_resultsZ page_tokenparamsr&rjrespr|rrr list_tabless  zDataset.list_tablescCst|||dS)aJConstruct a table bound to this dataset. :type name: string :param name: Name of the table. :type schema: list of :class:`gcloud.bigquery.table.SchemaField` :param schema: The table's schema :rtype: :class:`gcloud.bigquery.table.Table` :returns: a new ``Table`` instance )rPschemar)rr rrrrtable8s z Dataset.table)r)N)N)N)N)N)N)NN)r)$rrrrr(rpropertyr%r&r#setterr6r9r:r<r=r?rCrHrI classmethodrQrS staticmethodr\rMrdrerlrprsrwr`rzrrrrrrrXsj                         &  %r) rrAZgcloud._helpersrZgcloud.exceptionsrZgcloud.bigquery.tablerobjectrrrrrrs    A