E^edZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZddlmZmZmZejeZgd ZdZd Zdd ZddZdZdZdZdddefdZ dS)zCFunctions for discovering and executing various cookiecutter hooks.N)Path)UndefinedError)utils)StrictEnvironment)FailedHookException)create_tmp_repo_dirrmtreework_in) pre_promptpre_gen_projectpost_gen_projectctj|}tj|d}||k}|tv}|d}|o|o| S)zDetermine if a hook file is valid. :param hook_file: The hook file to consider for validity :param hook_name: The hook to find :return: The hook file validity r~)ospathbasenamesplitext_HOOKSendswith) hook_file hook_namefilenamer matching_hooksupported_hook backup_files 2lib/python3.11/site-packages/cookiecutter/hooks.py valid_hookrslw **Hw))!,H )M'N##C((K  ?^ ?K?hooksc tdtj|tj|stddSg}tj|D]b}t||rP|tjtj ||ct|dkrdS|S)aReturn a dict of all hook scripts provided. Must be called with the project template as the current working directory. Dict's key will be the hook/script's name, without extension, while values will be the absolute path to the script. Missing scripts will not be included in the returned dict. :param hook_name: The hook to find :param hooks_dir: The hook directory in the template :return: The absolute path to the hook script or None zhooks_dir is %szNo hooks/dir in template_dirNr) loggerdebugrrabspathisdirlistdirrappendjoinlen)r hooks_dirscriptsrs r find_hookr++s LL"BGOOI$>$>??? 7== # # 3444tGZ **PP i + + P NN27??27<< 9+M+MNN O O O 7||qt Nr.ctjd}|drtj|g}n|g}t j| tj|||}| }|tkrtd|ddS#t$r>}|j tjkrtd|td|d|d}~wwxYw) zExecute a script from a working directory. :param script_path: Absolute path to the script to run. :param cwd: The directory to run the script from. winz.py)shellcwdz!Hook script failed (exit status: )z?Hook script failed, might be an empty file or missing a shebangzHook script failed (error: N)sysplatform startswithr executablermake_executable subprocessPopenwait EXIT_SUCCESSrOSErrorerrnoENOEXEC) script_pathr0run_thru_shellscript_commandproc exit_statuserrs r run_scriptrDGs \,,U33NE""'.+6% +&&& Qn#NNNiikk , & &%BKBBB  ' & QQQ 9 % %%Q ""F"F"F"FGGSP QsA B'' C/19C**C/ctj|\}}t|d5}|}dddn #1swxYwYt jdd|5}t|d}||} | j d i|} | | ddddn #1swxYwYt|j |dS) zExecute a script after rendering it with Jinja. :param script_path: Absolute path to the script to run. :param cwd: The directory to run the script from. :param context: Cookiecutter project template context. zutf-8)encodingNFwb)deletemodesuffixT)contextkeep_trailing_newline)rrropenreadtempfileNamedTemporaryFiler from_stringrenderwriteencoderDname) r>r0rK_ extensionfilecontentstempenvtemplateoutputs rrun_script_with_contextr_dsn7##K00LAy kG , , ,99;;  $EY O O O+SWtLLL??8,, ++7++ 6==))*** +++++++++++++++ ty#s$AAA7ACC#&C#ct|}|std|dStd||D]}t|||dS)z Try to find and execute a hook from the specified project directory. :param hook_name: The hook to execute. :param project_dir: The directory to execute the script from. :param context: Cookiecutter project context. zNo %s hook foundNzRunning hook %s)r+r!r"r_)r project_dirrKr*scripts rrun_hookrcysv ""G  '333 LL"I...>> W====>>rct|5 t|||nB#ttf$r.|rt |t d|wxYw ddddS#1swxYwYdS)a|Run hook from repo directory, clean project directory if hook fails. :param repo_dir: Project template input directory. :param hook_name: The hook to execute. :param project_dir: The directory to execute the script from. :param context: Cookiecutter project context. :param delete_project_on_failure: Delete the project directory on hook failure? zCStopping generation because %s hook script didn't exit successfullyN)r rcrrr r!error)repo_dirrrarKdelete_project_on_failures rrun_hook_from_repo_dirrhs     Y W 5 5 5 5     ) ${### LL2      6s%A4$A4?A##A44A8;A8rfzos.PathLike[str]returnct|5td}|s|cdddS dddn #1swxYwYt|}t|5td}|D]0} t||#t$rt dwxYw dddn #1swxYwY|S)zeRun pre_prompt hook from repo directory. :param repo_dir: Project template input directory. r NzPre-Prompt Hook script failed)r r+rrDr)rfr*rbs rrun_pre_prompt_hookrks   L))   #8,,H   KKL)) K KF K68,,,,& K K K)*IJJJ K KKKKKKKKKKKKKKKK Os9=AA&B6;B  B6 B&&B66B:=B:)r)r,)!__doc__r<loggingrr7r2rPpathlibrjinja2.exceptionsr cookiecutterrcookiecutter.environmentrcookiecutter.exceptionsrcookiecutter.utilsrr r getLogger__name__r!rr:rr+rDr_rcrhrkrMrrrvs~II   ,,,,,,666666777777CCCCCCCCCC  8 $ $      @ @ @ 8QQQQ:*>>>":"4r