Python 接口自动化常用方法封装

雨点打透心脏的1/2处 2023-08-17 17:43 117阅读 0赞

1 #!/usr/bin/env python
      2 # -*- coding:utf-8 -*-
      3 # *************************************
      4 # @Time    : 2019/7/1
      5 # @Author  : Zhang Fan
      6 # @Desc    : RobotFramework Library
      7 # @File    : MyKeyworks.py
      8 # @Update  : 2019/8/23
      9 # *************************************
     10 from robot.api import logger
     11 import configparser
     12 import jsonpointer
     13 import jsonpatch
     14 import datetime
     15 import chardet
     16 import random
     17 import string
     18 import json
     19 import time
     20 
     21 
     22 class MyKeyword(object):
     23     """
     24     ===================================================================
     25     =====================       MyKeyword        ======================
     26     ===================================================================
     27     """
     28     @staticmethod
     29     def detect_code(self, byte_string):
     30         """检测字节串编码.
     31 
     32         参数：
     33             byte_string：字节串
     34 
     35         示例：
     36             |  Detect Code |    ${byte_string} | #返回字节串编码，比如'utf-8'   |
     37         """
     38         return chardet.detect(byte_string)['encoding']
     39 
     40     def get_config_value(self, cfg_path, section, name):
     41         """获取配置文件中指定节点下的指定选项值.
     42 
     43         参数：
     44             cfg_path：配置文件路径
     45             section：节点名
     46             name：选项名
     47 
     48         示例：
     49             |  Get Config Value |    ${CURDIR}\\config.ini  |    service_info   |  address  |
     50         """
     51         cfg = configparser.ConfigParser()
     52         cfg.read(cfg_path)
     53         return cfg.get(section, name)
     54 
     55     """
     56     ===============================================================
     57     =====================    Json Handle       ====================
     58     ===============================================================
     59     """
     60     def parse_json(self, json_string):
     61         """
     62         解析JSON文档并返回数据结构.
     63 
     64         参数：
     65             json_string：JSON文档
     66 
     67         示例:
     68         | ${result}=       | Parse Json  | [1, 2, 3] |
     69         | Length Should Be | ${result}   | 3         |
     70         """
     71         try:
     72             return json.loads(json_string)
     73         except ValueError as e:
     74             raise ValueError("Could not parse '%s' as JSON: %s" % (json_string, e))
     75 
     76     def stringify_json(self, data):
     77         """
     78         将数据结构转换为包含其JSON字符串表示形式的字符串.
     79 
     80         参数：
     81             data：数据结构
     82 
     83         示例:
     84         | ${data} =                   | Create List    |  1  2  3          |
     85         | ${json_string}=             | Stringify JSON |  ${data}          |
     86         | Should Be Equal As Strings  | ${json_string} |  ["1", "2", "3"]  |
     87         """
     88         try:
     89             return json.dumps(data, ensure_ascii=False)
     90         except ValueError as e:
     91             raise ValueError("Could not stringify '%r' to JSON: %s" % (data, e))
     92 
     93     def get_json_value(self, json_string, json_pointer):
     94         """
     95         获取JSON中指定目标节点值.
     96 
     97         参数：
     98             json_string：JSON文档
     99             json_pointer：JSON节点
    100 
    101         示例:
    102         | ${result}=       | Get Json Value   | {"foo": {"bar": [1,2,3]}} | /foo/bar |
    103         | Should Be Equal  | ${result}        | [1, 2, 3]                 |          |
    104         """
    105         try:
    106             json_string = json.loads(str(json_string))
    107         except:
    108             json_string = eval(str(json_string))
    109         return jsonpointer.resolve_pointer(json_string, json_pointer)
    110 
    111     def set_json_value(self, json_string, json_pointer, json_value):
    112         """
    113         设置JSON中指定目标节点值.
    114 
    115         参数：
    116             json_string：JSON文档
    117             json_pointer：JSON节点
    118             json_value：JSON值
    119 
    120         示例:
    121         | ${result}=       | Set Json Value | {"foo": {"bar": [1,2,3]}} | /foo | 12 |
    122         | Should Be Equal  | ${result}      | {"foo": 12}               |      |    |
    123         """
    124         try:
    125             json_string = json.loads(str(json_string))
    126         except:
    127             json_string = eval(str(json_string))
    128         json_new = jsonpatch.apply_patch(json_string, [{
        'op': 'add', 'path': json_pointer,
    129                                                         'value': self.parse_json(json_value)}])
    130         return self.stringify_json(json_new)
    131 
    132 
    133     """
    134     ===================================================================
    135     ====================    DateTime Handle       =====================
    136     ===================================================================
    137     """
    138     def calc_time_diff(self, date1, date2=None, format_=''):
    139         """计算与当前的时间差，返回秒数.
    140 
    141         参数：
    142             date: 日期字符串（支持多种日期格式，以及时间戳）
    143             format_: 日期格式，默认为空
    144 
    145         示例：
    146             |   Calc Time Diff  |   Jul 30, 2019 10:24:36 AM   |
    147             |   Calc Time Diff  |   2019-07-30T10:24:36Z       |
    148             |   Calc Time Diff  |   2019-07-30 10:24:36.000    |
    149             |   Calc Time Diff  |   2019-07-30 10:24:36        |
    150             |   Calc Time Diff  |   20190730102436             |
    151             |   Calc Time Diff  |   1564453476000              |
    152         """
    153         def format_date(date, format_=''):
    154             if not format_:
    155                 if all(x in date for x in ['-', ' ', ':', '.']):
    156                     format_ = "%Y-%m-%d %H:%M:%S.%f"
    157                 elif all(x in date for x in ['-', 'T', ':', 'Z']):
    158                     format_ = "%Y-%m-%dT%H:%M:%SZ"
    159                 elif all(x in date for x in [' ', ',', ':']):
    160                     format_ = "%b %d, %Y %I:%M:%S %p"
    161                 elif all(x in date for x in ['-', ' ', ':']):
    162                     format_ = "%Y-%m-%d %H:%M:%S"
    163                 else:
    164                     format_ = "%Y%m%d%H%M%S"
    165             try:
    166                 timestamp = time.mktime(time.strptime(date, format_))
    167                 return int(timestamp * 1000)
    168             except ValueError as e:
    169                 raise ValueError(e)
    170 
    171         if not date2:
    172             date2 = int(time.time() * 1000)
    173         else:
    174             date_string2 = str(date2).strip('b').strip('u').replace("'", '').replace('"', '')
    175             date2 = format_date(date_string2, format_)
    176 
    177         date_string1 = str(date1).strip('b').strip('u').replace("'", '').replace('"', '')
    178         if not date_string1:
    179             return date_string1
    180         if date_string1.isdigit() and len(date_string1) == 13:
    181             return int(abs(date2 - int(date_string1))/1000)
    182         date1 = format_date(date_string1, format_)
    183         return int(abs(date2 - date1)/1000)
    184 
    185     def generate_guid(self):
    186         """随机生成GUID字符串."""
    187         def _random(num):
    188             return ''.join(random.sample(string.ascii_uppercase + string.digits, num))
    189         guid = '-'.join([_random(8), _random(4), _random(4), _random(4), _random(12)])
    190         return guid
    191 
    192 
    193 if __name__ == '__main__':
    194     print('This is test.')
    195     mk = MyKeyword()

转载于:https://www.cnblogs.com/leozhanggg/p/11401616.html