U ޣ5hF@sNddlZddlZddlZddlmZddlmZmZmZm Z GdddZ dS)N) BeautifulSoup)AZURE_DEVOPS_URLAZURE_DEVOPS_ORGAZURE_DEVOPS_PROJECTAZURE_DEVOPS_PATc@s eZdZdddZdddZdS) AzureClientNcCs|r|dtnt|_|r&|dtnt|_|r<|dtnt|_|rR|dtnt|_|jr|j dsvd|j|_|j d|_dS)NurlorgZprojectpat)zhttp://https://r /) getr azure_urlr azure_orgr azure_projectr azure_pat startswithrstrip)self azure_configrK/var/www/html/testcasegenerator.evdpl.com/azure_integration/azure_client.py__init__s  zAzureClient.__init__c Cs|s$tddd}dd|D}|s4tddS|jsFtddS|jsXtddS|jsjtd dSg}|D] }|jd |jd |jd |d }d dtd|j   d}zt j ||d}|jdkr>|}dd}| di dd} || } | di dd} ||| | dtd|ntd|d|jWqrtk r} ztd|dt| W5d} ~ XYqrXqr|S)NZAZURE_DEVOPS_WORKITEM_IDS,cSsg|]}|r|qSr)strip).0idrrr sz6AzureClient.fetch_azure_work_items..uW⚠️ Work item IDs not found. Please set AZURE_DEVOPS_WORKITEM_IDS in your .env file.u$❌ Azure DevOps URL cannot be emptyu-❌ Azure DevOps organization cannot be emptyu(❌ Azure DevOps project cannot be emptyr z/_apis/wit/workitems/z?api-version=6.0zapplication/jsonzBasic :)Accept Authorization)headerscSst|d}|S)Nz html.parser)rZget_text)textZsouprrr clean_html<s z6AzureClient.fetch_azure_work_items..clean_htmlfieldszSystem.DescriptionzNo Description Foundz System.TitlezNo Title Found)rtitle descriptionu#✅ Successfully fetched work item u❌ Failed to fetch work item z: u❌ Error processing work item )osgetenvsplitprintrrrbase64 b64encoderencodedecoderequestsr status_codejsonappend Exceptionstr) rZ work_item_idsresultsZ work_item_idrr"response work_itemr%r(Zdescription_cleanedr'errrfetch_azure_work_itemssN " ,z"AzureClient.fetch_azure_work_items)N)N)__name__ __module__ __qualname__rr;rrrrrs r) r)r1r-Zbs4rconfig.settingsrrrrrrrrrs