官網解釋：用於在請求之間執行Web請求和保留（會話）cookie的類（以便能夠登入和退出網站）；記錄每個請求，以便locust可以顯示統計資訊；

from locust import TaskSet, task, HttpLocust, Locust
from locust.clients import HttpSession
import subprocess
import random
from common.sec_encrypt import AESEncrypt, md5_time
from common import read_config
from common.logger import
 
 print_file


class WebUser(TaskSet):
    def on_start(self):
        self.session = HttpSession('http://localhost:8000/api')
        self.r = random.randint(10000, 99999)

    @task(2)
    def user_auth(self):
        """使用者認證"""
        auth_user = ('lixiaofeng', 'fengzi802300')
        res = self.session.get('
 
/sec_get_event_list/', params={'eid': self.r}, auth=auth_user,
                               name='/sec_get_event_list/?eid=[eid]')
        print_file(res.json())
        # assert res.json()['status'] == 200, '使用者認證介面請求失敗！'

    @task(1)
    def aes_api(self):
        payload = {'name': '187187011{}'.format(self.r)}
        
 
# 加密
        encodeed = AESEncrypt(payload).encrypt_data()
        # print(encodeed)
        res = self.session.get('/sec_get_guest_list/', params={'data': encodeed},
                               name='/sec_get_guest_list/?data=[aes]')
        print_file(res.text)
        # assert res.json()['status'] == 200, 'AES介面加密請求失敗！'

    @task(2)
    def md5_api(self):
        payload = {"eid": "{}".format(self.r), "name": "一加8手機發佈會{}".format(str(self.r)), "limit": 2000, "status": "1",
                   "address": "深圳寶體會展中心",
                   "start_time": "2019-09-15 22:40:00", 'time': '', 'sign': ''}
        payload = md5_time(payload)
        res = self.session.post('/sec_add_event/', data=payload, name='/sec_add_event/?data=[md5]')
        print_file(res.text)
        # assert res.json()['status'] == 200, 'md5介面加密請求失敗！'

    @task(2)
    def add_guest(self):
        """新增嘉賓"""
        payload = {'eid': '{}'.format(self.r), 'realname': '趙小刀{}'.format(self.r),
                   'phone': '187011{}'.format(self.r),
                   'email': '187011{}@163.com'.format(self.r)}
        res = self.session.post('/add_guest/', data=payload, name='/add_guest/?data=[payload]')
        print_file(res.text)


class Run(Locust):
    # host = 'http://localhost:8000/api'
    task_set = WebUser
    max_wait = 6000
    min_wait = 3000

def run():
    subprocess.check_call(
        'locust -f D:\Interface_framework_Beauty\locust_files\\locust_test.py')


if __name__ == '__main__':
    run()

print_file，基於print函式，可以使locust列印寫入日誌檔案的方法；使用HttpSession類，需要傳入url引數;Run類，需要繼承Locust類，而不是HttpLocust類；name='/sec_add_event/?data=[md5]'，將請求分組，便於檢視；