阺h~UdZddlmZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z ddl mZddl mZdZd ed < ddlZn#e$rZeZdZYdZ[ndZ[wwxYween ejjZGd d eZd!dZGddejZGddeZGddeZdaded<d"dZ d#dZ!d$d Z"dS)%z1Module that holds a global gspread.client.Client.) annotationsN)AnyCallableMappingSequence) credentials) html_utils) ipython_env) sheets_idzException | None_gspread_import_errorceZdZdS)SpreadsheetNotFoundErrorN)__name__ __module__ __qualname__s/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/google/generativeai/notebook/gspread_client.pyrr*sDrrreturn ExceptioncPtdtS)Nz&"gspread" module not imported, got: {}) RuntimeErrorformatr rrr_get_import_errorr.s @GGH]^^ _ __rcxeZdZdZejddZejdd Zejdd ZdS) GSpreadClientztWrapper around gspread.client.Client. This adds a layer of indirection for us to inject mocks for testing. sidsheets_id.SheetsIdentifierrNonecdS)zValidates that `name` is the name of a Google Sheets document. Raises an exception if false. Args: sid: The identifier for the document. Nrselfrs rvalidatezGSpreadClient.validate8r worksheet_idint6tuple[Sequence[Mapping[str, str]], Callable[[], None]]cdS)z2Returns all records for a Google Sheets worksheet.Nrr"rr%s rget_all_recordszGSpreadClient.get_all_recordsBr$rrowsSequence[Sequence[Any]]cdS)z@Writes results to a new worksheet to the Google Sheets document.Nrr"rr+s r write_recordszGSpreadClient.write_recordsJr$rNrrrrrrr%r&rr'rrr+r,rr) rrr__doc__abcabstractmethodr#r*r/rrrrr2s       AAAA OOOOOOrrc:eZdZdZddZdd Zdd ZddZddZdS)GSpreadClientImplz)Concrete implementation of GSpreadClient.clientrenvipython_env.IPythonEnv | Nonec"||_||_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)r"r8r9s r__init__zGSpreadClientImpl.__init__Vs rrrcB |r,|j|S|r9|jt |S|r9|jt |Sn5#t$r(}td ||d}~wwxYwtd)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) namer<openkey open_by_keystrurl open_by_urlGSpreadExceptionrr)r"rexcs r_openzGSpreadClientImpl._openas axxzz 5|((444wwyy @|//CGGII???wwyy @|//CGGII??? @ a a a*+J+Q+QRU+V+VWW]` ` a&'KLLLs%?CA CA C D'#D  Drrc0||dSN)rIr!s rr#zGSpreadClientImpl.validatezs 3rr%r&r'c||||j|jfd}nfd}|fS)Nc dtjt jjdjjdS)Nz Reading inputs from worksheet {}{} in {}rEtext) display_htmlrr get_anchor_tagr SheetsURLrEtitle)r9sheet worksheetsr _display_fnz6GSpreadClientImpl.get_all_records.._display_fnst  6=="1 ) 3IM B B!+!2!29?EK!P!PrcbtdjjdS)Nz&Reading inputs from worksheet {} in {})printrrT)rUrVsrrWz6GSpreadClientImpl.get_all_records.._display_fns.>EEioW\Wbccdddddr)rI get_worksheetr=r*)r"rr%rWr9rUrVs @@@rr*z!GSpreadClientImpl.get_all_records}s  3'' 55   (#C         e e e e e e((**K77rr+r,c 6||}tj}d|dd|dd}||dd}|||jx|jd tj tj |j d |j |j  dStd |j |j dS) NzResults z%Y_%m_%dz (z%s))rTr+cols)valuesz#Results written to new worksheet {}rNrOz)Results written to new worksheet {} in {})rIdatetimenow add_worksheet append_rowsr=rQrr rRr rSrErTrY)r"rr+rUcurrent_datetimerTrVs rr/zGSpreadClientImpl.write_recordss'  3 $,0022N+NNN8HNNNN''e!!'DD T***   (   * *5<<-%/ >>'..y LL      =DDY_V[Vabb c c c c crN)r8rr9r:)rrr0r1r2) rrrr3r>rIr#r*r/rrrr7r7Ss33     MMMM288888ddddddrr7c*eZdZdZddZdd Zdd ZdS)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. rrrrctrKrr!s rr#zNullGSpreadClient.validates!!!rr%r&r'ctrKrhr)s rr*z!NullGSpreadClient.get_all_records  !!!rr+r,ctrKrhr.s rr/zNullGSpreadClient.write_recordsrjrNr0r1r2)rrrr3r#r*r/rrrrfrfsZ """"""""""""""rrfzGSpreadClient | None_gspread_clientcredscredentials.Credentialsr9r:rc~t'tj|}t||adSt adS)z Sets up credential for gspreads.N)r8r9)gspread authorizer7rlrf)rmr9r8s rrqrqs?"5))+6sCCC+--rc<tstdtS)NzMust call authorize() first)rlrrrr get_clientrss :8999 rr8c |adS)z(Overrides the global client for testing.N)rl)r8s rtestonly_set_clientrus OOOr)rr)rmrnr9r:rr)rr)r8rrr)#r3 __future__rr4r`typingrrrr google.authrgoogle.generativeai.notebookr r r r __annotations__rp ImportErrorer exceptionsrGrrrABCrr7rfrlrqrsrurrrrsC877"""""" 333333333333######333333444444222222 +/....NNNNGGGGGG !(99W5G5X     |   ````OOOOOCGOOOBbdbdbdbdbd bdbdbdJ""""" """4)-,,,,.... sAAAA