a 4jF@s,dZddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z ddl mZddlmZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZd(ddZd)ddZd*ddZddZd+d d!Zd,d"d#Z d-d$d%Z!d&d'Z"dS).z5 Contains utility functions and classes for Runners. )absolute_importbisectN) string_types)parser) FileLocation)ensure_stream_with_encoderc@s eZdZdS)FileNotFoundErrorN__name__ __module__ __qualname__rr8E:\soccer\ml_env\Lib\site-packages\behave/runner_util.pyr sr c@s eZdZdS)InvalidFileLocationErrorNr rrrrrsrc@s eZdZdS)InvalidFilenameErrorNr rrrrrsrc@s&eZdZedejZeddZdS)FileLocationParserz&^\s*(?P.*):(?P\d+)\s*$cCsF|j|}|r6|d}t|d}t||S|}t|S)Nfilenameline)patternmatchgroupstripintr)clstextrrrrrrparse(s  zFileLocationParser.parseN) r r r recompileUNICODEr classmethodrrrrrr%src@sHeZdZdZdddZddZddZed d Zdd d Z ddZ dS) FeatureScenarioLocationCollectora Collects FileLocation objects for a feature. This is used to select a subset of scenarios in a feature that should run. USE CASE: behave feature/foo.feature:10 behave @selected_features.txt behave @rerun_failed_scenarios.txt With features configuration files, like: # -- file:rerun_failed_scenarios.txt feature/foo.feature:10 feature/foo.feature:25 feature/bar.feature # -- EOF NcCsJ|s|r|j}||_||_d|_t|_t|_t|_|rF||dSNF)rfeatureuse_all_scenariossetscenario_lines all_scenariosselected_scenarios add_location)selfr#locationrrrr__init__Psz)FeatureScenarioLocationCollector.__init__cCs.d|_d|_d|_t|_t|_t|_dSr")r#rr$r%r&r'r(r*rrrclear\s z&FeatureScenarioLocationCollector.clearcCsN|js|j|_|j|jks.Jd|j|jf|jrD|j|jnd|_dS)Nz %s <=> %sT)rrr&addr$)r*r+rrrr)dsz-FeatureScenarioLocationCollector.add_locationcCs*|sdSt||d}|dkr"d}||S)a5 Select scenario line for any given line. ALGORITHM: scenario.line <= line < next_scenario.line :param line: A line number in the file (as number). :param scenario_lines: Sorted list of scenario lines. :return: Scenario.line (first line) for the given line. rr)rr&posrrrselect_scenario_line_forss z9FeatureScenarioLocationCollector.select_scenario_line_forFc Cs|js J|js|j|_dd|jD}t|j}|D]L}|||}||kr:|j||j||r:d||j|f}t |q:t |j}t }|jD]$} | j |vr|| || j q|rJ|S)a Discovers selected scenarios based on the provided file locations. In addition: * discover all scenarios * auto-correct BAD LINE-NUMBERS :param strict: If true, raises exception if file location is invalid. :return: List of selected scenarios of this feature (as set). :raises InvalidFileLocationError: If file location is no exactly correct and strict is true. cSsg|] }|jqSr)r).0scenariorrr zPFeatureScenarioLocationCollector.discover_selected_scenarios..z-Scenario location '...:%d' should be: '%s:%d') r#r'walk_scenarioslistr&r2remover/rrr%r) r*strictZexisting_linesZselected_linesrnew_linemsgr&r(r4rrrdiscover_selected_scenarioss.            z). :param stream: Output stream to use (default: sys.stderr). :param colored: Indicates if coloring should be used (default: True) Nz= You can implement step definitions for undefined steps with zthese snippets:  r)escapesZ undefinedreset) r_stderrrJrZbehave.formatter.ansi_escapesrrwriteflush)rstreamZcoloredr<rrrrprint_undefined_step_snippetss  rcCsDddlm}ddlm}||_|d|ji|j_|j|_dS)zVReset runtime environment. Best effort to reset module data to initial state. r) step_registryr~N) rrrZ StepRegistryregistryrZ ParseMatcherZ custom_typesr)rrrrr reset_runtimes    r)N)T)NN)N)N)NT)#rA __future__rrrLos.pathrGrr_sixrrrZbehave.model_corerZbehave.textutilr LookupErrorr r ValueErrorrobjectrr!rCr[rhrsr}rrrrrrrrrs4      > , %