def upload_file(request):
"""
实现文件上传
:param request:
:return:
"""
if request.method == "POST":
f = request.FILES.get("file")
print("文件对象", type(f), f)
print("文件name", f.name)
file_name = f.name
ff = file_name.split(".")[-1]
print(ff)
print("文件大小=》", f.size)
if ff == "exe" or ff == "js":
return ApiResponse(status=102, message="不支持这种类型的上传")
# 保存上传的文件
upload_dir = os.path.join(BASE_DIR, "api/upload")
print(upload_dir)
destination = open(os.path.join(upload_dir, file_name),
'wb+') # 打开特定的文件进行二进制的写操作
for chunk in f.chunks(): # 分块写入文件
destination.write(chunk)
destination.close()
return ApiResponse(status=200, message="successful")
def get_user(request):
"""
获取抽奖用户
"""
if request.method == "GET":
user_info = {"id": 1, "name": "张三"}
return ApiResponse(200, data=user_info)
else:
return ApiResponse(101, "request method error")
def user_login_data(request):
"""
通过Session读取用户数据
"""
username = request.session.get('user', '') # 读取浏览器 session
if username == "":
return ApiResponse(200, 'hello, stranger')
else:
return ApiResponse(200, 'hello, {}'.format(username))
def get_activity(request):
"""
获取抽奖活动
:return:
"""
if request.method == "GET":
activity_info = {"id": 1, "name": "618抽奖活动"}
return ApiResponse(200, data=activity_info)
else:
return ApiResponse(101, "request method error")
def user(request, uid):
"""
RESTful 风格的接口实现
实现用户的查询、添加、更新和删除
/event/1/ = GET
/event/2/ = POST
/event/1/ = PUT
/event/1/ = DELETE
:param request:
:param uid:
:return:
"""
if request.method == "GET":
try:
user = User.objects.get(id=uid)
except User.DoesNotExist:
return ApiResponse(101, message="用户信息不存在")
else:
user_info = model_to_dict(user)
return ApiResponse(200, data=user_info)
elif request.method == "POST":
post = json.loads(request.body)
name = post.get("name")
age = post.get("age")
try:
User.objects.get(id=uid)
except User.DoesNotExist:
user = User.objects.create(id=uid, name=name, age=age)
user_info = model_to_dict(user)
return ApiResponse(200, "add success", user_info)
else:
return ApiResponse(102, "用户id已存在")
elif request.method == "PUT":
put = json.loads(request.body)
name = put.get("name")
age = put.get("age")
try:
user = User.objects.get(id=uid)
except User.DoesNotExist:
return ApiResponse(101, "用户信息不存在")
else:
user.name = name
user.age = age
user.save()
user_info = model_to_dict(user)
return ApiResponse(200, "update success", user_info)
elif request.method == "DELETE":
try:
user = User.objects.get(id=uid)
except User.DoesNotExist:
return ApiResponse(101, "用户信息不存在")
else:
user.delete()
return ApiResponse(200, "delete success")
else:
return ApiResponse(101, "request method error")
def get_user_info(request, uid):
"""
根据用户id返回数据
"""
if request.method == "GET":
user_info = {"id": 1, "name": "tom", "age": 22}
if uid != 1:
return ApiResponse(101, "user id null")
else:
return ApiResponse(200, data=user_info)
else:
return ApiResponse(101, "request method error")
def search(request):
"""
一般GET请求
"""
if request.method == "GET":
search_word = request.GET["q"]
if search_word == "selenium":
data_list = ["selenium教程", "seleniumhq.org", "selenium环境安装"]
else:
data_list = []
return ApiResponse(200, data=data_list)
else:
return ApiResponse(101, "request method error")
def upload_file(request):
"""
文件上传
"""
if request.method == "POST":
my_file = request.FILES.get("file", None)
if not my_file:
return ApiResponse(101, "no files for upload!")
destination = open(os.path.join(BASE_DIR, "api/upload", my_file.name),
'wb+')
for chunk in my_file.chunks():
destination.write(chunk)
destination.close()
return ApiResponse(200, "upload success!")
def post_header(request):
"""
请求方法：POST
带请求提头：Header
"""
if request.method == 'POST':
token = request.headers.get("token")
ct = request.headers.get("Content-Type")
return ApiResponse(200, "header ok!", {
"token": token,
"Content-Type": ct
})
else:
return ApiResponse(101, "request method error")
def hello(request):
"""
最简单的json格式返回
:param request:
:return:
"""
return ApiResponse(status=100, message="Welcome to API testing")
def user_aes(request):
if request.method == 'POST':
data = request.POST.get("data", "")
if data == "":
return ApiResponse(102, "data null")
# 解密
decode = decryptAES(data)
# 转化为字典
dict_data = json.loads(decode)
print(dict_data)
return ApiResponse(200, "success")
else:
return ApiResponse(101, "请求方法错误")
def user_auth(request):
"""
用户认证
"""
if request.method == "GET":
auth_str = request.META["HTTP_AUTHORIZATION"]
auth_ = base64.b64decode(auth_str.split(" ")[1])
user_ = auth_.decode().split(":")[0]
pawd_ = auth_.decode().split(":")[1]
print(user_)
print(pawd_)
user = auth.authenticate(username=user_, password=pawd_)
if user is None:
return ApiResponse(102, "认证失败")
return ApiResponse(200, "认证通过")
else:
return ApiResponse(101, "请求方法错误")
def user_login(request):
"""
通过Session 记录登录状态
"""
if request.method == "POST":
login_username = request.POST.get("username")
login_password = request.POST.get("password")
if login_username == '' or login_password == '':
return ApiResponse(102, "username or password null")
else:
user = auth.authenticate(username=login_username,
password=login_password)
if user is not None:
auth.login(request, user)
# 将 session 信息记录到浏览器
request.session['user'] = login_username
return ApiResponse(200, "login success")
else:
return ApiResponse(103, "username or password error")
else:
return ApiResponse(101, "request method error")
def get_lucky_number(request):
"""
获取抽奖号码
"""
if request.method == "POST":
activity_id = request.form.get('aid')
user_id = request.form.get('uid')
if activity_id is None or user_id is None:
return ApiResponse(102, "username or password is None")
elif activity_id == "" or user_id == "":
return ApiResponse(10103, "username or password is null")
if int(activity_id) != 1:
return ApiResponse(104, "activity id exist")
if int(user_id) != 1:
return ApiResponse(105, "user id not exist")
number = randint(10000, 99999)
return ApiResponse(200, "Lucky draw number", number)
else:
return ApiResponse(101, "request method error")
def user_sign(request):
print(request.method)
if request.method == 'POST':
client_time = request.POST.get('time', '') # 客户端时间戳
client_sign = request.POST.get('sign', '') # 客户端签名
if client_time == '' or client_sign == '':
return ApiResponse(102, "sign null")
# 服务器时间
server_time = str(time.time()).split('.')[0]
print("客户端端时间:", client_time)
print("服务器端时间:", server_time)
# 获取时间差
time_difference = int(server_time) - int(client_time)
if time_difference >= 3:
return ApiResponse(102, "timeout")
# 签名检查
md5 = hashlib.md5()
sign_str = client_time + "&Guest-Bugmaster"
sign_bytes_utf8 = sign_str.encode(encoding="utf-8")
md5.update(sign_bytes_utf8)
sever_sign = md5.hexdigest()
print("客户端端签名:", client_sign)
print("服务器端签名:", sever_sign)
if sever_sign != client_sign:
return ApiResponse(103, "sign fail")
else:
return ApiResponse(200,
"sign success",
data={"password": "******"})
else:
return ApiResponse(101, "请求方法错误")
def post_add_user(request):
"""
请求方法：POST
参数类型：JSON
"""
if request.method == "POST":
try:
data = json.loads(request.body)
except json.decoder.JSONDecodeError:
return ApiResponse(105, "format error")
try:
name = data["name"]
age = data["age"]
height = data["height"]
except KeyError:
return ApiResponse(102, "key null")
else:
if name == "":
return ApiResponse(103, "name null")
elif name == "tom":
return ApiResponse(104, "name exist")
else:
data = {"name": name, "age": age, "height": height}
return ApiResponse(200, "add successful", data)
else:
return ApiResponse(101, "request method error")
def post_login(request):
"""
请求方法：POST
参数类型：from-data/x-www-from-urlencode
"""
if request.method == "POST":
username = request.POST('username')
password = request.POST('password')
if username is None or password is None:
return ApiResponse(102, "username or passwrord is None")
elif username == "" or password == "":
return ApiResponse(103, "username or passwrord is null")
elif username == "admin" and password == "a123456":
return ApiResponse(200, "login success")
else:
return ApiResponse(104, "username or password error")
else:
return ApiResponse(101, "request method error")
def header(request):
"""
获取 请求头Header的数据
:param request:
:return:
"""
print("header => ", type(request.headers), request.headers)
ct = request.headers["Content-Type"]
token = request.headers["Token"]
print("ct--->", ct)
print("token--->", token)
return ApiResponse(status=200, message="successful")
def post_req(request):
"""
post请求的几种参数：
添加用例
:param request:
:return:
"""
if request.method == "POST":
print("request==>", request.body)
# form-data 和 x-www-from-urlencode:
name = request.POST.get("name", "")
print("name-->", name)
# JSON
body = json.loads(request.body)
name = body.get("name")
print("json-name-->", name)
return ApiResponse(status=200, message="successful")
def user(request, username):
"""
通过 URI 传参
"""
msg = "hello, {}".format(username)
return ApiResponse(200, msg)
def hello_world(request):
"""
最简单的get请求，返回json格式返回
"""
return ApiResponse(200, "Welcome to API testing")