import requests  #導入requests模塊
class Base():   
    def method_post(self,url,params = None,data = None,headers = None,files = None):
        return requests.post(url = url,params = params,data = data,headers = headers,files = files)
    def method_get(self,url,params = None,data = None,headers = None,files = None):
        return requests.get(url = url,params = params,data = data,headers = headers,files = files)
    def method_put(self,url,params = None,data = None,headers = None,files = None):
        return requests.put(url = url,params = params,data = data,headers = headers,files = files)
    def method_delete(self,url,params = None,data = None,headers = None,files = None):
        return requests.delete(url = url,params = params,data = data,headers = headers,files = files

def requests_type(self,method,url,params = None,data = None,headers = None,files = None):
        if method =='post' or method =='POST':
            return self.method_post(url = url,params = params,data = data,headers = headers,files = files)
        elif method =='get' or method =='GET':
            return self.method_get(url = url,params = params,data = data,headers = headers,files = files)
        elif method =='put' or method =='PUT':
            return requests.put(url = url,params = params,data = data,headers = headers,files = files)
        elif method =='delete' or method =='DELETE':
            return requests.delete(url = url,params = params,data = data,headers = headers,files = files)

def Get_Login_cookie(self,data):
        res = self.method_post(url = "請求網址",data = data)
        return {"userId":str(res.json()['result']['userId']),"sessionId":res.json()['result']['sessionId']}

log_path = r'D:\PycharmProjects\自動化接口\自動化接口測試\report\logs'
report_html = r'D:\PycharmProjects\自動化接口\自動化接口測試\report\html'
read_xlrd = r'D:\PycharmProjects\自動化接口\自動化接口測試\data'
class Common():
    #封裝日志方法
    def get_logs(self,path = log_path):
        import logging,time
        logs = logging.getLogger()
        logs.setLevel(logging.DEBUG)
        path = path+'/' + time.strftime('%Y-%m-%d-%H-%M-%S') + '.log'
        write_file = logging.FileHandler(path,'a+',encoding='utf-8')
        write_file.setLevel(logging.DEBUG)
        set_logs = logging.Formatter('%(asctime)s - %(filename)s - %(funcName)s - %(levelname)s - %(message)s')
        write_file.setFormatter(set_logs)
        pycharm_text = logging.StreamHandler()
        pycharm_text.setFormatter(set_logs)
        logs.addHandler(write_file)
        logs.addHandler(pycharm_text)
        return logs

# 讀取Excel表方法，方便后續(xù)讀取接口用例數(shù)據(jù)
    def ReadExcelTypeDict(self,file_name,path = read_xlrd):
        path = path+'/' + file_name
        import xlrd
        work_book = xlrd.open_workbook(path)  # 打開Excel表
        sheets = work_book.sheet_names()  # 獲取所有的sheets頁
        DatasList = []
        for sheet in sheets:
            sheets = work_book.sheet_by_name(sheet)
            nrows = sheets.nrows
            for i in range(0,nrows):
                values = sheets.row_values(i)
                DatasList.append(values)
        title_list = DatasList[0]
        content_list = DatasList[1:]
        new_list = []
        for content in content_list:
            dic = {}
            for i in range(len(content)):
                dic[title_list[i]] = content[i]
            new_list.append(dic)
        return new_list   #最終返回為字典形式 有鍵和值

# 封裝一個HTML報告方法
    def GetHtmlResult(self,suite,title,path = report_html):
        import HTMLTestReportCN,time
        path = path + '/' + time.strftime('%Y-%m-%d-%H-%M-%S') + '.html'
        with open(path,'wb+') as f:
            run = HTMLTestReportCN.HTMLTestRunner(stream=f,description='用戶相關接口測試報告',tester='小白',title = title)
            run.run(suite)

import unittest
import ddt
import 自動化接口測試.common.commons as common
from 自動化接口測試.base.Base_Page import Base

import unittest
import ddt
import 自動化接口測試.common.commons as common
from 自動化接口測試.base.Base_Page import Base
r = common.Common().ReadExcelTypeDict('cezxhi .xlsx')  # 拿到具體的Excel表數(shù)據(jù)
@ddt.ddt  #導入ddt模塊
class TestLogin(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:  # setupclass類方法  全部用例開始前執(zhí)行一次
        cls.logs = common.Common().get_logs() # 導入日志方法
        cls.logs.debug('開始寫入接口自動化測試用例')
    @classmethod
    def tearDownClass(cls) -> None:
        cls.logs.debug('自動化接口用例結束')
 
    def setUp(self) -> None:
        self.logs.debug('開始本條接口用例')
 
    def tearDown(self) -> None:
        self.logs.debug('結束本條用例')
 
    @ddt.data(*r) #  引入ddt模塊，讀取拿到的數(shù)據(jù)
    def test_logins(self,pars):  # 用例方法名開頭必須已test  pars參數(shù)為接收的表數(shù)據(jù)值
        import json  #導入json模塊
        dic = json.loads(pars['body參數(shù)值'])  # 將Excel數(shù)據(jù)中的參數(shù)值轉變?yōu)閖son格式
        url = pars['接口地址']  # 拿到請求url
        yuqi = pars['預期結果']  # 拿到預期結果
        fs = pars['請求方式'] # 拿到請求方式
        result = Base().requests_type(method = fs,url = url,data = dic)  # 填充base頁的請求api
        self.assertEqual(result.text,yuqi)  # 進行斷言 看用例是否通過

if __name__ == '__main__':
    load = unittest.TestLoader().loadTestsFromTestCase(TestLogin)  #使用loader加載方式 來找尋所有已test開頭的用例
    suite = unittest.TestSuite([load,])
    common.Common().GetHtmlResult(suite,'登錄測試用例')