三、auth自定义授权
客户端代码：
import requests
import hashlib
import time
current_time = time.time()
#自意义的字符串app_id,和服务端ck一致
app_id =‘8kasoimnasodn8687asdfkmasdf‘
app_id_time = "%s|%s" % (app_id,current_time,)
m = hashlib.md5()
m.update(bytes(app_id_time,encoding=‘utf-8‘))
authkey = m.hexdigest()
authkey_time ="%s|%s" % (authkey,current_time,)
print(authkey_time)
host_data = {
‘status‘:True,
‘data‘:{
‘hostname‘:‘c1.com‘,
‘disk‘:{‘status‘:True,‘data‘:‘xxx‘},
‘mem‘:{‘status‘:True,‘data‘:‘xxx‘},
‘nic‘:{‘status‘:True,‘data‘:‘xxx‘},
},
}
response=requests.post(
url=‘http://127.0.0.1:8000/api/asset/‘,
json=host_data,
headers={‘authkey‘:authkey_time},
)
print(response.text)
# requests.get(url=‘http://127.0.0.1:8000/api/asset/?k1=123‘)
# requests.get(url=‘http://127.0.0.1:8000/api/asset/‘,params={‘k1‘:‘v1‘,‘k2‘:‘v2‘})
# requests.post(
# url=‘http://127.0.0.1:8000/api/asset/‘,
# params={‘k1‘:‘v1‘,‘k2‘:‘v2‘}, # GET形式传值
# data={‘username‘:‘1123‘,‘pwd‘: ‘666‘}, # POST形式传值
# headers={‘a‘:‘123‘} # 请求头数据
# )
服务端代码：
from django.shortcuts import render,HttpResponse
from django.views.decorators.csrf import csrf_exempt,csrf_protect
import hashlib
import time
#自定义的字符串
ck=‘8kasoimnasodn8687asdfkmasdf‘
#5秒钟的授权列表
auth_list = []
@csrf_exempt
def asset(request):
auth_key_time = request.META.get(‘HTTP_AUTHKEY‘)
auth_key_client,client_ctime = auth_key_time.split(‘|‘)
server_current_time = time.time()
if server_current_time - 5 > float(client_ctime):
#太久远了
return HttpResponse(‘时间太久远了‘)
if auth_key_time in auth_list:
#已经访问过了
return HttpResponse(‘你来晚了‘)
key_time = ‘%s|%s‘ %(ck,client_ctime)
m = hashlib.md5()
m.update(bytes(key_time,encoding=‘utf-8‘))
authkey = m.hexdigest()
if authkey != auth_key_client:
return HttpResponse(‘授权失败‘)
auth_list.append(auth_key_time)
if request.method == ‘POST‘:
import json
print(authkey)
host_info = json.loads(str(request.body,encoding=‘utf-8‘))
print(host_info)
return HttpResponse(‘授权成功‘)
def test(request):
# print(request.POST,type(request.POST))
# from django.http.request import QueryDict
response = render(request,‘index.html‘)
response.set_signed_cookie(‘kkkk‘,‘vvvv‘,salt=‘asdf‘)
return response
线程池和进程池
#!/usr/bin/env python
# -*- coding:utf8 -*-
####### 编写方式一 ##########
from concurrent.futures import ThreadPoolExecutor
from concurrent.futures import ProcessPoolExecutor
import requests
import time
def task(url):
response = requests.get(url)
print(url,response)
pool = ThreadPoolExecutor(5)
url_list = [
‘https://www.baidu.com‘,
‘http://www.sina.com.cn‘,
‘http://cn.bing.com‘,
‘https://home.cnblogs.com/u/liaoboshi/‘,
‘https://www.linkedin.com‘,
‘http://mil.news.baidu.com‘,
]
for url in url_list:
pool.submit(task,url)
pool.shutdown(wait=True)
####### 编写方式一 ################################
from concurrent.futures import ThreadPoolExecutor
import requests
import time
def task(url):
‘‘‘
下载页面
:param url:
:return:
‘‘‘
response = requests.get(url)
return response
def done(future,*args,**kwargs):
response = future.result()
print(response.status_code,response.content)
pool = ThreadPoolExecutor(5)
url_list = [
‘https://www.baidu.com‘,
‘http://www.sina.com.cn‘,
‘http://cn.bing.com‘,
‘https://home.cnblogs.com/u/liaoboshi/‘,
‘https://www.linkedin.com‘,
‘http://mil.news.baidu.com‘,
]
for url in url_list:
v = pool.submit(task,url)
# 每一个线程函数走完，再走下面的另一个回调函数
v.add_done_callback(done)
pool.shutdown(wait=True)
自定义异步IO框架
#!/usr/bin/env python
# -*- coding:utf8 -*-
# IO多路复用: 监听多个socket对象，感知变化，利用其特性可以并发出异步IO模块
# 异步IO: 异步是非阻塞 非阻塞 ＋ IO多路复用
# setblocking(False)
import select
import socket
class HttpRequest:
def __init__(self,sk,host,callback):
self.socket = sk
self.host = host
self.callback = callback
def fileno(self):
return self.socket.fileno()
class HttpResponse:
def __init__(self,recv_data):
self.recv_data = recv_data
self.header_dict = {}
self.body = None
self.initialize()
def initialize(self):
headers, body = self.recv_data.split(b‘\r\n\r\n‘, 1)
self.body = body
header_list = headers.split(b‘\r\n‘)
for h in header_list:
h_str = str(h, encoding=‘utf-8‘)
v = h_str.split(‘:‘, 1)
if len(v) == 2:
self.header_dict[v[0]] = v[1]
class AsyncRequest:
def __init__(self):
self.conn = []
self.connection = [] # 用于检测是否已经连接成功
def add_request(self,host,callback):
try:
sk = socket.socket() # 创建 socket 对象
sk.setblocking(False) # 设置socket为非阻塞
sk.connect((host,80,)) # 连接 主机
except BlockingIOError as e: # 设置socket为非阻塞后，会报错，要抓住异常
pass
request = HttpRequest(sk,host,callback) # 创建一个socket对象 要返回self.socket.fileno()
self.conn.append(request) # 把对象加到列表里
self.connection.append(request) # 把对象加到列表里
def run(self):
while True:
rlist,wlist,elist = select.select(self.conn,self.connection,self.conn,0.05) # 创建select对象
for w in wlist:
print(w.host,‘连接成功...‘)
# 只要能循环到，表示socket和服务器端已经连接成功
tpl = ‘GET / HTTP/1.0\r\nHost:%s\r\n\r\n‘ % w.host
w.socket.send(bytes(tpl,encoding=‘utf8‘)) # 给发服务器送消息
self.connection.remove(w) # 发送完消息后，删除对象
for r in rlist:
# r,是HttpRequest
recv_data = bytes()
while True:
try:
chunck = r.socket.recv(8096) # 接收服务器返回消息
recv_data += chunck
except Exception as e:
break
response = HttpResponse(recv_data) # 返回的消息包装成字典（请求头和请求体）
r.callback(response) # 执行回调函数
r.socket.close() # 关闭连接
self.conn.remove(r) #删除对象
if len(self.conn) == 0:
break
def f1(response): # 回调函数拿到返回的请求头和请求体
print(‘保存到文件‘, response.header_dict)
def f2(response):
print(‘保存到数据库‘, response.header_dict)
url_list = [
{‘host‘: ‘www.baidu.com‘, ‘callback‘: f1},
{‘host‘: ‘cn.bing.com‘, ‘callback‘: f2},
{‘host‘: ‘www.cnblogs.com‘, ‘callback‘: f2},
]
req = AsyncRequest() # 创建一个对象
for item in url_list:
req.add_request(item[‘host‘],item[‘callback‘]) # 运行类的add_request方法，把 主机名 和 回调函数 传进去
req.run() # 运行类的run方法
自定义异步IO框架
时间： 2024-08-25 22:04:40