實(shí)現(xiàn)原理

• 在請(qǐng)求之前,對(duì)請(qǐng)求的url進(jìn)行替換

1. 需要二次封裝requests，對(duì)請(qǐng)求進(jìn)行定制化。
2. 將請(qǐng)求的結(jié)構(gòu)體的url從一個(gè)寫(xiě)死的ip地址改為一個(gè)（任意的）域名。
3. 使用一個(gè)env 配置文件，存放各個(gè)環(huán)境的配置信息。
4. 然后將請(qǐng)求結(jié)構(gòu)體中的url替換為env配置文件中個(gè)人選擇的url。
5. 將env配置文件使用yaml進(jìn)行管理。

實(shí)戰(zhàn)

• env.yaml 內(nèi)容

default: dev
env_select:
  dev: 127.0.0.1
  test: 127.0.0.2

• env_demo.py 內(nèi)容

import requests
import yaml


class Api:
    env = yaml.safe_load(open("env.yaml"))
    data = {
        "method": "get",
        "url": "http://testing-studio:9999/demo.txt",
        "headers": None,
    }

    # data是一個(gè)請(qǐng)求的信息
    def send(self, data:dict):
        # 進(jìn)行替換
        data["url"] = str(data["url"]).replace("testing-studio", self.env["env_select"][self.env["default"]])
        r = requests.request(method=data["method"], url=data["url"], headers=data["headers"])
        return r

if __name__ == '__main__':
    api = Api()
    print(api.send(api.data).text)

• 運(yùn)行結(jié)果

"D:\Program Files\Python38\python.exe" D:/Programs/DevOps/Python_Practice/lessons/api/env_demo.py
eyJ0b3BpY3MiOgp7CiJvcmFuZ2UiOiJtb3ZpZSIsCiJzaG9vbCI6InRlc3Rpbmctc3R1ZGlvIiwKInByZXNpZGVudCI6InNldmVuaXJ1YnkiCn0KfQo=

• 將字典轉(zhuǎn)成yaml文件腳本

import yaml

def test_yaml():
    env = {
        "default":"dev",
        "env_select":
            {
                "dev": "127.0.0.1",
                "test": "127.0.0.2"
            }

    }
    with open("env.yaml", "w") as f:
        yaml.safe_dump(data=env, stream=f)