li/<zddlZddlZddlZddlZddlmZddlmZddlZddl Z ddl Z ddl Z ddlmZmZddlmZddlmZddlmZmZmZmZmZmZmZmZmZm Z ddl!m"Z"dd l#m$Z$m%Z%dd l&m'Z'dd l(m)Z)dd l*m+Z+d Z,dZ-ddgZ.ej^ej`ejbejdddZ3Gdde$Z4y)N)Path)Optional)get_include_paths load_plugin) read_profile) Identifier) ConfigFolderDirectoryInvalidProfileTemplateYAMLNoSampleProfileFound%ProfileWrittenWithProjectTemplateYAMLProfileWrittenWithSample$ProfileWrittenWithTargetTemplateYAMLProjectCreatedProjectNameAlreadyExistsSettingUpProfileStarterProjectPath) get_flags)BaseTaskmove_to_nearest_project_dir)_get_adapter_plugin_names) fire_event)DbtRuntimeErrorz3https://docs.getdbt.com/docs/configure-your-profilezhttps://community.getdbt.com/z __init__.py __pycache__)stringintfloatboolNceZdZdeddfdZdedefdZdedefd Zifd ed edefd Z defd Z dedefdZ d edefdZ dedefdZ dedefdZddeedefdZdZdefdZdeddfdZdefdZdedefdZdZy)InitTask project_namereturnNcddlm}tt|t j ||t j ty)Nr) PACKAGE_PATHdir)ignore)dbt.include.starter_projectr#rrshutilcopytreeignore_patterns IGNORE_FILES)selfr starter_project_directorys R/var/www/html/strategist-ai/venv_dbt/lib/python3.12/site-packages/dbt/task/init.pycopy_starter_repozInitTask.copy_starter_repo5s7  %*CDE %|F<.PQ@RTbcN $Y[%=%= >nAU U  '')+S13QGGD>1233+S1,QGGN+,,,SIZE[\ * *33,,s$E <E,&E8 E),E58Fprofile_templatetargetc tj|}|jD]Z\}}|jdr|ddj dd}t |j }djt|D cgc]\}} d|dzd | c} }d |d z} tj| tj } || dz } |j||| |}|jd r |||dd<|jdd} |jdd}|jdd}t|jdd}||rd|dndz}tj||| |||<]|Scc} }w)zEGenerate a target configuration from profile_template and user input._chooseN_ rC[] z Desired z option (enter a number)type_fixed hide_inputFdefaulthintrdz ())rhrgrd)copydeepcopyitems startswithreplacerJrMjoin enumerateclickpromptINTgenerate_target_from_inputgetclick_type_mapping)r,rYrZprofile_template_localkeyvalue choice_type option_listnv prompt_msgnumeric_choicechoicergrhrirdtexts r.rvz#InitTask.generate_target_from_inputfs!%/?!@0668 JC~~i(!!"gooc37 "5::<0 II)K:PQ$!Q1Q3%r!~QR";-/GHI"'juyy!I$^a%7888*3/7>>(+&+F3qr7O!&<!?J#ii 48G 99VT2D-eii.EFD$BtfA,B?D"',,g*4#F3K1 6 -RsF c~td5}tj|}ddd|dS#1swYdSxYw)z]Reads dbt_project.yml in the current directory to retrieve the profile name. dbt_project.ymlNprofile)rHrKrL)r,rT dbt_projects r.%get_profile_name_from_current_projectz.InitTask.get_profile_name_from_current_projectsC# $ ,..+K ,9%% ,9%%s/<rcjttjtdz }||i}|jr3t |d5}t j |xsi}|||<dddt |d5}t j||dddy#1swY5xYw#1swYyxYw)z~Given a profile, write it to the current project's profiles.yml. This will overwrite any profile with a matching name.rAr>NrD)rrrPr3rHrKrLdump)r,rr<rWprofilesrTs r. write_profilezInitTask.write_profiles!!9!9:T.=QQ '*  # # %'- 1>>!,2)0& 1 #S ) #Q IIh " # #  1 1  # #s B=B)B&)B2c|jdi}|jdi}|j||}|jdd}||i|d}|j||y)z?Create and write a profile using the supplied profile_template.fixedpromptsrZdev)outputsrZN)rwrvpopr)r,rYr<initial_targetrrZ target_namers r.$create_profile_from_profile_templatez-InitTask.create_profile_from_profile_templatesg)--gr:"&&y"500.Ijj51 *F3{K 7L1r0ct|t|d}|dz }|jrt|5}t j |}ddd|j |ttjtdz }tt|t|y|j||y#1swYwxYw)zCreate a profile without defaults using target's profile_template.yml if available, or sample_profiles.yml as a fallback.rprofile_template.ymlNrArE)rrr3rHrKrLrrrrPrrr4rX)r,r;r<rRprofile_template_pathrTrYrWs r.create_profile_from_targetz#InitTask.create_profile_from_targets G(1!4 ,/E E ' ' )+, 5#'>>!#4  5  5 56F U $Y[%=%= >nAU U  4%C0A,B   + +G\ B 5 5s CCcHtj}t|}||vS)z Validate that the specified profile exists. Can't use the regular profile validation routine because it assumes the project file exists )rrPr)r,r<r1 raw_profiless r.check_if_profile_existsz InitTask.check_if_profile_existss& !{// #L1 |++r0ctttjtdz }|jsy|xs|j }t |d5}t j|xsi}ddd|jvrtjd|d|d}|Sy#1swY:xYw)zUsing either a provided profile name or that specified in dbt_project.yml, check if the profile already exists in profiles.yml, and if so ask the user whether to proceed and overwrite it.rATr>Nz The profile z already exists in z. Continue and overwrite it?) rrrPr3rrHrKrLrMrsconfirm)r,r< profiles_filerTrresponses r.check_if_can_write_profilez#InitTask.check_if_can_write_profilesY[556n9MM ##%#St'Q'Q'S - % /~~a(.BH / 8==? *}}|n,? /!=?HO / /s B..B7c"td5}tj|}ddd|j|t t j t dz }tt|t|y#1swYdxYw)z9Create a profile using the project's profile_template.ymlrNrArE) rHrKrLrrrrPrr r4)r,r<rTrYrWs r.-create_profile_using_project_profile_templatez6InitTask.create_profile_using_project_profile_templatesw ( ) 1Q#~~a0  1 112BLQ !9!9:T.=QQ 1|#N_J` a  1 1s BBc Ltt}|stjj ddj t |Dcgc]\}}d|dzd|c}}zdzdz}tj|tj}||dz Scc}}w) z9Ask the user which adapter (database) they'd like to use.z&Which database would you like to use? rCr`rarbzO (Don't see the one you want? https://docs.getdbt.com/docs/available-adapters)z Enter a numberrc) rJrdbt exceptionsNoAdaptersAvailableErrorrqrrrsrtru)r,available_adaptersr~rrrs r.ask_for_adapter_choicezInitTask.ask_for_adapter_choices!";"=>!..99; ; 6iiI>P4QRDAq1QqSEA3RS Ta b# #  juyyA!.1"455 SsB c6tt|j|sytd}|j r |j |y|j}|j||y#t $rttYBwxYw)z"Set up a new profile for a project)r<Nr) rrrrr3r Exceptionr rr)r,r<rr;s r. setup_profilezInitTask.setup_profiles#%&..L.I !%%; < ' ' ) 9BB<P--/ ''l'K 9578 9sA99BBcfddlm}|jj}|h}t t }|D]}|j d|tj|r||vrJ|rtj|dztjd}tj|sE||vrJ|S)zAReturns a valid project name, either from CLI arg or user prompt.r) PROJECT_NAMEdbt_z is not a valid project name.z;Enter a name for your project (letters, digits, underscore)) dbt.include.global_projectrargsr rJrupdate ProjectNameis_validrsechort)r,GLOBAL_PROJECT_NAMErFinternal_package_namesr adapter_names r.get_valid_project_namezInitTask.get_valid_project_names Syy%%"5!6!";"=>. AL " ) )D*? @ A&&t,8N0N 4"AAB<< ]^D&&t,8N0N  r0cx|j|tj|tdd5}|j j ||}dddtdd5}|j dddtt|tty#1swYOxYw#1swY5xYw)Nrr>)r r<rD)r docs_url slack_url) r/oschdirrHrIformatrQrrDOCS_URL SLACK_URL)r,r r<rTcontents r.create_new_projectzInitTask.create_new_projects |,  #S ) aQ ** S_*`G a #S ) Q GGG   )!#   a a  s#B$*B0$B-0B9c@tj}|j| t|jj d}|r[|jjr td|jjs"|j}|j|yy|j}t|}|jrt!t#|y|jj}|r?|j%|stdj'||j)||y|}|j)|||jjs|j|yy#t jj$rd}YTwxYw)zEntry point for the init task.TFzRCan not init existing project with specified profile, edit dbt_project.yml instead)msg)rFNz!Could not find profile named '{}')rrPr:rr project_dirr5rrrskip_profile_setuprrrrr3rrrrr)r,r1 in_projectr<r project_pathuser_profile_names r.runz InitTask.run)ss {//    .  ' (=(= >J yy  %l 99//#IIK ""<00  668L -L""$3FG!% 1 1  334EF)?FFGXY'' 6GH+ '' lCyy33&&|44Q$$44 J s!E::FF)N)__name__ __module__ __qualname__r4r/rr:rXdictrvrrrrrrrrrrrrrr0r.rr4s  c  d  #S:QS4W[@&s&#T##"2T2Y\2C#CSC.,C,D,x}PT& 66 L#L$L&$  s  #  15r0r)5rlrrNr(pathlibrtypingrrsrK dbt.configrdbt_common.clients.systemr5dbt.adapters.factoryrrdbt.config.profilerdbt.contracts.utilrrdbt.events.typesr r r r r rrrrr dbt.flagsr dbt.task.baserr dbt.versionrdbt_common.events.functionsrdbt_common.exceptionsrrrr+STRINGruFLOATBOOLrxrrr0r.rs   ?+8    ?121 @ + }- ll 99 [[ JJ  f5xf5r0