edZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZmZmZmZej(eZgdZdZd Zdd Zdd Zd Zd ZdZdddefdZy)zCFunctions for discovering and executing various cookiecutter hooks.N)Path)UndefinedError)utils)FailedHookException)create_env_with_contextcreate_tmp_repo_dirrmtreework_in) pre_promptpre_gen_projectpost_gen_projectctjj|}tjj|d}||k(}|tv}|j d}|xr|xr| 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.12/site-packages/cookiecutter/hooks.py valid_hookr sgww *Hww)!,H )M'N##C(K  ?^ ?K?c tjdtjj |tjj |stjdyg}tj |D][}t||s|jtjj tjj||]t|dk(ry|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*0s LL"BGGOOI$>? 77== # 34GZZ *P i + NN277??277<< 9+MN OP 7|q Nrctjjd}|jdrtj|g}n|g}t j | tj|||}|j}|tk7rtd|dy#t$r>}|jtjk(r td|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_pathr.run_thru_shellscript_commandproc exit_statuserrs r run_scriptrBLs \\,,U3NE"..+6% +& Qn#Niik , &%3K=B  ' Q 99 %%Q "$?uA"FGSP QsAB C%'9C  C%ctjj|\}}t|d5}|j }dddt j dd|5}t|}|j} | jdi|} |j| jddddtj|y#1swYxYw#1swY,xYw)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)deletemodesuffix)rrropenreadtempfileNamedTemporaryFiler from_stringrenderwriteencoderBname) r<r.context_ extensionfilecontentstempenvtemplateoutputs rrun_script_with_contextr\is77##K0LAy kG ,99;  $ $EY O+SW%g.??8, +7+ 6==)* + tyy#++sC!ACCC$ct|}|stjd|ytjd||D]}t|||y)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_dirrSr)scripts rrun_hookr`~sM "G  '3 LL"I.> W=>rct|5 t||| dddy#ttf$r%|r t |t j d|wxYw#1swYyxYw)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 r`rrr r error)repo_dirrr^rSdelete_project_on_failures rrun_hook_from_repo_dirresv    Y W 5  ){# LL2   sA %4AAA%rczos.PathLike[str]returnc@t|5td}|s |cdddS dddt|}t|5td}|D]} t|| ddd|S#1swYJxYw#t$r t dwxYw#1swY|SxYw)zeRun pre_prompt hook from repo directory. :param repo_dir: Project template input directory. r NzPre-Prompt Hook script failed)r r*rrBr)rcr)r_s rrun_pre_prompt_hookrhs  L) #8,H  KL) KF K68, KK O' K)*IJJ K K Os/A/B A;"B/A8;BBB)hooks).) __doc__r:loggingrr5r0rLpathlibrjinja2.exceptionsr cookiecutterrcookiecutter.exceptionsrcookiecutter.utilsrrr r getLogger__name__r rr8rr*rBr\r`rerhrIrrrtsI   ,7   8 $    @ 8Q:*>":"4r