o ;Di_@sddlmZmZmZddlmZmZddlmZddl m Z ddZ ddd Z dd d Z dd dZGdddeZ dddZ dddZee_ee_d S))absolute_importprint_functiondivision)Tableiterdata) text_type) ArgumentErrorcCsrddl}t||jr |St|tr||}|Sddl}t||jjjr+| |}|S|dur5|}|St d)Nrz$gspread: Invalid account credentials) gspread isinstanceClientdictservice_account_from_dictgoogleoauth2service_account Credentials authorize PetlArgError) auth_infor gdrgcgarG/var/www/Datamplify/venv/lib/python3.10/site-packages/petl/io/gsheet.py_get_gspread_client s    rFcCs^|r ddlm}z||}W|S|y||}Y|Sw|dur+||}|Std)Nr)SpreadsheetNotFoundz%gspread requires argument spreadsheet)gspread.exceptionsr open_by_keyopenr)gspread_client spreadsheetrrwbrrr_open_spreadsheets    r"cCs|dur |j}|St|tr||}|St|trBt|}|r;|dd|Dvr2||}|S||dd}|S||}|Std)NcSsg|]}|jqSrtitle).0wbsrrr 3sz%_select_worksheet..z,Only can find worksheet by name or by number) sheet1r int get_worksheetr worksheets worksheet add_worksheetr)r!r-find_or_createws sheetnamerrr_select_worksheet*s"    r2NcCst|||||S)a Extract a table from a google spreadsheet. The `credentials_or_client` are used to authenticate with the google apis. For more info, check `authentication`_. The `spreadsheet` can either be the key of the spreadsheet or its name. The `worksheet` argument can be omitted, in which case the first sheet in the workbook is used by default. The `cell_range` argument can be used to provide a range string specifying the top left and bottom right corners of a set of cells to extract. (i.e. 'A1:C7'). Set `open_by_key` to `True` in order to treat `spreadsheet` as spreadsheet key. .. note:: - Only the top level of google drive will be searched for the spreadsheet filename due to API limitations. - The worksheet name is case sensitive. Example usage follows:: >>> from petl import fromgsheet >>> import gspread # doctest: +SKIP >>> client = gspread.service_account() # doctest: +SKIP >>> tbl1 = fromgsheet(client, 'example_spreadsheet', 'Sheet1') # doctest: +SKIP >>> tbl2 = fromgsheet(client, '9zDNETemfau0uY8ZJF0YzXEPB_5GQ75JV', credentials) # doctest: +SKIP This functionality relies heavily on the work by @burnash and his great `gspread module`_. .. _gspread module: http://gspread.readthedocs.io/ .. _authentication: http://gspread.readthedocs.io/en/latest/oauth2.html )GoogleSheetView)credentials_or_clientr r- cell_rangerrrr fromgsheet?s)r6c@s0eZdZdZddZddZddZdd Zd S) r3z2Conects to a worksheet and iterates over its rows.cCs"||_||_||_||_||_dSN)rr r-r5r)selfr4r r-r5rrrr__init__ts  zGoogleSheetView.__init__cCsDt|j}t||j|j}t||j}|jdur||S| |Sr7) rrr"r rr2r-r5_yield_by_range_yield_all_rows)r8rr!r0rrr__iter__~s     zGoogleSheetView.__iter__ccs|D]}t|VqdSr7)get_all_valuestuple)r8r0rowrrrr;s  zGoogleSheetView._yield_all_rowsccs&||j}|D]}t|Vq dSr7) get_valuesr5r>)r8r0foundr?rrrr:s   zGoogleSheetView._yield_by_rangeN)__name__ __module__ __qualname____doc__r9r<r;r:rrrrr3qs   r3readerc Cspt|}||}|j} | jddd|dur| j|d| j||d|dur5|D] } |j| d|dq*|jS)a Write a table to a new google sheet. The `credentials_or_client` are used to authenticate with the google apis. For more info, check `authentication`_. The `spreadsheet` will be the title of the workbook created google sheets. If there is a spreadsheet with same title a new one will be created. If `worksheet` is specified, the first worksheet in the spreadsheet will be renamed to its value. The spreadsheet will be shared with all emails in `share_emails` with `role` permissions granted. For more info, check `sharing`_. Returns: the spreadsheet key that can be used in `appendgsheet` further. .. _sharing: https://developers.google.com/drive/v3/web/manage-sharing .. note:: The `gspread`_ package doesn't support serialization of `date` and `datetime` types yet. Example usage:: >>> from petl import fromcolumns, togsheet >>> import gspread # doctest: +SKIP >>> client = gspread.service_account() # doctest: +SKIP >>> cols = [[0, 1, 2], ['a', 'b', 'c']] >>> tbl = fromcolumns(cols) >>> togsheet(tbl, client, 'example_spreadsheet') # doctest: +SKIP r()rowscolsNr#) table_rangeuser) perm_typerole)rcreater)resize update_title append_rowsshareid) tabler4r r-r5 share_emailsrLrr!r0 user_emailrrrtogsheets&  rVc CsDt|}t|||}t||d}|r|ntt|} || |jS)a Append a table to an existing google shoot at either a new worksheet or the end of an existing worksheet. The `credentials_or_client` are used to authenticate with the google apis. For more info, check `authentication`_. The `spreadsheet` is the name of the workbook to append to. The `worksheet` is the title of the worksheet to append to or create when it does not exist yet. Set `open_by_key` to `True` in order to treat `spreadsheet` as spreadsheet key. Set `include_header` to `True` if you don't want omit fieldnames as the first row appended. .. note:: The sheet index cannot be used, and None is not an option. T)rr"r2listrrPrR) rSr4r r-rinclude_headerrr!r0rGrrr appendgsheets    rY)F)NNF)NNNrF)NFF) __future__rrrpetl.util.baserr petl.compatr petl.errorsrrrr"r2r6r3rVrYrrrrs&     2" 7 "