144.完成证书下载功能

大家好~我是米洛！
我正在从0到1打造一个开源的接口测试平台, 也在编写一套与之对应的教程，希望大家多多支持。
欢迎关注我的公众号米洛的测开日记，获取最新文章教程!

回顾

上一节我们梳理了代理的难题，这一节就让我们来解决证书下载问题。其实可能这个功能不是很紧急，但起码目前pity要想做一个完整版的演示功能，这个功能是必不可少的。

找规律

我们需要先找到各个证书的下载地址规律，从而根据规律拼凑出新的url。先挂上代理，然后看看对应的mitm.it：

注意这个后缀，分别是p12,pem和cer。接着我们来看看F12，对应的链接:

其实前面都是固定的，也就是:

http://mitm.it/cert/

根据pem，p12等所有区分，所以我们写定对应的规律即可。话不多说，开始编码:

设置枚举类

新建app/enums/CertEnum.py文件

from enum import IntEnum


class CertType(IntEnum):
    windows = 0
    linux = 1
    macos = 2
    ios = 3
    android = 4

    def get_suffix(self):
        if self == CertType.windows:
            return "p12"
        if self in (CertType.linux, CertType.macos, CertType.ios):
            return "pem"
        if self == CertType.android:
            return "cer"
        raise Exception("unsupported cert type")

除了定义windows这些枚举值以外，我们还需要把他们和pem/p12/cer对应起来，所以我们需要写一个get_url的方法，给他们自动归类。

编写下载证书接口

修改app/routers/request/http.py，我们加入以下代码。

@router.get("/cert")
async def http_request(cert: CertType):
    try:
        suffix = cert.get_suffix()
        client = AsyncRequest(CERT_URL + suffix)
        content = await client.download()
        shuffle = list(range(0, 9))
        random.shuffle(shuffle)
        filename = f"{''.join(map(lambda x: str(x), shuffle))}mitmproxy.{suffix}"
        with open(filename, 'wb') as f:
            f.write(content)
        return PityResponse.file(filename, f"mitmproxy.{suffix}")
    except Exception as e:
        return PityResponse.failed(e)