# !/usr/bin/python3
|
# -*- coding: utf-8 -*-
|
import datetime
|
import os
|
import random
|
import string
|
import time
|
|
from requests_toolbelt import MultipartEncoder
|
|
from httprunner.compat import builtin_str, integer_types
|
from httprunner.exceptions import ParamsError
|
|
|
def gen_random_string(str_len):
|
"""generate random string with specified length"""
|
return "".join(
|
random.choice(string.ascii_letters + string.digits) for _ in range(str_len)
|
)
|
|
|
def get_timestamp(str_len=13):
|
"""get timestamp string, length can only between 0 and 16"""
|
if isinstance(str_len, integer_types) and 0 < str_len < 17:
|
return builtin_str(time.time()).replace(".", "")[:str_len]
|
|
raise ParamsError("timestamp length can only between 0 and 16.")
|
|
|
def get_current_date(fmt="%Y-%m-%d"):
|
"""get current date, default format is %Y-%m-%d"""
|
return datetime.datetime.now().strftime(fmt)
|
|
|
def multipart_encoder(field_name, file_path, file_type=None, file_headers=None):
|
if not os.path.isabs(file_path):
|
file_path = os.path.join(os.getcwd(), file_path)
|
|
filename = os.path.basename(file_path)
|
with open(file_path, "rb") as f:
|
fields = {field_name: (filename, f.read(), file_type)}
|
|
return MultipartEncoder(fields)
|
|
|
def multipart_content_type(multipart_encoder):
|
return multipart_encoder.content_type
|
|
|
""" built-in hooks
|
"""
|
|
|
def setup_hook_prepare_kwargs(request):
|
pass
|
