o ߭j @s>UdZddlmZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z ddl mZddl mZdZd ed <zddlZWneyWZz eZdZWYdZ[ndZ[wwedur^enejjZGd d d eZd%ddZGdddejZGdddeZGdddeZdaded<d&ddZ d'd d!Z!d(d#d$Z"dS))z1Module that holds a global gspread.client.Client.) annotationsN)AnyCallableMappingSequence) credentials) html_utils) ipython_env) sheets_idzException | None_gspread_import_errorc@s eZdZdS)SpreadsheetNotFoundErrorN)__name__ __module__ __qualname__rrh/var/www/html/CropPilot/venv/lib/python3.10/site-packages/google/generativeai/notebook/gspread_client.pyr *sr return ExceptioncCstdtS)Nz&"gspread" module not imported, got: {}) RuntimeErrorformatr rrrr_get_import_error.rc@s@eZdZdZejdddZejdd d ZejdddZdS) GSpreadClientztWrapper around gspread.client.Client. This adds a layer of indirection for us to inject mocks for testing. sidsheets_id.SheetsIdentifierrNonecCdS)zValidates that `name` is the name of a Google Sheets document. Raises an exception if false. Args: sid: The identifier for the document. Nrselfrrrrvalidate8zGSpreadClient.validate worksheet_idint6tuple[Sequence[Mapping[str, str]], Callable[[], None]]cCr)z2Returns all records for a Google Sheets worksheet.Nrrrr!rrrget_all_recordsBr zGSpreadClient.get_all_recordsrowsSequence[Sequence[Any]]cCr)z@Writes results to a new worksheet to the Google Sheets document.Nrrrr&rrr write_recordsJr zGSpreadClient.write_recordsNrrrrrrr!r"rr#rrr&r'rr) r rr__doc__abcabstractmethodrr%r)rrrrr2s  rc@sBeZdZdZdddZdd d ZdddZdddZdddZdS)GSpreadClientImplz)Concrete implementation of GSpreadClient.clientrenvipython_env.IPythonEnv | NonecCs||_||_dS)zConstructor. Args: client: Instance of gspread.client.Client. env: Optional instance of IPythonEnv. This is used to display messages such as the URL of the output Worksheet. N)_client _ipython_env)rr1r2rrr__init__Vs zGSpreadClientImpl.__init__rrc Csz0|r|j|WS|r|jt|WS|r,|jt|WSWt dtyD}zt d ||d}~ww)acOpens a Sheets document from `sid`. Args: sid: The identifier for the Sheets document. Raises: SpreadsheetNotFoundError: If the Sheets document cannot be found or cannot be opened. Returns: A gspread.Worksheet instance representing the worksheet referred to by `sid`. zUnable to find Sheets with {}Nz"Invalid sheets_id.SheetsIdentifier) namer4openkey open_by_keystrurl open_by_urlGSpreadExceptionr r)rrexcrrr_openaszGSpreadClientImpl._openrrcCs||dSN)r@rrrrrzrzGSpreadClientImpl.validater!r"r#csP||||jdur|jfdd}nfdd}|fS)Nc s2dtjtjdjjddS)Nz Reading inputs from worksheet {}{} in {}r<text) display_htmlrrget_anchor_tagr SheetsURLr<titlerr2sheet worksheetrr _display_fns z6GSpreadClientImpl.get_all_records.._display_fncstdjjdS)Nz&Reading inputs from worksheet {} in {})printrrHr)rJrKrrrLs)r@ get_worksheetr5r%)rrr!rLrrIrr%}s    z!GSpreadClientImpl.get_all_recordsr&r'c Cs||}tj}d|dd|dd}|j|ddd}|j|d|jdurB|jd tj t |j d |j |j d dStd |j |j dS) NzResults z%Y_%m_%dz (z%s))rHr&cols)valuesz#Results written to new worksheet {}rBrCz)Results written to new worksheet {} in {})r@datetimenow add_worksheet append_rowsr5rErrrFr rGr<rHrM)rrr&rJcurrent_datetimerHrKrrrr)s      zGSpreadClientImpl.write_recordsN)r1rr2r3)rrr*r+r,) r rrr-r6r@rr%r)rrrrr0Ss   r0c@s.eZdZdZdddZdd d ZdddZdS)NullGSpreadClientzNull-object implementation of GSpreadClient. This class raises an error if any of its methods are called. It is used when the gspread library is not available. rrrrcCtrArrrrrrszNullGSpreadClient.validater!r"r#cCrYrArZr$rrrr%z!NullGSpreadClient.get_all_recordsr&r'cCrYrArZr(rrrr)r[zNullGSpreadClient.write_recordsNr*r+r,)r rrr-rr%r)rrrrrXs   rXzGSpreadClient | None_gspread_clientcredscredentials.Credentialsr2r3rcCs,tdurt|}t||dadStadS)z Sets up credential for gspreads.N)r1r2)gspread authorizer0r\rX)r]r2r1rrrr`s  r`cCststdtS)NzMust call authorize() first)r\rrrrr get_clientsrar1cCs|adS)z(Overrides the global client for testing.N)r\)r1rrrtestonly_set_clientsrb)rr)r]r^r2r3rr)rr)r1rrr)#r- __future__rr.rStypingrrrr google.authrgoogle.generativeai.notebookrr r r __annotations__r_ ImportErrorer exceptionsr>rr rABCrr0rXr\r`rarbrrrrs6        !e