python为封装的api工具类添加批次号功能

秦羽 2023-07-10 PM 154℃ 0条

## 背景

现在有一个Python封装的api接口工具类，每个业务操作需要调用多个api接口，现在想将一个业务请求内的多次调用使用同一个批次号，便于后续日志记录与统计分析。

以下示例中，假设`TestOld`和`TestNew`为改造前后的接口工具类；`x,y,z`为测试的`a`和`b`函数为两个不同的api接口；`handle_ts`为一个业务操作；`t1`和`t2`为测试函数

## 改造前

```python
class TestOld:
    def __init__(self, x, y, z):
        self.x = x
        self.y = y
        self.z = z

def a(self, a):
        print('a', a)

def b(self, b):
        print('b', b)
        
    def handle_ts(self, a, b):
        self.a(a)
        self.b(b)

def t1():
    t = TestOld(0, 0, 0)
    t.handle_ts(1, 2)
    t.handle_ts(5, 6)
```
此时调用`t1`测试函数，输出如下：
```bash
a 1
b 2
a 5
b 6
```
此时无法区分每个api请求所属的批次

## 改造后

```python
class TestNew:
    def __init__(self, x, y, z, sid=None):
        self.x = x
        self.y = y
        self.z = z
        self.sid = sid

def __enter__(self):
        return self

def __call__(self, sid):
        return TestNew(x=self.x, y=self.y, z=self.z, sid=sid)

def __exit__(self, exc_type, exc_val, exc_tb):
        return False

def a(self, a):
        print(self.sid, 'a', a)

def b(self, b):
        print(self.sid, 'b', b)
    
    def handle_ts(self, a, b):
        self.a(a)
        self.b(b)

def t2():
    t = TestNew(0, 0, 0)
    with t(1001) as tm:
        tm.handle_ts(1, 2)
    with t(1002) as tm:
        tm.handle_ts(5, 6)
```

此时调用`t2`测试函数，输出如下：
```bash
1001 a 1
1001 b 2
1002 a 5
1002 b 6
```
其中`1001`和`1002`为两次业务操作的两个批次号，可以看到在同一个业务操作（handle_ts函数）内，可以保证多个api请求（a和b）的批次号相同

标签: none

非特殊说明，本博所有文章均为博主原创。

上一篇 python解析app安装包apk内基本信息

下一篇 python使用requests库下载大文件占用内存高