o ihL@sddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd l mZd Zd d ddiiZGdddeZGdddeZGdddeZddZGddde ZGddde e ZdS)N)errors)MediaIoBaseUploadwraps) ApiAttribute)ApiAttributeMixin) ApiResource)ApiResourceList)LoadAuthiz$application/vnd.google-apps.document text/plainuutf8c@eZdZdZdS)FileNotUploadedErrorz=Error trying to access metadata of file that is not uploaded.N__name__ __module__ __qualname____doc__rrH/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pydrive/files.pyrrc@r)ApiRequestErrorz$Error while making any API requests.Nrrrrrrrrc@r)FileNotDownloadableErrorz7Error trying to download file that is not downloadable.Nrrrrrrrrcstfdd}|S)zwDecorator to check if the file has metadata and fetches it if not. :raises: ApiRequestError, FileNotUploadedError cs$|js||g|Ri|SN)uploaded FetchMetadata)selfargskwargs decorateerr _decorated'sz LoadMetadata.._decoratedr)r!r"rr r LoadMetadata"sr#cs.eZdZdZdfdd ZeddZZS)GoogleDriveFileListzOGoogle Drive FileList instance. Equivalent to Files.list() in Drive APIs. Ncstt|j||ddS)z*Create an instance of GoogleDriveFileList.)authmetadataN)superr$__init__)rr%param __class__rrr(5szGoogleDriveFileList.__init__cCsZ|jjjdit|j|jd|_g}|jdD]}t|j|dd}| |q|S)zOverwritten method which actually makes API call to list files. :returns: list -- list of pydrive.files.GoogleDriveFile. httpitemsT)r%r&rNr) r%servicefileslistdictexecuter-r&GoogleDriveFileappend)rresult file_metadatatmp_filerrr_GetList9s zGoogleDriveFileList._GetList)NN)rrrrr(r r9 __classcell__rrr*rr$/s r$c@s\eZdZdZedZedZedZdddZ d?ddZ ed?ddZed?ddZd@ddZd@ddZd@ddZd@ddZd d!Zed"d#Zd$d%Zed@d&d'Zed@d(d)Zed@d*d+Zed@d,d-Zeed@d.d/Zeed@d0d1Zd2d3Zed4d5Z ed6d7Z!e"e#fd8d9Z$e"e#fd:d;Z%dS)Ar4z}Google Drive File instance. Inherits ApiResource which inherits dict. Can access and modify metadata like dictionary. contentrr&NFcCs^t|t|i|_ddi|_||_||_|r ||n|r'||d|_ d|_ dS)aDCreate an instance of GoogleDriveFile. :param auth: authorized GoogleAuth instance. :type auth: pydrive.auth.GoogleAuth :param metadata: file resource to initialize GoogleDriveFile with. :type metadata: dict. :param uploaded: True if this file is confirmed to be uploaded. :type uploaded: bool. r;FaalternateLink,appDataContents,canComment,canReadRevisions,copyable,createdDate,defaultOpenWithLink,description,downloadUrl,editable,embedLink,etag,explicitlyTrashed,exportLinks,fileExtension,fileSize,folderColorRgb,fullFileExtension,headRevisionId,iconLink,id,imageMediaMetadata,indexableText,isAppAuthorized,kind,labels,lastModifyingUser,lastModifyingUserName,lastViewedByMeDate,markedViewedByMeDate,md5Checksum,mimeType,modifiedByMeDate,modifiedDate,openWithLinks,originalFilename,ownedByMe,ownerNames,owners,parents,permissions,properties,quotaBytesUsed,selfLink,shareable,shared,sharedWithMeDate,sharingUser,spaces,thumbnail,thumbnailLink,title,userPermission,version,videoMediaMetadata,webContentLink,webViewLink,writersCanShareTN) rr(r r&dirtyr%rUpdateMetadataupdate _ALL_FIELDShas_bom)rr%r&rrrrr(Us     zGoogleDriveFile.__init__c Csfzt||WSty2}z|jrt||dr+|t||WYd}~Std}~ww)a:Overwrites manner of accessing Files resource. If this file instance is not uploaded and id is specified, it will try to look for metadata with Files.get(). :param key: key of dictionary query. :type key: str. :returns: value of Files resource :raises: KeyError, FileNotUploadedError idN)r2 __getitem__KeyErrorrgetrr)rkeyerrrrBzs  zGoogleDriveFile.__getitem__utf-8cCs0t|||_|ddurd|d<dSdS)aQSet content of this file to be a string. Creates io.BytesIO instance of utf-8 encoded string. Sets mimeType to be 'text/plain' if not specified. :param encoding: The encoding to use when setting the content of this file. :type encoding: str :param content: content of the file in string. :type content: str mimeTypeNr )ioBytesIOencoder;rD)rr;encodingrrrSetContentStrings  z GoogleDriveFile.SetContentStringcCsJt|d|_|ddur||d<|ddur#t|d|d<dSdS)a,Set content of this file from a file. Opens the file specified by this method. Will be read, uploaded, and closed by Upload() method. Sets metadata 'title' and 'mimeType' automatically if not specified. :param filename: name of the file to be uploaded. :type filename: str. rbtitleNrHr)openr;rD mimetypes guess_type)rfilenamerrrSetContentFiles zGoogleDriveFile.SetContentFilecCs@|jdust|jtjus|j|kr||||j|S)aGet content of this file as a string. :param mimetype: The mimetype of the content string. :type mimetype: str :param encoding: The encoding to use when decoding the byte string. :type encoding: str :param remove_bom: Whether to strip a known BOM. :type remove_bom: bool :returns: str -- utf-8 decoded content of the file :raises: ApiRequestError, FileNotUploadedError, FileNotDownloadableError N)r;typerIrJr@ FetchContentgetvaluedecode)rmimetyperL remove_bomrrrGetContentStrings   z GoogleDriveFile.GetContentStringcCsV|jdust|jtjus|j|kr|||t|d}||j| dS)agSave content of this file as a local file. :param filename: name of the file to write to. :type filename: str :param mimetype: mimeType of the file. :type mimetype: str :param remove_bom: Whether to remove the byte order marking. :type remove_bom: bool :raises: ApiRequestError, FileNotUploadedError, FileNotDownloadableError Nwb) r;rUrIrJr@rVrPwriterWclose)rrSrYrZfrrrGetContentFiles     zGoogleDriveFile.GetContentFilec Cs|jdp |d}|r|j}|r@z|jjj||dj|jd}Wntj y5}zt |d}~wwd|_ | |dSt )aDDownload file's metadata from id using Files.get(). :param fields: The fields to include, as one string, each entry separated by commas, e.g. 'fields,labels'. :type fields: str :param fetch_all: Whether to fetch all fields. :type fetch_all: bool :raises: ApiRequestError, FileNotUploadedError rA)fileIdfieldsr,NT)r&rDr?r%r/r0r3r-r HttpErrorrrr=r)rrb fetch_allfile_idr&errorrrrrs"  zGoogleDriveFile.FetchMetadatacCs|jd}|jd}|rt|||_d|jd<n|r6||r6t||||_d|jd<ntd|dkrS|rU||jt |d|| |_ dSdSdS) z}Download file's content from download_url. :raises: ApiRequestError, FileNotUploadedError, FileNotDownloadableError downloadUrl exportLinksFr;z:No downloadLink/exportLinks for mimetype found in metadatar rHN) r&rDrIrJ_DownloadFromUrlr;r<r _RemovePrefixMIME_TYPE_TO_BOMr@)rrYrZ download_url export_linksrrrrVs&      zGoogleDriveFile.FetchContentcCsN|js |ddur|jdr|j|ddS|j|ddS|j|ddS)zUpload/update file by choosing the most efficient method. :param param: additional parameter to upload file. :type param: dict. :raises: ApiRequestError rANr;r))rrDr< _FilesUpdate _FilesPatch _FilesInsertrr)rrrUploads  zGoogleDriveFile.UploadcC|j|ddS)zr3r-rrcrrr=rrrrros"      zGoogleDriveFile._FilesUpdatec Cs|duri}||d<|jd|d<z|jjjdi|j|jd}Wnt j y8}zt |d}~ww| |dS)zUpdate metadata using Files.Patch(). :param param: additional parameter to upload file. :type param: dict. :raises: ApiRequestError, FileNotUploadedError Nr{rArar,r) rr&rDr%r/r0patchr3r-rrcrr=rrrrrps   zGoogleDriveFile._FilesPatchcCs*|ddur d|d<t|j|dddS)zBuild MediaIoBaseUpload to get prepared to upload content of the file. Sets mimeType as 'application/octet-stream' if not specified. :returns: MediaIoBaseUpload -- instance that will be used to upload content. rHNzapplication/octet-streamT) resumable)rDrr;rrrrrszGoogleDriveFile._BuildMediaBodycCs*|j|\}}|jdkrtd||S)zDownload file from url using provided credential. :param url: link of the file to download. :type url: str. :returns: str -- content of downloaded file in string. :raises: ApiRequestError zCannot download file: %s)r-requeststatusr)rurlrespr;rrrris  z GoogleDriveFile._DownloadFromUrlc s|jdp |d}z|jjj|dWntjy*}zt |d}~wwd|vrLd|jvrL|d}fdd}t ||}||d<||jd<dS)zDeletes the permission remotely, and from the file object itself. :param permission_id: The ID of the permission. :type permission_id: str :return: The permission :rtype: object rA)ra permissionIdNr|cs |dkS)NrAr)perrrrs z3GoogleDriveFile._DeletePermission..T) r&rDr%r/r|rr3rrcrfilter)rrrerfr|is_not_current_permissionrrrrs"      z!GoogleDriveFile._DeletePermissionc Cst|}||}||krO||}t|}||}d}|dkrF|||||t|7}||||}t|}||7}|dks |||dSdS)a*Deletes passed prefix by shifting content of passed file object by to the left. Operation is in-place. Args: file_object (obj): The file object to manipulate. prefix (str): The prefix to insert. block_size (int): The size of the blocks which are moved one at a time. rN)lenreadseekr]truncate) file_objectprefix block_size prefix_length content_startblock_to_writecurrent_block_length read_locationwrite_locationrrrrjs$       zGoogleDriveFile._RemovePrefixcCs||}||}|d}|d||t|}t|rD||||||7}|}||||}||7}t|s dSdS)zInserts the passed prefix in the beginning of the file, operation is in-place. Args: file_object (obj): The file object to manipulate. prefix (str): The prefix to insert. rN)rrr]r)rrr first_block second_blockrrrrr _InsertPrefix=s       zGoogleDriveFile._InsertPrefix)NNF)rG)NrGF)NFr)&rrrrrr;rr&r(rBrMrTr[r`r rr#rVrsrvrxrzrr~rrqrwruryrorprrir staticmethod BLOCK_SIZErjrrrrrr4Ks\ %                  'r4)rIrQ apiclientrapiclient.httpr functoolsrapiattrrrr r r%r rrKrk RuntimeErrorrIOErrorrrr#r$r4rrrrs*