登录成功后将结果写回excel

import unittest
from common.get_request import GetRequest
from common.read_data import ReadData
from ddt import ddt, data
from common import constant


@ddt
class TestLogin(unittest.TestCase):
    """登录"""
    get_request = GetRequest()
    read_data = ReadData(constant.data_file)
    login_data = read_data.read_data('login')

    @data(*login_data)
    def test_login(self, case):
        resp = self.get_request.get_request(case.method, case.url, case.data)
        try:
            self.assertTrue(case.expected == resp.json()['msg'])
            case.result = 'SUCCESS'
        except AssertionError as e:
            case.result = 'FAILED'
            print('断言失败{}'.format(e))
        finally:
            self.read_data.write_back(case.case_id+1, 7, str(resp.json()))
            self.read_data.write_back(case.case_id + 1, 8, case.result)
if __name__=='__main__':
    unittest.main()