Python：numpy模块实现保存/缓存（类似环境变量）_numpy保存变量

目的：通过脚本验证某个模块功能（包括动作1、动作2），要求“动作2”重复完成N次，执行1次“动作1”…为了方便记录总的验证次数，程序重新打开会自动获取并按照原来的计数进行累加。

NumPy 是一个 Python 包。 它代表 “Numeric Python”。 它是一个由多维数组对象和用于处理数组的例程集合组成的库。

#!/usr/bin/python
import numpy as np
from scipy import io
import os

class Optical():
    '''
    引用保存的变量（类似postman、jmeter引用环境变量）
    '''
    def __init__(self):
        self.text='b.mat'
    def create_text(self):
        '''
        “create_text()”函数创建文件
        “testname”文件名称 创建“b.mat"文件
        '''
        try:
            b = np.array(0)
            io.savemat(self.text, {'array': b})
            return b
        except Exception as ex:
            print(ex)
    def text_exists(self):
        '''
        "text_exists()"函数：判定文件是否存在
        '''
        try:
            if os.path.exists(self.text):
                return True
            else:
                return False
        except Exception as ex:
            print(ex)
    def read_cache(self):
        '''
        "read_cache"函数：读取缓存数据
        '''
        try:
            d = io.loadmat(self.text)  # 读取缓存信息
            sum = int(d['array']) + 1  # 实现计数累加
            # print('sum=',sum)
            return sum
        except  Exception as ex:
            print(ex)
    def save_cache(self):
        '''
        "save_cache()"函数：保存最新缓存数据
        '''
        try:
            '''
            #"read_cache"函数可以合并到“save_cache"函数中,如下：
            d = io.loadmat(self.text)  # 读取缓存信息
            sum = int(d['array']) + 1  # 实现计数累加
            b = np.array(sum)  # 获取新缓存信息
            io.savemat(self.text, {'array': b})  # 保存缓存信息
            print('验证结果sum=',sum)
        except  Exception as ex:
            print(ex)
            '''
            b = np.array(Optical.read_cache(self))  # 获取新缓存信息
            io.savemat(self.text, {'array': b})  # 保存缓存信息
            print('验证结果sum=',int(io.loadmat(self.text)['array']))
        except  Exception as ex:
            print(ex)
    def repeat_set(self,arg0=2,arg1=5):
        '''
        "repeat_set"函数：设置“动作1”、“动作2”重复次数
        "arg0"变量：”动作1“重复次数
        "arg1"变量：”动作2“重复次数
        '''
        try:
            for i in range(arg0):
                print('i=',i)
                if i % 2 == 0:
                    # Optical.read_cache(self)
                    Optical.save_cache(self)
                else:
                    for j in range(arg1):
                        print('j=',j )
                        # Optical.read_cache(self)
                        Optical.save_cache(self)
        except Exception as ex:
            print(ex)

op=Optical()

if __name__ == "__main__":
    if op.text_exists()==True:
        op.repeat_set()
    else:
        op.create_text()
        op.repeat_set()


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94