a .b\!@sdZddlmZddlmZddlmZddlmZmZm Z m Z m Z m Z ddl mZddlmZmZmZmZmZddlmZdd lmZmZmZmZGd d d Zed d d fe ee ee efeedddZedkredS)z7 Created on Mon Feb 14 17:54:58 2022 @author: nicolas )glob) expanduser)Path)DictListIterableTypeUnionOptional) WebsiteTest)TestInfoTestName TestsResultsScoresPathLike) LOG_DIR_NAME) list2xlsxcached_propertyas_pathwarnc@sjeZdZedfeeeeedddZe e e efdddZ e e e e fddd Ze edd d Ze edd d Ze e eeefdddZe e e e eeffdddZe e e efdddZe e e eefdddZe e eefdddZe e eefdddZe e eefdddZd%edddZd&ed!d"d#Zd$S)'CollectTestsResults/)paths test_classpath_on_servercCs$tdd|D|_||_||_dS)Ncss|]}t|VqdSNr.0pathr ]/home/manager/.local/lib/python3.9/site-packages/websites_test_framework/testing_framework.py ,z/CollectTestsResults.__init__..)sortedrr_path_on_server)selfrrrr r r!__init__%szCollectTestsResults.__init__)returncsfddjDS)Ncsi|]}|j|jdqS))r)rr%rr&r r! 2sz0CollectTestsResults._testers..rr)r r)r!_testers0s zCollectTestsResults._testerscCsdd|jDS)NcSsi|]\}}||qSr )runrrtesterr r r!r*8r#z/CollectTestsResults.results..)r,itemsr)r r r!results6szCollectTestsResults.resultscsfddjDS)Ncs&i|]}|ddj|DqS)cSsi|]\}}||dqS)scorer rnameresultr r r!r*=r#z>CollectTestsResults._raw_scores...r1r0rr)r r!r*<sz3CollectTestsResults._raw_scores..r+r)r r)r! _raw_scores:s zCollectTestsResults._raw_scorescCsvi}|jD]b\}}i||<|D]H\}}||||<t|j|jr&|j|}dkr&||||<q&q|S)zReturn the scores for each test for all websites. For each test marked as relative, scores are divided by the best score of all websites. r)r7r0getattrrZ is_relative best_scores)r&scoresrr1 test_namer2bestr r r!r:As  zCollectTestsResults.scorescs:i}jD]&tfddjDdd|<q|S)Nc3s|]}j|VqdSr)r7rr&r;r r!r"Yr#z2CollectTestsResults.best_scores..)default)rZget_all_tests_namesmaxr)r&r9r r=r!r9Ts  zCollectTestsResults.best_scorescsfddjDS)Ncs&i|]}|ddj|DqS)cSsi|]\}}||dqS)logr r3r r r!r*`r#z7CollectTestsResults.logs...r6rr)r r!r*_sz,CollectTestsResults.logs..r+r)r r)r!logs]s zCollectTestsResults.logscs\tjtjfdddkrDtdjjdfddjDS)zReturn the global score of each website. The global score is the weighted arithmetic mean of all tests scores. csj|Sr)r:values)rr)r r!_scoresmsz2CollectTestsResults.global_scores.._scoresrzNo test found ! ()c s.i|]&}|tddt|DqS)css|]\}}||VqdSrr )rweightr2r r r!r"sr#z?CollectTestsResults.global_scores...)sumzipr)rCtotalweightsr r!r*rsz5CollectTestsResults.global_scores..)rFrIrBlist RuntimeErrorrZ tests_namesr1r)r )rCr&rHrIr! global_scoresds z!CollectTestsResults.global_scorescCsbzdd|jDWSty\|jtkrDtdd|jjdddt|jDYS0dS)NcSsi|]\}}||qSr )Z get_authorsr.r r r!r*zr#z/CollectTestsResults.authors..z No `.get_authors()` method foundz'you should overwrite it in your class `z`.cSs$i|]\}}||d|jgqS)-)r4)rirr r r!r*r#) r,r0NotImplementedErrorrr r __qualname__ enumeraterr)r r r!authorsws  zCollectTestsResults.authorscCs |jSr)rZget_tests_infosr)r r r! _tests_infossz CollectTestsResults._tests_infoscCsdd|jDS)NcSsi|]\}}||dqS)titler rr4infosr r r!r*r#z.CollectTestsResults.titles..rSr0r)r r r!titlesszCollectTestsResults.titlescCsdd|jDS)NcSsi|]\}}||dqS)rEr rUr r r!r*r#z/CollectTestsResults.weights..rWr)r r r!rIszCollectTestsResults.weights.)folderc Cst|t}|jdd|jD]}d|g}|j|}|j|}|j|D]V}|j|}|d||d|||d|j ||||qLd |}|j |D]} || dj |d d qqdS) NTexist_okz # Website: z ## zscore: zweight:  z.mdutf8)encoding) rrmkdirrr:rAr1rXappendrIjoinrR write_text) r&rZrZfull_logr:rAr;rTZlog_textr4r r r! write_logs        zCollectTestsResults.write_log scores.xlsx) xlsx_pathcCsddgdd|jDddg|jg}|jD]T}dtd|j|dg|j|}|j|D]}||d<||d d qhq4t |j d d t ||t d |d d S)z,Write all tests results inside an XSLX file.NameZScorecss|] }|VqdSrr )rrTr r r!r"r#z6CollectTestsResults.write_xlsx_file..rNT) missing_okz! XLSX file generated: rY) rXrBrIrroundrLr:rRrarunlinkrprint)r&rfdatarrowr4r r r!write_xlsx_files & z#CollectTestsResults.write_xlsx_fileN)rY)re) __name__ __module__rPr rrrstrr'rrrr,rr1rr7r:r r floatr9rArLrrRr rSrXrIrdrqr r r r!r$s>   rrYr)rinput_dir_or_globoutput_file_or_dirrcCstdt|tr2d|vr2ddtt|D}nPt|ttfrtt|}|s^td|dd|dD}ndd|D}t|}|j dkr|j }|}n |}|d }t d d|D||}|j d d | |||d S)a^Run all tests, write log files and generate XLSX file with results. `input_dir_or_glob` is either: - the name of the parent folder containing all the websites' root directories, - a glob like '/home/*/*/www', - a list of paths, each path corresponding to the root directory of a website. `output_file_or_dir` is either a directory or the name of a XLSX file. `path_on_server` is used to rewrite absolute links, when server root is not website root. For example, if the website folder is /path/to/website on the server, path_on_server should be "/path/to/website". z Starting tests... *css|]}t|VqdSrrrpthr r r!r"r#z"run_and_collect..zDirectory not found: css|]}|r|VqdSr)is_dirryr r r!r"r#css|]}t|VqdSrrryr r r!r"r#z.xlsxrecss|]}|jtkr|VqdSr)r4rryr r r!r"r#Tr[N)rn isinstancertrrrrr{FileNotFoundErrorsuffixparentrr`rdrq)rrvrwrr output_dirZ output_file collectorr r r!run_and_collects*   r__main__N) __doc__ros.pathrpathlibrtypingrrrrr r Z"websites_test_framework.test_classr Z$websites_test_framework.custom_typesr r rrrZwebsites_test_framework.paramrZwebsites_test_framework.toolsrrrrrrtrrrr r r r!s,       2