python 接口自动化测试框架设计（六）

七、操作json（operation_json.py）

#coding:utf-8

import json

class OperationJson:

    def __init__(self,file_path=None):
        if file_path == None:
            self.file_path = ''
        else:
            self.file_path = file_path
            self.data = self.read_data()

    def read_data(self):
        with open(self.file_path) as fp:
            data = json.load(fp)
            return data

    def get_data(self,id):
        print(type(self.data))
        return self.data[id]

    def write_data(self,data):
        with open('','w') as fp:
            fp.write(json.dumps(data))

if __name__ == '__main__':
    operjson = OperationJson()