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

 #!/usr/bin/env python
 # -*- coding:utf-8 -*-
 # *************************************
 # @Time    : 2019/7/1
 # @Author  : Zhang Fan
 # @Desc    : RobotFramework Library
 # @File    : MyKeyworks.py
 # @Update  : 2019/8/23
 # *************************************
 from robot.api import logger
 import configparser
 import jsonpointer
 import jsonpatch
 import datetime
 import chardet
 import random
 import string
 import json
 import time

 class MyKeyword(object):
     """
     ===================================================================
     =====================       MyKeyword        ======================
     ===================================================================
     """
     @staticmethod
     def detect_code(self, byte_string):
         """检测字节串编码.

         参数：
             byte_string：字节串

         示例：
             |  Detect Code |    ${byte_string} | #返回字节串编码，比如'utf-8'   |
         """
         return chardet.detect(byte_string)['encoding']

     def get_config_value(self, cfg_path, section, name):
         """获取配置文件中指定节点下的指定选项值.

         参数：
             cfg_path：配置文件路径
             section：节点名
             name：选项名

         示例：
             |  Get Config Value |    ${CURDIR}\\config.ini  |    service_info   |  address  |
         """
         cfg = configparser.ConfigParser()
         cfg.read(cfg_path)
         return cfg.get(section, name)

     """
     ===============================================================
     =====================    Json Handle       ====================
     ===============================================================
     """
     def parse_json(self, json_string):
         """
         解析JSON文档并返回数据结构.

         参数：
             json_string：JSON文档

         示例:
         | ${result}=       | Parse Json  | [1, 2, 3] |
         | Length Should Be | ${result}   | 3         |
         """
         try:
             return json.loads(json_string)
         except ValueError as e:
             raise ValueError("Could not parse '%s' as JSON: %s" % (json_string, e))

     def stringify_json(self, data):
         """
         将数据结构转换为包含其JSON字符串表示形式的字符串.

         参数：
             data：数据结构

         示例:
         | ${data} =                   | Create List    |  1  2  3          |
         | ${json_string}=             | Stringify JSON |  ${data}          |
         | Should Be Equal As Strings  | ${json_string} |  ["1", "2", "3"]  |
         """
         try:
             return json.dumps(data, ensure_ascii=False)
         except ValueError as e:
             raise ValueError("Could not stringify '%r' to JSON: %s" % (data, e))

     def get_json_value(self, json_string, json_pointer):
         """
         获取JSON中指定目标节点值.

         参数：
             json_string：JSON文档
             json_pointer：JSON节点

         示例:
         | ${result}=       | Get Json Value   | {"foo": {"bar": [1,2,3]}} | /foo/bar |
         | Should Be Equal  | ${result}        | [1, 2, 3]                 |          |
         """
         try:
             json_string = json.loads(str(json_string))
         except:
             json_string = eval(str(json_string))
         return jsonpointer.resolve_pointer(json_string, json_pointer)

     def set_json_value(self, json_string, json_pointer, json_value):
         """
         设置JSON中指定目标节点值.

         参数：
             json_string：JSON文档
             json_pointer：JSON节点
             json_value：JSON值

         示例:
         | ${result}=       | Set Json Value | {"foo": {"bar": [1,2,3]}} | /foo | 12 |
         | Should Be Equal  | ${result}      | {"foo": 12}               |      |    |
         """
         try:
             json_string = json.loads(str(json_string))
         except:
             json_string = eval(str(json_string))
         json_new = jsonpatch.apply_patch(json_string, [{'op': 'add', 'path': json_pointer,
                                                         'value': self.parse_json(json_value)}])
         return self.stringify_json(json_new)

     """
     ===================================================================
     ====================    DateTime Handle       =====================
     ===================================================================
     """
     def calc_time_diff(self, date1, date2=None, format_=''):
         """计算与当前的时间差，返回秒数.

         参数：
             date: 日期字符串（支持多种日期格式，以及时间戳）
             format_: 日期格式，默认为空

         示例：
             |   Calc Time Diff  |   Jul 30, 2019 10:24:36 AM   |
             |   Calc Time Diff  |   2019-07-30T10:24:36Z       |
             |   Calc Time Diff  |   2019-07-30 10:24:36.000    |
             |   Calc Time Diff  |   2019-07-30 10:24:36        |
             |   Calc Time Diff  |   20190730102436             |
             |   Calc Time Diff  |   1564453476000              |
         """
         def format_date(date, format_=''):
             if not format_:
                 if all(x in date for x in ['-', ' ', ':', '.']):
                     format_ = "%Y-%m-%d %H:%M:%S.%f"
                 elif all(x in date for x in ['-', 'T', ':', 'Z']):
                     format_ = "%Y-%m-%dT%H:%M:%SZ"
                 elif all(x in date for x in [' ', ',', ':']):
                     format_ = "%b %d, %Y %I:%M:%S %p"
                 elif all(x in date for x in ['-', ' ', ':']):
                     format_ = "%Y-%m-%d %H:%M:%S"
                 else:
                     format_ = "%Y%m%d%H%M%S"
             try:
                 timestamp = time.mktime(time.strptime(date, format_))
                 return int(timestamp * 1000)
             except ValueError as e:
                 raise ValueError(e)

         if not date2:
             date2 = int(time.time() * 1000)
         else:
             date_string2 = str(date2).strip('b').strip('u').replace("'", '').replace('"', '')
             date2 = format_date(date_string2, format_)

         date_string1 = str(date1).strip('b').strip('u').replace("'", '').replace('"', '')
         if not date_string1:
             return date_string1
         if date_string1.isdigit() and len(date_string1) == 13:
             return int(abs(date2 - int(date_string1))/1000)
         date1 = format_date(date_string1, format_)
         return int(abs(date2 - date1)/1000)