【Python】使用re.sub定制参数化Jmeter测试脚本

心已赠人 2023-08-17 15:17 133阅读 0赞

## 背景 ##

见上一篇文章：

str_temp = """
      <ThreadGroup guiclass="ThreadGroupGui" testclass="ThreadGroup" testname="Thread Group" enabled="true">
        <stringProp name="ThreadGroup.on_sample_error">continue</stringProp>
        <elementProp name="ThreadGroup.main_controller" elementType="LoopController" guiclass="LoopControlPanel" testclass="LoopController" testname="Loop Controller" enabled="true">
          <boolProp name="LoopController.continue_forever">false</boolProp>
          <stringProp name="LoopController.loops">$loops</stringProp>
        </elementProp>
        <stringProp name="ThreadGroup.num_threads">$num_threads</stringProp>
        <stringProp name="ThreadGroup.ramp_time">$ramp_time</stringProp>
        <boolProp name="ThreadGroup.scheduler">false</boolProp>
        <stringProp name="ThreadGroup.duration"></stringProp>
        <stringProp name="ThreadGroup.delay"></stringProp>
      </ThreadGroup>
    """
    
    str_l = re.sub(r"\$loops", "101", str_temp)
    str_t = re.sub(r"\$num_threads", "102", str_l)
    str_r = re.sub(r"\$ramp_time", "103", str_t)
    
    print(str_r)

输出：

···
          <boolProp name="LoopController.continue_forever">false</boolProp>
          <stringProp name="LoopController.loops">101</stringProp>
        </elementProp>
        <stringProp name="ThreadGroup.num_threads">102</stringProp>
        <stringProp name="ThreadGroup.ramp_time">103</stringProp>
    ···

相信大家也注意到了，我们每替换一个参数都需要调用一次`re.sub`，而且要将上一次调用的输出作为下一次的输入，像极了递归调用。但是我们今天不介绍递归改写的方法，而是使用闭包的方式，具体的例子如下：

import re
    
    
    def multiple_replace(text, adict):
        rx = re.compile('|'.join(map(re.escape, adict)))
    
        def one_xlat(match):
            return adict[match.group(0)]
    
        return rx.sub(one_xlat, text)  # 每遇到一次匹配就会调用回调函数
    
    
    # 把key做成了 |分割的内容，也就是正则表达式的OR
    map1 = {'1': '2', '3': '4', '5': '6'}
    _str = '113355'
    print(multiple_replace(_str, map1))

转载于:https://www.cnblogs.com/Detector/p/10448500.html