iUddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z ddlmZmZddlmZddlmZddlmZd Ze e gZeZeeed <d efd ZGd deZedfdZ dZ!y)N) BaseSettings) UnsetProfile load_project)MANIFEST_FILE_NAMERUN_RESULTS_FILE_NAME)ArtifactUploadSkippedArtifactUploadSuccess)DbtProjectError) fire_event)DbtBaseExceptionPRODUCED_ARTIFACTS_PATHS artifact_pathc.tj|yN)radd)rs Z/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/dbt/utils/artifact_upload.pyadd_artifact_producedrs  /cJeZdZUeed<eed<eed<eed<dZdZd dZy) ArtifactUploadConfigtenant_hostnameDBT_CLOUD_TOKENDBT_CLOUD_ACCOUNT_IDDBT_CLOUD_ENVIRONMENT_IDcVd|jd|jd|jdS)Nzhttps://z/api/private/accounts/z/environments/z /ingests/)rrr)selfs rget_ingest_urlz#ArtifactUploadConfig.get_ingest_url"sL$..//EdF_F_E``nospMpMoNNWX Xrc,|j|dS)N/)r)r ingest_ids rget_complete_urlz%ArtifactUploadConfig.get_complete_url%s%%'( 155rNcf|ttj}d|d|jdS)Nzapplication/jsonzToken )AcceptzX-Invocation-Id Authorization)struuiduuid4r)r invocation_ids r get_headersz ArtifactUploadConfig.get_headers(s9   -M(,%d&:&:%;<  rr)__name__ __module__ __qualname__r&__annotations__rr"r*rrrrs*!!X6 rrc z|gd}d}t|dzD]c} |\}}|r|cS|j|vrtd|d|||k(rtd|d| t j||dz}ey#tj$r)}||k(rtd|dt |Yd}~Td}~wwxYw)aExecute a function with exponential backoff retry logic. Args: operation_name: Name of the operation for error messages func: Function to execute that returns (success, result) max_retries: Maximum number of retry attempts Returns: The result from the function if successful Raises: DbtException: If all retry attempts fail N)iiiizError z: )range status_code DbtExceptionrequestsRequestExceptionr&timesleep) operation_namefunc max_retries retry_codes retry_delayattemptsuccessresultes r_retry_with_backoffrC2s* Kq) H"fOGV !!4"VN+;2fX#FGG+%"VN+;2fX#FGG& ;q (( H+%"VN+;2c!fX#FGG& Hs A>5A>>B:B55B:ctsttdy t|dt d}|j rd|j vr t d|j d}|s t d t| |sd }d }tj|d 5}tD]&}|j||jd d( dddfd} td| } | j} | dd| ddt!|d5} | j#fd} td| dddfd}td|tt%d|dt&j(j*!t&j(j-d|itj/y#t$r}t dt|d}~wwxYw#1swYxYw#1swYxYw)Nz*No artifacts to upload for current command)msgF) version_checkprofilecli_varsrz6dbt_cloud.tenant_hostname not found in dbt_project.ymlz5dbt_cloud.tenant_hostname is empty in dbt_project.ymlz>Error reading dbt_cloud.tenant_hostname from dbt_project.yml: )rtargetz target.zipwr ctjjj}|jdk(|fS)N)urlheaders)r6postrr*r4)responseconfigs r create_ingestz'upload_artifacts..create_ingestvs;==V%:%:%.upload_files)&llz JO"..*.complete_ingestsM$NN'' 2&&(!9-  !,,35FFFrzcompleting ingestzcommand z completed successfullycommand)rr rrr dbt_cloudr Exceptionr&rzipfileZipFilewritesplitrCraopenreadr dbttracking active_usertrack_artifact_uploadclear) project_dir target_pathreprojectrrB zip_file_namezrrSrQ response_datafr]rdrRr\r!rVs @@@@rupload_artifactsrzUs #(-YZ[   ulnt   $5W=N=N$N!"Z[ [!++,=>!"YZ Z"/ BF  !M  ,A5 AM GGM=#6#6s#;B#? @ AA 5##rsx +9CI*2B )+@A%(U#c(*00 < ,;FSW FB%r