U h(@svddlmZmZmZmZddlmZmZddlZddl Z ddl Z ddl m Z ddl mZmZmZmZGdddZdS))capture_exceptioncapture_messageset_tag set_context)monitor_azure_apimonitor_critical_systemN) BeautifulSoup)AZURE_DEVOPS_URLAZURE_DEVOPS_ORGAZURE_DEVOPS_PROJECTAZURE_DEVOPS_PATc@sDeZdZd ddZed ddZedddZdeed d d Z dS) AzureClientNc CsXtd||r|dt|_|dt|_|dt|_|dt|_ td|jd|jd|jd |j rzd t |j nd n^|pt|_|pt|_d|_|pt|_ td |jd|jd|jd |j rd t |j nd d|_ |jr*t |jt r*|jd sd|j|_|jd|_n*t |jt sTtdt|jdd|_dS)Nu7🔧 AzureClient constructor called with: azure_config=urlorgprojectpatu 🔧 Using azure_config - URL: ' ', Org: ' ', Project: '', PAT: *Noneu!🔧 Using direct params - URL: ')zhttp://https://r/u❌ Invalid azure_url type: z, expected string)printgetr azure_urlr azure_orgr azure_projectr azure_patlen last_error isinstancestr startswithrstriptype)selfrrr azure_configr(K/var/www/html/testcasegenerator.evdpl.com/azure_integration/azure_client.py__init__ s(<   : zAzureClient.__init__c s|s$tddd}dd|D}|s4tddStd|jd|jd |jd |jrdd t|jnd |jsd |_ tddS|jsd|_ tddS|jsd|_ tddS|jsd|_ tddSg}d}ddt d|j  ddd}tdt||D]}||||}ddd|D}|jd|jd|jd|d ztd!d"fd#d$}|} | jdkr| pi} | d%g} | D]N} t| d&} | d'i}|d(d)}|d*d+}|| |||d,qtd-t| d.n&d/| j}||_ td0|d1|Wn`tk r}z@d2t|}||_ td0|t|||j|j|jd3W5d}~XYnXq|S)4NZAZURE_DEVOPS_WORKITEM_IDS,cSsg|]}|r|qSr()strip).0idr(r(r) 0sz6AzureClient.fetch_azure_work_items..uW⚠️ Work item IDs not found. Please set AZURE_DEVOPS_WORKITEM_IDS in your .env file.u%🔍 Validating Azure fields - URL: 'rrrrrz Azure DevOps URL cannot be emptyu$❌ Azure DevOps URL cannot be emptyz)Azure DevOps organization cannot be emptyu-❌ Azure DevOps organization cannot be emptyz$Azure DevOps project cannot be emptyu(❌ Azure DevOps project cannot be emptyz2Azure DevOps Personal Access Token cannot be emptyu6❌ Azure DevOps Personal Access Token cannot be emptyapplication/jsonBasic :Accept AuthorizationcSst|pdd}|S)Nr+z html.parser)rZget_text)textZsoupr(r(r) clean_htmlUsz6AzureClient.fetch_azure_work_items..clean_htmlrcSsg|] }t|qSr()r")r.xr(r(r)r0[srz/_apis/wit/workitems?ids=z&api-version=6.0T)criticalcstjddS)Nheaderstimeout)requestsrr(r>rr(r)make_azure_requestbsz>AzureClient.fetch_azure_work_items..make_azure_requestvaluer/fieldsz System.TitlezNo Title FoundzSystem.DescriptionzNo Description Found)r/title descriptionu✅ Successfully fetched z work items in batchz"Failed to fetch work items batch: u❌ z - ids: z#Error processing work items batch: ) work_item_idsrrr)osgetenvsplitrrrrrrr base64 b64encodeencodedecoderangejoinr status_codejsonrr"append Exceptionr)r&rGresultsZ CHUNK_SIZEr9iZ chunk_idsZ ids_paramrBresponsebodyitemsZwiZwidrDrErF error_msger(rAr)fetch_azure_work_items,s:        z"AzureClient.fetch_azure_work_items) project_namec Cszb|jd|jd|d}ddtd|jd}tj||dd }| | WSt k r}zt d |d |WYd Sd }~XYnXd S) zGet project informationrz/_apis/projects/?api-version=6.0r2r3r4r5r<r=u❌ Failed to get project : N) rrrKrLrrMrNr@rraise_for_statusrRrTr)r&r]rr>rWr[r(r(r) get_projects zAzureClient.get_project)r]limitc Cs:ztd||jd|jd|d}dddtd|jd}d|d g}|rt|t t frt |d krd d d |D}| d|dd |}d|d} d| i} td|tj||| dd} td| j| jdkrtd| jd| jgWS| | } | dg} tdt | dg}|dkr`| n | d|}|D]p}|d}|jd|jd|d |d!}tj||dd"}|jdkr| |ntd#|d$|jqptd%t |d&|WStk r4}ztd'|gWYSd}~XYnXdS)(a4Get recent work items for suggestions :param project_name: Azure DevOps project name :param limit: Max number of items to return (None for all, details fetch still respects this cap) :param states: Optional list of state names to filter by (e.g., ['Ready for QA', 'Re-open']) u&🔍 Fetching work items for project: rz/_apis/wit/wiql?api-version=6.0r2r3r4)r6z Content-Typer7z[System.TeamProject] = ''rz, cSsg|]}d|dqS)rcr()r.sr(r(r)r0sz5AzureClient.get_recent_work_items..z[System.State] IN ()z AND z_SELECT [System.Id], [System.Title], [System.WorkItemType], [System.State] FROM WorkItems WHERE z# ORDER BY [System.ChangedDate] DESCqueryu🔍 Making WIQL request to: r<)r>rRr?u🔍 WIQL response status: r1u❌ WIQL request failed: z - Z workItemsu 🔍 Found z work items in WIQL responseNr/z/_apis/wit/workitems/r^r=u(❌ Failed to get details for work item r_u&🔍 Successfully fetched details for z work itemsu%❌ Failed to get recent work items: )rrrrKrLrrMrNr!listtuplerrPrSr@postrQr8r`rRrrT)r&r]rbstatesrr>Z where_clausesZ quoted_statesZ where_sqlZ wiql_querypayloadrWresult work_itemsZdetailed_itemsZitems_to_processitemitem_idZitem_urlZ item_responser[r(r(r)get_recent_work_itemssN      z!AzureClient.get_recent_work_items)NNNN)N)NN) __name__ __module__ __qualname__r*rr\r"raintrpr(r(r(r)r s   Yr )utils.error_loggerrrrrutils.error_monitorrrrHr@rKZbs4rconfig.settingsr r r r r r(r(r(r)s