U ÕÃhâ‡ã@s<ddlmZmZmZmZddlZddlZddlm Z ddl m Z m Z m Z mZmZddlmZddlZddlmZddlZddlZddlmZmZmZejejdd e e¡Zeeeed œd d „Z eeeed œdd„Z!eeeeeee"dœdd„Z#eedœdd„Z$eeeefdœdd„Z%deeeedœdd„Z&dS)é)Úcapture_exceptionÚcapture_messageÚset_tagÚ set_contextN©ÚWorkbook©ÚFontÚ PatternFillÚ AlignmentÚBorderÚSide)Údataframe_to_rows)Údatetime)ÚOptionalÚListÚDictz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)ÚlevelÚformat)ÚcontentÚ base_nameÚreturnc Csä|r|st d¡tdƒdStj dd¡}|›d}tj ||¡}zHtj|ddt|d d d }| |¡W5QRXt  d |›¡|WSt k rÞ}z4t d |›d|›¡td|›d|›ƒWY¢dSd}~XYnXdS)zÕSave test script to file. Args: content (str): Content to write to file base_name (str): Base name for the file Returns: Optional[str]: Filename if successful, None otherwise z$Filename and content cannot be emptyu(⌠Filename and content cannot be emptyNÚtestsÚ generatedz.txtT©Úexist_okÚwzutf-8)Úencodingz Test script saved successfully: zError saving file ú: u⌠Error saving file ) ÚloggerÚerrorÚprintÚosÚpathÚjoinÚmakedirsÚopenÚwriteÚinfoÚ Exception)rrÚ output_dirÚfilenameÚ file_pathÚfileÚe©r/ú?/var/www/html/testcasegenerator.evdpl.com/utils/file_handler.pyÚsave_test_scripts"   r1)Ú test_casesrrc sÖ|r|st d¡tdƒdS|›d}tj dd¡}tj|ddtj ||¡}z4t|ƒ}|svt d ¡t |ƒ}nBt d t |ƒ›d ¡g}|  ¡D]\}}t ||d } |  | ¡q˜|rÆt  |¡nt  ¡‰ˆjr t d ¡tdƒt  dgd gdgdgdgdgdœ¡‰dˆjkr"dˆd<nˆd d¡ˆd<ddddddddg} | D]} | ˆjkrLdˆ| <qLdˆjkrŠˆd dd„¡ˆd<ˆ d¡‰dddddddddg } ˆjD]} | | kr°|  | ¡q°‡fdd„| Dƒ} ˆj| d ‰t j|d!d"v} ˆj| d#d$d%| jd$}tˆjƒD]J\}} tˆ|  t¡ t ¡ ¡t | ƒƒ}t|d&d'ƒ|jtd(|ƒ_ qW5QRXt d)|›¡|WSt!k rÐ}z(t d*|›¡td+|›ƒWY¢dSd}~XYnXdS),zéSave test cases to Excel file. Args: test_cases (str): Test cases content to write to Excel base_name (str): Base name for the file Returns: Optional[str]: Filename if successful, None otherwise z'Filename and test cases cannot be emptyu+⌠Filename and test cases cannot be emptyNz.xlsxrrTrz6No TEST TYPE sections found, using traditional parsingzFound z TEST TYPE sections)Údefault_sectionzNo test cases could be parsedu$âš ï¸ No test cases could be parsedzNo test cases foundÚ)ÚSectionÚTitleÚScenarioÚStepsúExpected Resultú Actual Resultr5ÚGeneralr6r7Ú Preconditionsr8r9r:ÚStatusÚPrioritycSs*t|tƒr"d dd„t|ƒDƒ¡S|p(dS)NÚ cSs"g|]\}}|d›d|›‘qS©éz. r/©Ú.0ÚiÚstepr/r/r0Ú tsz7save_excel_report....r4)Ú isinstanceÚlistr$Ú enumerate)Úxr/r/r0Útóz#save_excel_report..csg|]}|ˆjkr|‘qSr/©Úcolumns)rCÚcol©Údfr/r0rFs z%save_excel_report..rMÚopenpyxl)ÚengineFú Test Cases)ÚindexZ sheet_nameéé2éAz!Excel report saved successfully: zError saving Excel report: u⌠Error saving Excel report: )"rr r!r"r#r$r%Úextract_test_type_sectionsr(Úparse_traditional_formatÚlenÚitemsÚextendÚpdÚ DataFrameÚemptyÚwarningrNZfillnaÚapplyÚappendZreindexZ ExcelWriterÚto_excelZsheetsrIÚmaxZastypeÚstrÚminÚcolumn_dimensionsÚchrÚwidthr))r2rr+r*ÚfilepathÚsectionsÚ test_dataÚ section_nameÚsection_contentZ section_dataZrequired_columnsrOZ column_orderÚwriterZ worksheetÚidxÚ max_lengthr.r/rPr0Úsave_excel_report2sz        ú          þ,rs)r2Ú status_valuesÚ source_typeÚitem_idsrc# Csºzzddlm}ddlm}m}m}m}m} ddlm } |ƒ} |   | j ¡|   d¡} d|›| d<|dd d | d_ d t ¡ d ¡›| d <d|›| d<dd |¡›| d<dt|ƒ›| d<i} |D]0}| dd¡}| |d¡}|  |d¡d| |<qÈd| d<|dd| d_ d}|  ¡D]*\}}|›d|›| d|›<|d7}q|   d¡}dd d!d"d#d$g}t|dƒD]6\}}|jd||d%}|dd|_ |d&d&d'd(|_qnt|d)ƒD] \}}| dd*¡}d*}d+|kröd,|krö| d+¡d- d,¡d}|j|d| dd*¡d%|j|d)| d d*¡d%| d!d*¡}t|tƒrVd. d/d0„t|ƒDƒ¡}|j|d1|d%| d"d*¡}t|tƒrˆd. |¡}|j|d2|d%| dd*¡}| |d¡}|j|d3|d%|j|d4|d%q°|jD]r}d}| |djƒ}|D]:}z&tt|jƒƒ|krtt|jƒƒ}Wn YnXqôt|d)d5ƒ}||j |_!qÚdd6l"m#} | ƒ}!|  $|!¡|! %d¡|! &¡WSt'k r´}"zt( )d7|"›¡|"‚W5d8}"~"XYnXd8S)9abCreate an Excel report from test cases data. Args: test_cases (List[Dict]): List of test case dictionaries status_values (Dict): Status values for test cases source_type (str): Source type (e.g., 'Jira', 'Azure') item_ids (List[str]): List of item IDs Returns: bytes: Excel file as bytes rrr)Úget_column_letterÚSummaryzTest Cases Report - ÚA1Té)ÚboldÚsizezGenerated on: z%Y-%m-%d %H:%M:%SZA3z Source Type: ZA4z Item IDs: z, ZA5zTotal Test Cases: ÚA6r6ÚUnknownú Not TestedrAzStatus Summary:ZA8)r{é rÚArTr7r8r9r=zItem ID)ÚrowÚcolumnÚvalueZCCCCCCZsolid)Ú start_colorZ end_colorZ fill_typerVr4ú(ú)éÿÿÿÿr?cSs"g|]\}}|d›d|›‘qSr@r/rBr/r/r0rFåsz'create_excel_report..éééérW)ÚBytesIOzError creating Excel report: N)*rRrÚopenpyxl.stylesr r r r r Zopenpyxl.utilsrwÚremoveÚactiveZ create_sheetZfontrÚnowÚstrftimer$r[Úgetr\rIÚcellÚfillÚsplitrGrHrNrƒrfr„rgrhrjÚiorÚsaveÚseekÚgetvaluer)rr )#r2rtrurvrr r r r r rwÚwbZ summary_wsZ status_countsÚtcZtc_idÚstatusr‚ÚcountZtc_wsÚheadersrOÚheaderr”ÚtitleÚitem_idÚstepsZexpected_resultrƒrrZ column_letterZadjusted_widthrÚoutputr.r/r/r0Úcreate_excel_report™sˆ                        r¥)Ú test_caserc Cs zÂ| d¡r*|d ¡ dd¡ ¡|d<| d¡rn|d ¡}|rd|dkrdt dd|¡}||d<qvd|d<nd|d<| d¡rJ|d}t|tƒrþg}|D]N}|rœ| ¡rœ| ¡ dd¡ ¡}t d d|¡}t d d|¡}|rœ| |¡qœ|rô|nd |d<nJt|tƒr@| ¡r@| ¡ dd¡ ¡}t dd|¡}||d<nd |d<nd |d<| d ¡rœ|d  ¡}|r’|d kr’t dd|¡}||d <nd |d <nd |d <| d¡rØ|d ¡}|rØt dd|¡}||d<| d¡dkr„| d ¡rft|d tƒrf|d  ¡rf|d  ¡}|  ¡  d¡s\t |ƒdkrXd|d  ¡|dd…nd}||d<n| d¡r„d|d›|d<| d¡s˜d|d<| d¡rÀ|d ¡ dd¡ ¡|d<|WSt k r} z t  dt| ƒ›¡|WY¢Sd} ~ XYnXdS)zFValidate and clean up test case data to ensure quality and consistencyr6ú**r4r7zNo scenario providedú\s+Ú r8z ^\s*\d+\.\s*z ^\s*[-*]\s*zNo steps providedr9zNo expected result providedr:)ÚverifyÚensureÚvalidateÚcheckrAzVerify rNz Verify resultzExecute the steps to validate: r=rr>zError validating test case: )r“ÚstripÚreplaceÚreÚsubrGrHrcrfÚlowerÚ startswithr[r)rra) r¦Zscenarior£Z cleaned_stepsrEÚ cleaned_stepÚexpectedÚactualÚderivedr.r/r/r0Ú_validate_and_clean_test_casesp                  * .    r¸)r2rc CsŒi}d}tt ||¡ƒ}|s iSt|ƒD]^\}}| d¡ ¡}| ¡}|dt|ƒkrf||d ¡nt|ƒ}|||… ¡} | ||<q(|S)zêExtract sections from test cases content based on TEST TYPE markers. Args: test_cases (str): The full test cases content Returns: Dict[str, str]: Dictionary of section name to section content zTEST TYPE:\s*([^\n]+)rA) rHr°ÚfinditerrIÚgroupr®Úendr[Ústart) r2rlZsection_patternZsection_matchesrDÚmatchrnÚ start_posÚend_posror/r/r0rYes ( rYr;)r2r3rc) Csè g}i}|}d}g}t d|›¡t d|dd…›¡d|krLd|kstd|kr`t d|¡std |kr d |kr t d ¡t d |¡}|D]Z}| ¡}|rŽt|ƒd kr®qŽd|ksÎd |ksÎt d| ¡¡rŽi} || d<t d|¡} | r |  d¡ ¡  dd¡ ¡| d<nft d|¡} | r>|  d¡ ¡  dd¡ ¡| d<n4| d¡d ¡} t d| ¡rr|   dd¡ ¡| d<t d|tj ¡} | rÈ|  d¡ ¡  dd¡ ¡}t  d d|¡}t  dd|¡}|| d<nªt d|tj ¡}|r| d¡ ¡  dd¡ ¡}t  d d|¡}t  dd|¡}|| d<nTt d|tj ¡}|rr| d¡ ¡  dd¡ ¡}t  d d|¡}t  dd|¡}|| d<t d|tj ¡}|rÆ| d¡ ¡  dd¡ ¡| d<t d| ddd …›d!¡nt  d"|dd#…›d!¡t d$|tj ¡}|rœ| d¡ ¡}t d%|dd#…›d!¡t  d&|tj¡}|rbd'd(„|Dƒ| d)<t d*t| d)ƒ›d+¡n6d,d(„| d¡Dƒ}|| d)<t d*t| d)ƒ›d-¡n8t d.|tj ¡}|rV| d¡ ¡}t d/|dd#…›d!¡t  d&|tj¡}|rd0d(„|Dƒ| d)<t d*t| d)ƒ›d+¡n6d1d(„| d¡Dƒ}|| d)<t d*t| d)ƒ›d-¡n~t  d2|dd#…›d!¡t d3|tj ¡}|rº| d¡ ¡}t d4|dd#…›d!¡|g| d)<nt  d5|dd#…›d!¡t d6|tj ¡}|r*| d¡ ¡  dd¡ ¡}t  d d|¡}t  dd|¡}|| d7<nªt d8|tj ¡}|r€| d¡ ¡  dd¡ ¡}t  d d|¡}t  dd|¡}|| d7<nTt d9|tj ¡}|rÔ| d¡ ¡  dd¡ ¡}t  d d|¡}t  dd|¡}|| d7<t d:|¡}|rø| d¡ ¡| d;<t d<|tj ¡}|rB| d¡ ¡}|rB| d=¡sB|  dd¡ ¡| d><t d?|¡}|rr| d¡ ¡  dd¡ ¡| d@<|  d¡rŽt| ƒ} | | ¡t dA| d›¡t dBt|  ¡ƒ›¡t dC|  ddD¡›¡t dE|  d)dD¡›¡qŽ|r t d*t|ƒ›dF¡|St dG¡| d¡} d}!|!t| ƒk rP| |! ¡}"|!d7}!|"sPq$|" dH¡rv|"  dId¡ ¡pp|}q$t dJ|"¡} | rÒ|r¦|rœ||d)<| |¡||  d¡ ¡  dd¡ ¡dKœ}d}g}q$t dL|"¡} | r |r |  d¡ ¡  dd¡ ¡|d<q$t dM|"¡}|rd|rd| d¡ ¡  dd¡ ¡|d<t dN|ddd …›d!¡q$t dO|"¡}|r |r dP}g}t dQ| ddR¡›¡q$| rÀt dS|"¡ rd}||d)<t dTt|ƒ›dU| ddR¡›¡t dV|"¡}#|#r$|# d¡ ¡  dd¡ ¡|d7<q$t dW|"¡}$|$ r~|$ ¡dX r@|$ dY¡n|"}%|% ¡  dd¡ ¡}&|&r$| |&¡t dZ|&›¡q$nB|" ¡ rÀ|" ¡  dd¡ ¡}'|'r$| |'¡t d[|'›¡q$t dV|"¡}| rú| rú| d¡ ¡  dd¡ ¡|d7<q$t d\|"¡}(|( r(| r(|( d¡ ¡|d;<q$t d]|"¡}| rb| rb| d¡ ¡  dd¡ ¡|d><q$t d^|"¡}| rœ| rœ| d¡ ¡  dd¡ ¡|d@<q$|r$|s$|"r$d@|k rÎ|d@d|"7<n~d>|k rî|d>d|"7<n^d7|k r|d7d|"7<n>d|k r.|dd|"7<nd|kr$|dd|"7<q$| rÎ| rd||d)<| |¡t d_| dd`¡›¡t dat| ¡ƒ›¡t db| ddD¡›¡t dc| d)dD¡›¡t d*t|ƒ›dd¡|S)ea Parse test cases using the traditional format. Args: test_cases (str): Test cases content to parse default_section (str): Default section name if none is specified Returns: List[Dict]: List of test case dictionaries Fz)Parsing test cases with default section: z)Raw test case content (first 500 chars): NiôzTitle:zSteps to reproduce:z TC_[A-Z]+_\d+z**Test Case ID:**z**Test Steps:**zDetected test case formatz\n\s*\né r5zTitle:\s*(.*?)(?:\n|$)rAr§r4r6z%\*\*Test Case ID:\*\*\s*(.*?)(?:\n|$)r?rzYScenario:\s*(.*?)(?:\n\s*Preconditions:|\n\s*Steps to reproduce:|\n\s*Expected Result:|$)r©r¨r7zR\*\*Test Scenario:\*\*\s*(.*?)(?:\n\s*\*\*Test Steps:|\n\s*\*\*Expected Result:|$)zHDescription:\s*(.*?)(?:\n\s*Steps to reproduce:|\n\s*Expected Result:|$)zJPreconditions:\s*(.*?)(?:\n\s*Steps to reproduce:|\n\s*Expected Result:|$)r<zFound preconditions: édz...z!No preconditions found in block: éÈzX(?:\*\*)?Steps to reproduce:(?:\*\*)?\s*\n([\s\S]*?)(?=\n\s*(?:\*\*)?Expected Result:|$)zFound steps text: zK(?:^|\n)\s*(?:\d+\.\s*|\*\s*|\-\s*)(.*?)(?=\n\s*(?:\d+\.\s*|\*\s*|\-\s*)|$)cSs(g|] }| ¡r| ¡ dd¡ ¡‘qS©r§r4©r®r¯©rCrEr/r/r0rFàsz,parse_traditional_format..r8z Extracted z steps using numbered formatcSs(g|] }| ¡r| ¡ dd¡ ¡‘qSrÃrÄrÅr/r/r0rFäsz steps using line-by-line formatzA\*\*Test Steps:\*\*\s*\n([\s\S]*?)(?=\n\s*\*\*Expected Result:|$)zFound test steps text: cSs(g|] }| ¡r| ¡ dd¡ ¡‘qSrÃrÄrÅr/r/r0rFðscSs(g|] }| ¡r| ¡ dd¡ ¡‘qSrÃrÄrÅr/r/r0rFôszNo steps found in block: z)Steps:\s*(.*?)(?:\n\s*Expected Result:|$)z&Found steps with alternative pattern: z*No steps found with any pattern in block: zMExpected Result:\s*(.*?)(?:\n\s*Actual Result:|\n\s*Status:|\n\s*Priority:|$)r9zP\*\*Expected Result:\*\*\s*(.*?)(?:\n\s*\*\*Actual Result:|\n\s*\*\*Priority:|$)zNExpected Outcome:\s*(.*?)(?:\n\s*Actual Result:|\n\s*Status:|\n\s*Priority:|$)z\n\s*Status:\s*(.*?)(?:\n|$)r=z*Actual Result:\s*(.*?)(?:\n\s*Priority:|$)z Priority:r:zPriority:\s*(.*?)(?:\n|$)r>zExtracted test case: zTest case fields: z Scenario: z NOT FOUNDzSteps: z test cases using block parsingz)Using line-by-line parsing for test casesz###ú#z/^(?:\d+\.\s*)?(?:\*\*)?Title:(?:\*\*)?\s*(.*?)$)r5r6z%^(?:\*\*)?Scenario:(?:\*\*)?\s*(.*?)$z*^(?:\*\*)?Preconditions:(?:\*\*)?\s*(.*?)$z%Found preconditions in line-by-line: z+^(?:\*\*)?Steps(?: to reproduce)?:(?:\*\*)?Tz(Started collecting steps for test case: r~z#^(?:\*\*)?Expected Result:(?:\*\*)?zFinished collecting z steps for test case: z,^(?:\*\*)?Expected Result:(?:\*\*)?\s*(.*?)$z^(?:(\d+)\.|\-|\*)\s*(.*?)$rˆrVz Added step: zAdded step (no pattern): z^Status:\s*(.*?)$z*^(?:\*\*)?Actual Result:(?:\*\*)?\s*(.*?)$z%^(?:\*\*)?Priority:(?:\*\*)?\s*(.*?)$zFinal test case: zNO TITLEzFinal test case fields: zFinal Scenario: z Final Steps: z& test cases using line-by-line parsing)rr(r°Úsearchr–r®r[r½rºr¯ÚDOTALLr±raÚfindallÚ MULTILINEr³r“r¸rcÚdebugrHÚkeysÚgroups))r2r3rmZ current_testZcurrent_sectionZcollecting_stepsZ current_stepsZtest_case_blocksÚblockr¦Z title_matchZtest_case_id_matchÚ first_lineZscenario_matchZ scenario_textZtest_scenario_matchZalt_scenario_matchZpreconditions_matchZ steps_matchÚ steps_textZ step_linesr£Ztest_steps_matchZalt_steps_matchZexpected_matchZ expected_textZtest_expected_matchZalt_expected_matchZ status_matchZ actual_matchZ actual_resultZpriority_matchÚlinesrDÚlineZer_matchZ step_matchZ step_textr´Z cleaned_lineZ status_aloner/r/r0rZ„s® 8                        þ      "                   rZ)r;)'Úutils.error_loggerrrrrr"Úpandasr^rRrrŽr r r r r Zopenpyxl.utils.dataframerr°rÚuuidÚloggingÚtypingrrrÚ basicConfigÚINFOÚ getLoggerÚ__name__rrfr1rsÚbytesr¥r¸rYrZr/r/r/r0Ús$    guW