U L{jQ<@sddlZddlZddlmZmZddlZddlZddlZddlZzddlm Z Wne k rhe Z YnXGddde Z ejejBdfddZdd ZdS) N)LockThread)JSONDecodeErrorc@seZdZdZdZdZdZeddZd#ddZ d d Z ed d Z d dZ ddZ ddZddZddZeddZeddZejddZeddZejddZdd Zd!d"ZdS)$ BaseScopeN) --disable-gpuz--allow-file-access-from-filesz--disable-breakpadz--disable-dev-shm-usagecCs |j|jS)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_argsclsrrR/var/www/html/bwave/python/venv/lib/python3.8/site-packages/kaleido/scopes/base.pydefault_chromium_args s zBaseScope.default_chromium_argsTcCs`|dkr|}n |dkrd}|dkr4dd|D}t||_t|_d|_d|_t|_ dS)NTFrcSsg|]}|dkr|qSrr.0argrrr 9sz&BaseScope.__init__..) r tuple_chromium_argsioBytesIO _std_error_std_error_thread_procr _proc_lock)self disable_gpu chromium_argsrrr __init__,s   zBaseScope.__init__cCs |dSN)_shutdown_kaleidorrrr __del__CszBaseScope.__del__c Cs~tjtjtjtjtdd}tj|r:|}n@td}|dkrztj dtj }| tj d}t dj||d|S)N executableZkaleidoPATHz 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__existswhichenvirongetdefpathreplacepathsep ValueErrorformat)r r%executable_pathr(r&rrr r6Fs$  zBaseScope.executable_pathcCs||jg}|jD]f}t||}|dkr|jdk r |jdk rR|jt|_|}tj |tj tj tj t j dkd|_|j dks|j st|jd|_ |j d|j |jjd}|sd|}t|n6t|}|dd d kr |jt|d d W5QRXdS) z` Launch the kaleido subprocess if it is not already running and in a good state Nwin32)stdinstdoutrFshell)targetTutf-8z3Failed to start Kaleido subprocess. Error stream: codermessagez"Failed to start Kaleido subprocess)rpollrwaitrrrrE subprocessPopenPIPEsysplatformris_aliverrK setDaemonstartrNrGdecode_get_decoded_std_errorr4jsonloadsr0)rrAZstartup_response_stringrSZstartup_responserrr _ensure_kaleidos:          zBaseScope._ensure_kaleidocCsd}ztjj}|j|}Wntk r4YnX|dkrrztd}|j|}Wntk rpYnX|dkr~d}|S)zK Attempt to decode standard error bytes stream to a string NFz1Failed to decode Chromium's standard error stream) rYrFencodingrgetvaluer^ Exceptionlocalegetpreferredencoding)rZ std_err_strrcrrr r_s z BaseScope._get_decoded_std_errorc Csj|jdk rf|jP|jdk r\|jdkr6|jjz|jjddWn YnXd|_W5QRXdS)z_ Shut down the kaleido subprocess, if any, and self the _proc property to None Ng@)timeout)rrrTrMcloserUr!rrr r s   zBaseScope._shutdown_kaleidocCstdSr)NotImplementedErrorr!rrr r:szBaseScope.scope_namecCs d|jkS)zT If True, asks chromium to disable GPU hardware acceleration with --disable-gpu flagr)rr!rrr rszBaseScope.disable_gpucCs,dd|jD}|r|dt||_dS)NcSsg|]}|dkr|qSrrrrrr rsz)BaseScope.disable_gpu..r)rr?r)rrJnew_argsrrr rs cCs|jSr)rr!rrr rszBaseScope.chromium_argscCst||_|dSr)rrr rIrrr r s c Ks|tjt||d|jdd}|jJt|_ |j j ||j j dd|j j |j j}W5QRX|d}|sd|}t|zt|}Wn&tk rtdt|YnX|S)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 rQ z!Transform failed. Error stream: zInvalid JSON: )rbr`dumpsdict _json_encoderencoderrrrrrMrHflushrNrGr^r_r4rarprintr=)rrlkwargsZ export_specresponseZresponse_stringrSrrr _perform_transforms4     zBaseScope._perform_transformcKsT|j|f|}|dd}|dkr@|dd}tdj||d|dd}|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 rRrrSNz2Transform failed with error code {code}: {message})rRrSresult)rvpopr0r4r5rq)rrlrtrurRrSZ img_stringrrr transform:s    zBaseScope.transform)TT)__name__ __module__ __qualname__rpr;rr classmethodr rr"r6rErKrbr_r propertyr:rsetterrrvryrrrr rs:   #1     ,rc s"dd}tjr&||r"SdS|dkr>tjdtj}|sFdS|tj}tj dkrtj |krt| dtj tjddtj}t fd d |Drg}q‡fd d |D}ng}t }|D]P}tj|}||kr|||D](} tj|| } || |r| SqqdS) 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. cSs&tj|o$t||o$tj| Sr)r'r(r-accessisdir)fnmoderrr _access_checkesz which_py2.._access_checkNr$rLrPATHEXTc3s |]}|VqdSr)lowerendswithrextcmdrr szwhich_py2..csg|] }|qSrrrrrr rszwhich_py2..)r'r(r*r/r0r1splitr3rYrZcurdirinsertanysetnormcaseaddr)) rrr(rpathextfilesseendirnormdirthefilenamerrr which_py2Us8        rcCs(tjdkrddl}||St|SdS)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)rY version_infoshutilr.r)rrrrr r.s  r.)rVr` threadingrrrr'rYrfr ImportErrorr4objectrF_OKX_OKrr.rrrr s HA