python通過接口上傳圖片造測試數據

最近客戶端上有個發送圖文的需求，大致是添加圖片，文本數據然后發送。

一般像這樣的情況都是有個單獨上傳圖片的接口，返回上傳圖片的路徑，發送信息的接口接收這個圖片地圖參數，信息存儲的數據庫。

測試中期需要造一些數據來進行測試及觀察整體效果，圖片上傳是個很費事的手機活，那么我又想到了python http神器requests

還是直接上代碼加注釋吧

#-*-coding:utf-8 -*-
__author__ = 'DongJie'
import requests
import glob
import os
import time
import random
import urllib
import sys
reload(sys)
sys.setdefaultencoding('utf-8')

#選擇測試（從測試圖片目錄隨機選擇多張圖片）
def chosePic(number):
    pic_list = glob.glob('E:\\testpic\\*.jpg')
    up_pic = random.sample(pic_list, number)
    return up_pic


#上傳測試圖片（通過接口將圖片上傳至服務器,得到服務器返回的路徑：http圖片上傳是以二進制附件流上傳到服務器的）
def upPic(pic_list):
    up_url = '127.0.0.1/api/User?sort=tourDating'
    re = []
    for pic in pic_list:
        f = open(u'%s' % pic, 'rb')
        files = {'file':[os.path.split(pic)[-1], f, 'application/octet-stream']}
        req = requests.post(up_url, files=files)
        server_path = req.json()[0]
        re.append(server_path)
    path = ','.join(re)
    return path


#這是一個應該用接口
def sendYue(account, content, startTime, endTime, lat, lng, address, isNearVisible, picList, contactInformation):
    add = "http://127.0.0.1/api/TourDating/Publish"
    value = {'accountId':account,
             'content':content,
             'startTime':startTime,
             'endTime':endTime,
             'lat':lat,
             'lng':lng,
             'address':address,
             'isNearVisible':isNearVisible,
             'picList':picList,
             'contactInformation':contactInformation}
    #必須用urlencode將參數值編碼
    args = urllib.urlencode(value)
    send_url = add + '?' + args
    try:
        req = requests.post(send_url)
        return req.json()
    except Exception, e:
        print e


if __name__ == "__main__":
    #從account取已經從數據庫取出來用戶ID數據
    account_list = open('account', 'r').readlines()
    #地置數據（位置名稱\t經度\t緯度）當然也可以做成你自己文本格式，容易解析最好
    where = open('coordinate', 'r').readlines()
    position = [x.strip().split('\t') for x in where if x!='']
    content_all = open('content', 'r').read()
    for x in range(180):
        account = account_list[x].strip()
        address = position[x][0]
        lng = position[x][1]
        lat = position[x][2]
        #內容也是從一個文本里面隨機（10-140個字）這個根據自己的需要
        content = ''.join(random.sample(content_all.decode('utf-8'), random.randint(10, 140)))
        #下面是我隨機開始時間和結束時間的方法（從一個時間戳段中取一個值，然后往后推隨機天數）
        t = random.randint(1433541966,1451581261)
        startTime = time.strftime('%Y-%m-%d', time.localtime(t))
        endTime = time.strftime('%Y-%m-%d',time.localtime(t+random.randint(1,60)*86400))
        #這個參數也是一個隨機數了
        contactInformation = random.randint(111111,19999999999)
        #0和1隨機取一個
        isNearVisible = random.randint(0,1)
        #上傳圖片返回的路徑在這里用到
        picList = upPic(chosePic(random.randint(1,3)))
        #發送請求造數據
        sendYue(account, content, startTime, endTime, lat, lng, address, isNearVisible, picList, contactInformation)

代碼很多地方偷了懶，特別是對文件操作，大家別學。