8. 框架优化-2

接口请求部分优化

# project>api>api.py
import pytest
import requests
from utils.read_test import base_data

url = base_data.read_ini()['host']['api_sit_url']


def mobile_query(param):
    r = requests.get(url + "/shouji/query",
                     params={"shouji": param['shouji'], "appkey": param['appkey']})
    print(r.status_code)
    result = r.json()
    return result

# project > test_cases > test_case_optimize.py
import pytest

from api.api import mobile_query
from utils.read_test import base_data

url = base_data.read_ini()['host']['api_sit_url']


def test_mobile():
    param = base_data.read_data()["mobile"]
    result = mobile_query(param)
    print(result)

if __name__ == '__main__':
    pytest.main()

将url、root_url分别放在两个独立的文件里面进行管理

# project > data > data.yaml
mobile: {shouji: 13456759012,appkey: 0c818521d38759e1}

# project > utils > read_test.py
import yaml
import configparser
import os

current_path = os.path.realpath(__file__)
parent_path = os.path.dirname(current_path)
data_path = os.path.join(os.path.dirname(parent_path), "data", "data.yaml")
ini_path = os.path.join(os.path.dirname(parent_path), "config", "setting.ini")


class FileRead:
    def __init__(self):
        self.data_path = data_path
        self.ini_path = ini_path

    def read_data(self):
        f = open(self.data_path, encoding="utf8")
        data = yaml.safe_load(f)
        return data

    def read_ini(self):
        config = configparser.ConfigParser()
        config.read(self.ini_path, encoding='utf8')
        return config


base_data = FileRead()

# project > core > api_util.py
from core.rest_client import get

def get_mobile_belong(params,**kwargs):
    return get('/shouji/query',params,**kwargs)

# project > core > api.py
from core.api_util import get_mobile_belong

def mobile_query(params):
    response = get_mobile_belong(params=params)
    return response.json()

# project > test_cases > test_case_optimize.py
import pytest

from api.api import mobile_query
from utils.read_test import base_data

url = base_data.read_ini()['host']['api_sit_url']


def test_mobile():
    param = base_data.read_data()["mobile"]
    result = mobile_query(param)
    print(result)

if __name__ == '__main__':
    pytest.main()

# project > config > setting.ini
[host]
api_sit_url = https://api.binstd.com

# project > core > rest_client.py
import requests

from utils.read_test import base_data

api_root_url = base_data.read_ini()['host']['api_sit_url']


def get(url, params, **kwargs):
    return requests.get(api_root_url + url,params,**kwargs)

进一步优化，用类方法/ **kwargs用于传递字典

api_util.py
from core.rest_client import RestClient


class Api(RestClient):
    def __init__(self):
        super().__init__()

    def get_mobile_belong(self, **kwargs):
        return self.get('/shouji/query', **kwargs)

api_util = Api()

# rest_client.py
import requests

from utils.read_test import base_data

api_root_url = base_data.read_ini()['host']['api_sit_url']


class RestClient:
    def __init__(self):
        self.api_root_url = api_root_url

    def get(sefl, url, **kwargs):
        return requests.get(sefl.api_root_url + url, **kwargs)