a 7j3@sdZddlZddlZddlZddlZddlZddlmZddl m Z ddZ ddZ d d Z d d Zd dZedkr|eddS)a Launches SeleniumBase Commander | GUI for pytest. Usage: seleniumbase commander [OPTIONAL PATH or TEST FILE] sbase commander [OPTIONAL PATH or TEST FILE] seleniumbase gui [OPTIONAL PATH or TEST FILE] sbase gui [OPTIONAL PATH or TEST FILE] Examples: sbase gui sbase gui -k agent sbase gui -m marker2 sbase gui test_suite.py sbase gui offline_examples/ Output: Launches SeleniumBase Commander | GUI for pytest. N) shared_utils) ScrolledTextcCsd}d}d}d}d}d}d}|rtjjtjj}tjjtjj}tjjtjj}tjjtjj}tjjtjj}tjjtjj}tj j }|||||||fS)N) coloramaZForeZBLUEZBackZ LIGHTCYAN_EXZ LIGHTGREEN_EXZREDZLIGHTYELLOW_EXZBLACKStyleZ RESET_ALL) use_colorsc0c1c2c3c4c5crrOE:\soccer\ml_env\Lib\site-packages\seleniumbase/console_scripts/sb_commander.py set_colorss rcCs(||dd||jdddS)Nz-topmostTF)Zlift attributesZ after_idle)rootrrrsend_window_to_front0s rcCsg}|D]<}|dr:|dddd}||q||q|}t|}d}|D]}||rZ|d7}qZdtj}|dks||kr|r|d7}||7}n^t|D]T\}}||r|d7}d|vr||7}qd|vr|d|7}q||dd 7}qd |vr|d 7}n&d |vr*|d 7}nd|vr<|d7}d|vrP|d7}n:d|vrd|d7}n&d|vrx|d7}nd|vr|d7}d|vr|d7}nvd|vr|d7}nbd|vr|d7}nNd|vr|d7}n:d |vr|d!7}n&d"|vr|d#7}nd$|vr|d%7}|r"|d&7}| r0|d'7}| r>|d(7}| rL|d)7}| r\|d*7}nt rn|d+7}| r||d,7}d-}d.|vrd/}| }|r|d7}||7}|r|d07}|r|d17}t |t j|d/d2t|dS)3Nz(FILE) z => rz"%s" -m pytest "z"%s"z\ z(--edge)z --edgez (--firefox)z --firefoxz (--safari)z --safariz(--rs)z --rsz(--rs --crumbs)z --rs --crumbsz(--rcs)z --rcsz(--rcs --crumbs)z --rcs --crumbsz(-n=2)z -n=2z(-n=3)z -n=3z(-n=4)z -n=4z(-n=5)z -n=5z(-n=6)z -n=6z(-n=7)z -n=7z(-n=8)z -n=8z --demoz --mobilez --dashboardz --html=report.htmlz --headlessz --guiz --screenshotFz --captureTz -vz --capture=tee-sys)shell) startswithsplitappendlengetsys executable enumeratereplaceris_linuxstripprint subprocessPopenr)rtestsZselected_testscommand_stringZbrowser_stringZ rs_stringZ thread_stringverboseZ demo_modeZ mobile_modeZ dashboardZ html_reportZheadlessZsave_screenshotsadditional_optionsZ cleaned_teststestZ clean_testZ total_testsZtotal_selected_testsZ selected_testZfull_run_commandZ test_numberZcapture_neededrrr do_pytest_run6s                                     r,cst dtr( ddn ddtj ddgd}tr`| dt  |d tj g|R}|gd }t  |d tj g|R}|gd }z>t td kr| d | d| d| dWnty*Yn0t  |d tj g|R}|ttj dd d}||ttj dd d}|ttj dd d}|ttj dd d}||ttj dd d}||ttj dd d}|t tj d d d}|tj dddt t|t|f}t|dkrd}| nt|dkrdt|}| tj |dddt d d!d"tjd#}|jtjtjd$d%d } i D]} | d&7} t| <d} trZtj|| d'd(d)d dd| d* } ntj|| d'd(d)d | d+} |jd,| d-|d,d.| d7} qtj ddd} tj | d/tj d0dd1dtj d2} | | | d3 fd4d5tj d6d7 fd8d5d9tj d.dt t}| d:|!|"|#|$ %dS);Nz'SeleniumBase Commander | GUI for pytesti4iir)text)zUse Chrome Browser (Default)zUse Edge Browser (--edge)z Use Firefox Browser (--firefox)zUse Safari Browser (--safari)r)zNew Session Per Test (Default)z-Reuse Session for ALL tests in thread (--rs)z5Reuse Session and also clear cookies (--rs --crumbs)z0Reuse Session for tests with same CLASS (--rcs)z;Reuse Session for class and clear cookies (--rcs --crumbs))zNumber of Threads: 1 (Default)zNumber of Threads: 2 (-n=2)zNumber of Threads: 3 (-n=3)zNumber of Threads: 4 (-n=4)zNumber of Threads: 5 (-n=5)zNumber of Threads: 6 (-n=6)zNumber of Threads: 7 (-n=7)zNumber of Threads: 8 (-n=8)zVerbose Output (-v))r-variablepadyzDemo Mode (--demo)zMobile Mode (--mobile)zDashboard (--dashboard)zReport (--html=report.html)zHeadless Browser (--headless)z Save Screenshots (--screenshot)zXSelect from %s rows (%s files with %s tests): (All tests will run if none are selected)rz(Only ONE TEST was found and will be run:z@Select from %s tests: (All tests will run if none are selected)yellowmagenta)r-bgfgd word)widthheightwrapstateT)Zsidefillexpandz whiteblackw)r-r3r4anchorr0Z borderwidthZhighlightthicknessr/)r-r3r4rAr0r/end)Zwindow )valuez8Additional "pytest" Options: (Eg. "--incognito --slow")blue)Z textvariablezcsPt    SNr,r)_ZaoptsZaraZbrxr(ZdbxZdmxZhbxZhrxZmmxZntxrZrsxZssxr'ZvoxrrQs z$create_tkinter_gui..zRun Selected TestsgreencsPt    SrFrGrrIrrrJhs )r-r4commandZ1x1)&tkZTktitler is_windowsZminsizeLabelpackZis_macrZ StringVarsetZ OptionMenuintos cpu_count ExceptionZIntVarZ CheckbuttonselectrrZDISABLEDZTOPZBOTHZ window_createinsertZEntryZfocusbindZButtonrZgeometryZiconifyupdateZ deiconifydestroyZmainloop)r'r(files solo_testsZ options_listZ question_menuZchkZ run_displayZ text_areacountrowcbr*entryZdecoyrrIrcreate_tkinter_guis(                  &&  rbcCsXd}trd}t|\}}}}}}}tjdd}d|} d} | |7} | d7} | |7} | d7} | |7} | d7} | |7} | d 7} | |7} | d7} | |7} | d 7} | |7} | d 7} | |7} | d 7} | |7} t| tjd tj | ftj dd} | \} } | r$dt | }|||}t|dSg}t r>| d} n | d} | dddD]}d|vrZ||qZ|sd}|d||}t|dSg}|D]2}|ddkrd|dd}||qg}g}|D]P}||vr||d}||dkrd}d||||f}||q|}g||}t|| ||dS)NTFrr*z Starting the ZSeleniumZBaseZ Commanderz Desktop Appz...z2"%s" -m pytest --collect-only -q --rootdir="./" %s)stdoutrzError collecting tests: %slatin1zutf-8 rCz::z1No tests found! Exiting SeleniumBase Commander...zERROR: rz (FILE) %srsz%s => (%s Test%s))rr"rrargvjoinr$r%r&rPIPE communicatestrrOdecoder!rrr^rb)rrr r r r r r command_argsr(messageprocoutputerror error_msgr'r_groupsZg_namer\Z used_filespluralZf_rowr]rrrmains           rw__main__z>To open SBase Commander, type "sbase commander" or "sbase gui")__doc__rrTr%rtkinterrMZseleniumbase.fixturesrZtkinter.scrolledtextrrrr,rbrw__name__r$rrrrs  v^I