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)