接口数据写入CSV

2022-09-05 23:31:22 阅读：200 来源： 互联网

标签：csv url res 写入接口 ---- print SendRequest CSV

# api_util.py

import requests


class SendRequest:
    sess = requests.session()

    def all_send_request(self, method, url, **kwargs):
        print('\n----接口测试开始----')
        # print('\n----请求url:%s----'%url)
        # print('\n----**kwargs:%s----'%kwargs)
        res = SendRequest.sess.request(method, url, **kwargs)
        print('\n----接口测试结束----')
        return res

    def login_system(self):
        method = 'post'
        url = 'http://127.0.0.1:80/api/mgr/signin'
        data = {
            'username': 'byhy',
            'password': '88888888'
        }
        try:
            res = SendRequest().all_send_request(method, url, data=data)
        except BaseException as e:
            print('\n****登录系统失败****')
        else:
            print('\n****登录系统成功****')
        # return res

# test_writecsv.py

import csv
import json
import time

from common.api_util import SendRequest


class TestApi:
    # 前置步骤
    def setup(self):
        print('\n****执行前置步骤:登录系统****')
        SendRequest().login_system()

    # 用例
    def test_customer_list(self):
        method = 'get'
        url = 'http://127.0.0.1:80/api/mgr/customers'
        params = {
            'action': 'list_customer',
            'pagesize': 100,
            'pagenum': 1,
            'keywords': ''
        }
        res = SendRequest().all_send_request(method, url, params=params)
        print(res.json())
        assert res.status_code == 200
        # 结果写入csv文件
        # 文件地址与名称
        filepath = f'customerlist_{time.strftime("%Y%m%d%H%M%S",time.localtime())}.csv'
        header = ["id", "name", "phonenumber", "address"]
        header = []
        #  获取表头
        dict1 = res.json()['retlist'][0]

        for key in dict1.keys():
            header.append(key)
        # print(header)
        # 获取数据列表
        rows = res.json()['retlist']
        with open(filepath, 'w', newline='', encoding='utf-8') as f:
            f_csv = csv.DictWriter(f, fieldnames=header)
            f_csv.writeheader() # 写入header
            f_csv.writerows(rows) # 写入列表数据

        return res

结果:

标签：csv,url,res,写入,接口,----,print,SendRequest,CSV
来源： https://www.cnblogs.com/Avicii2018/p/16660054.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

接口数据写入CSV