7VfR>ddlZddlZddlmZmZddlZddlZddlZddlZddl Z ddlm Z Gdde Zejej zdfdZdZy#e $re Z Y5wxYw)N)LockThread)JSONDecodeErrorc4eZdZdZdej j ddrdndzZdZe dZ ddZ dZ e d Z d Zd Zd Zd ZdZedZedZej,dZedZej,dZdZdZdZy) BaseScope) --disable-gpuz--allow-file-access-from-filesz--disable-breakpadz--disable-dev-shm-usageLAMBDA_RUNTIME_DIRN)z--single-processc4|j|jzS)a~ Get tuple containing the default chromium arguments that will be passed to chromium if not overridden. chromium arguments can be overridden in the Scope constructor using the chromium_args argument, or they can be overridden by assigning a tuple to the chromium_args property of an already constructed Scope instance :return: tuple of str )_default_chromium_args_scope_chromium_args)clss \/var/www/html/software/conda/envs/catlas/lib/python3.12/site-packages/kaleido/scopes/base.pydefault_chromium_argszBaseScope.default_chromium_args$s))C,D,DDDc|dur|j}n|durd}|dur|Dcgc] }|dk7s | }}t||_tj|_d|_d|_t|_ ycc}w)NTFrr ) rtuple_chromium_argsioBytesIO _std_error_std_error_thread_procr _proc_lock)self disable_gpu chromium_argsargs r__init__zBaseScope.__init__0s D  668M e #M % ,9TSSO=SSTMT#M2**,!% &Us A<A<c$|jyN)_shutdown_kaleidors r__del__zBaseScope.__del__Gs  rc jtjjtjjtjjtjj t dd}t jdk(r|dz }tjj|r|}|Std}|jtjjdtj}|jtjd}tdj!|||S) N executablekaleidoWindowsz.cmdPATHz aC The kaleido executable is required by the kaleido Python library, but it was not included in the Python package and it could not be found on the system PATH. Searched for included kaleido executable at: {vendored_executable_path} Searched for executable 'kaleido' on the following system PATH: {formatted_path} )vendored_executable_pathformatted_path)ospathjoindirnameabspath__file__platformsystemexistswhichenvirongetdefpathreplacepathsep ValueErrorformat)rr*executable_pathr-r+s rr=zBaseScope.executable_pathJs#%77<< GGOOBGGOOBGGOOH,EF G  $   ??  ) $ . $ 77>>2 37O0'$I.O&zz~~fbjj9!%bjj(!C !&1I'5!  rcl|j|jg}|jD]m}t||}|durd|j ddz}n4|dus|0d|j ddzdzt t |z}|j|o|j|j|S)z Build list of kaleido command-line arguments based on current values of the properties specified by self._scope_flags and self.chromium_args :return: list of flags Tz--_-F=) r= scope_name _scope_flagsgetattrr9reprstrappendextendr)r proc_argskvflags r_build_proc_argszBaseScope._build_proc_argsts))+T__= "" #Aa ADyaiiS11eqyaiiS11C7$s1v,F   T " # ++,rc |j@|jjj}|jj |nyN)z Write standard-error of subprocess to the _std_error StringIO buffer. Intended to be called once in a background thread N)rstderrreadlinerwritervals r_collect_standard_errorz!BaseScope._collect_standard_errorsE zz%jj''002%%c*rc |j|jj|j5|j|jj|j|jjt j |_|j}tj|tjtjtjtjdk(|_|j|jjsPt|j |_ |jj#d|jj%|jj&j)j+d}|sd|j-z}t/|t1j2|}|j5dd d k7r5|jjt/|j5d d dddyy#1swYyxYw) z` Launch the kaleido subprocess if it is not already running and in a good state Nwin32)stdinstdoutrOshell)targetTutf-8z3Failed to start Kaleido subprocess. Error stream: codermessagez"Failed to start Kaleido subprocess)rpollrwaitrrrrM subprocessPopenPIPEsysr2ris_aliverrT setDaemonstartrXrPdecode_get_decoded_std_errorr;jsonloadsr7)rrIstartup_response_stringr]startup_responses r_ensure_kaleidozBaseScope._ensure_kaleidos :: !2!>( t::%):)Fzz- )')jjlDO !% 5 5 7I!+!1!1!(oo))!llg5 "DJ--5T=S=S=\=\=^17t?[?[1\...88>..446/3jj.?.?.H.H.J.Q.QRY.Z+2S 779: )11+/::6M+N(+//:a? JJOO-",-=-A-A)Mq-r"ssQ( t( t"?( t( ts G9H77IcXd} tjj}|jj j |}|? tjd}|jj j |}|d}|S#t $rYRwxYw#t $rY wxYw)zK Attempt to decode standard error bytes stream to a string NFz1Failed to decode Chromium's standard error stream) rcrOencodingrgetvaluerg Exceptionlocalegetpreferredencoding)r std_err_strros rrhz BaseScope._get_decoded_std_errors  zz**H//224;;HEK   !66u="oo668??I   MK     s$AB >B BB B)(B)cJ|j|j5|jb|jj$|jjj  |jj dd|_dddyy#YxYw#1swYyxYw)z_ Shut down the kaleido subprocess, if any, and self the _proc property to None Ng@)timeout)rrr^rWcloser_r#s rr"zBaseScope._shutdown_kaleidos :: ! &::)zz(0 ((..0  4 "&DJ# & & " & &s$A B%BBBBB"ctr!)NotImplementedErrorr#s rrBzBaseScope.scope_names!!rcd|jvS)zT If True, asks chromium to disable GPU hardware acceleration with --disable-gpu flagr )rr#s rrzBaseScope.disable_gpus$"4"444rc|jDcgc] }|dk7s | }}|r|jdt||_ycc}w)Nr )rrGr)rrSrnew_argss rrzBaseScope.disable_gpusC#'#5#5PC9OCPP  OOO ,"8_Qs AAc|jSr!)rr#s rrzBaseScope.chromium_argss"""rcDt||_|jyr!)rrr"rRs rrzBaseScope.chromium_argss#Cj  rc,tj|Sr!)ridumpsrRs r _json_dumpszBaseScope._json_dumpsszz#rc |j|jt||jd}|j5t j |_|jjj||jjjdjd|jjj|jjj}dddjd}|sd|jz}t!| t#j$|}|S#1swYRxYw#t&$rt)dt+|zwxYw)aS Transform input data using the current scope, returning dict response with error code whether successful or not. :param data: JSON-serializable object to be converted :param kwargs: Transform arguments for scope :return: Dict of response from Kaleido executable, whether successful or not )datar[ Nz!Transform failed. Error stream: zInvalid JSON: )rmrdictencoderrrrrrWrQflushrXrPrgrhr;rirjrprintrE)rrkwargs export_specresponseresponse_stringr]s r_perform_transformzBaseScope._perform_transformsA &&tF'>?FFwO __ 4 jjlDO JJ   " "; / JJ   " "4;;w#7 8 JJ   " " $zz((113H 4#//'29//12 W% % zz/2H / 4 4&  "T/%:: ;  sB;E=E E "Fc |j|fi|}|jdd}|dk7r.|jdd}tdj |||jdd}|j S)a Transform input data using the current scope Subclasses should provide a more helpful docstring :param data: JSON-serializable object to be converted :param kwargs: Transform arguments for scope :return: Transformed value as bytes r\rr]Nz2Transform failed with error code {code}: {message})r\r]result)rpopr7r;r<r)rrrrr\r] img_strings r transformzBaseScope.transformEs+4**4:6:||FA& 19ll9d3GDKKwL  \\(D1   ""r)TT)__name__ __module__ __qualname__rCr,r6r7r r classmethodrrr$r=rMrTrmrhr"propertyrBrsetterrrrrrrrrrsL"$0Dd!KQS  E E!.!''R2 /tb.&4""55-- ##!!(T#rrchd}tjjr ||rSy|.tjj dtj }|sy|j tj}tjdk(rtj|vr |jdtjtjj ddj tj}tfd|Drg}n|Dcgc]}|z }}ng}t}|D]m}tjj|} | |vs'|j| |D]1} tjj!|| } || |s-| ccSoycc}w) a Backport (unmodified) of shutil.which command from Python 3.6 Remove this when Python 2 support is dropped Given a command, mode, and a PATH string, return the path which conforms to the given mode on the PATH, or None if there is no such file. `mode` defaults to os.F_OK | os.X_OK. `path` defaults to the result of os.environ.get("PATH"), or can be overridden with a custom search path. ctjj|xr8tj||xr tjj | Sr!)r,r-r4accessisdir)fnmodes r _access_checkz which_py2.._access_checkps;ww~~b!SbiiD&9S"''--PRBS>SSrNr)rVrPATHEXTc3xK|]1}jj|j3ywr!)lowerendswith).0extcmds r zwhich_py2..s(DSsyy{##CIIK0Ds7:)r,r-r/r6r7r8splitr:rcr2curdirinsertanysetnormcaseaddr.) rrr-rpathextfilesrseendirnormdirthefilenames ` r which_py2r`sa T  wws d #J |zz~~fbjj1  ::bjj !D ||wyyD KK299 %**..B/55bjjA DGD DEE*123S3Y2E2 5D ''""3'$ HHW   ww||C1 t,K   3s F/chtjdkDrddl}|j|St |S)a Return the absolute path of the input executable string, based on the user's current PATH variable. This is a wrapper for shutil.which that is compatible with Python 2. Parameters ---------- cmd: str String containing the name of an executable on the user's path. Returns ------- str or None String containing the absolute path of the executable, or None if the executable was not found. )rrN)rc version_infoshutilr5r)rrs rr5r5s.& & ||C  ~r)r`ri threadingrrrr,rcrrr2r ImportErrorr;objectrF_OKX_OKrr5rrrrsj " !$ L#L#` "'')>Bk ! O!sAA"!A"