一、基础设计

实验环境 Python3+jupyter
一、程序框架


图片.png

二、工具层 utils.py
1. 单例函数
类只能实例化一次

def singleton(cls):
    instance = cls()
    instance.__call__ = lambda: instance
    return instance

三、网络层 jquery.py
要求:
1. 实现get、post请求
2. 添加cookie和headers
3. 处理http状态码异常
4. 判断返回数据是json 还是str
5. 每个用户的cookies独立
6. 实现cookie文件和cookie的转化
核心目标还是要方便简单
通过requests.session+cookiejar 来自动管理cookie的问题
在初始化函数中传入参数headers来给请求添加默认头部
cookie_file确保了session不会丢失,可重新运行
_type是为了区分开json数据和原数据

代码:

import json,requests
import  http.cookiejar as cookielib
from utils import singleton

class session(requests.Session):
    def __init__(self,headers={},cookie_file=None):
        requests.Session.__init__(self)
        self.cookies = cookielib.LWPCookieJar(filename=cookie_file)
        try:self.cookies.load(ignore_discard=True)
        except:print("failed load cookie_file")
        self.headers=headers
        self.auth = ('user', 'pass')
        self._type='str'
        
    def save(self):
        self.cookies.save()
@singleton
class http():
    def get(self,s,url,data=None):
        r=s.get(url,params=data)
        self.error(r.status_code)
        return self.res_data(r,s._type)
        
    def post(self,s,url,data):
        r=s.post(url,data=data)
        self.error(r.status_code)
        return self.res_data(r,s._type)
        
    def error(self,code):
        if code==200:return
        elif code==403:raise RuntimeError('403 - Forbidden')
        elif code==404:raise RuntimeError('404 -  Not Found')
        
    def res_data(self,r,_type):
        if _type=='json':return json.loads(r.content)
        return r.content

四、代码测试
获取b站直播房间的flv下载地址

headers={
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:62.0) Gecko/20100101 Firefox/62.0',
    'Accept': 'application/json, text/plain, */*',
    'Accept-Language': 'zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2',
    'Accept-Encoding': 'gzip, deflate, br',
    'Referer': 'https://live.bilibili.com/',
    'Origin': 'https://live.bilibili.com',
    'Connection': 'keep-alive'
    }
s=session(headers,'cookie.txt')
s._type='json'
url='https://api.live.bilibili.com/room/v1/Room/playUrl'
data={
    'cid':7603080,
    'quality':0,
    'platform': 'web'
}
r=http.get(s,url,data)
for d in r['data']['durl']:print(d['url'])

下载地址

failed load cookie_file
[https://qn.live-play.acgvideo.com/live-qn/165966/live_267605959_3403844.flv?wsSecret=06028657c2cc08ff816df9d9d505a5a3&wsTime=1538217562](https://qn.live-play.acgvideo.com/live-qn/165966/live_267605959_3403844.flv?wsSecret=06028657c2cc08ff816df9d9d505a5a3&wsTime=1538217562)
[https://qn.live-play.acgvideo.com/live-qn/165966/live_267605959_3403844.flv?wsSecret=06028657c2cc08ff816df9d9d505a5a3&wsTime=1538217562](https://qn.live-play.acgvideo.com/live-qn/165966/live_267605959_3403844.flv?wsSecret=06028657c2cc08ff816df9d9d505a5a3&wsTime=1538217562)
[https://qn.live-play.acgvideo.com/live-qn/165966/live_267605959_3403844.flv?wsSecret=06028657c2cc08ff816df9d9d505a5a3&wsTime=1538217562](https://qn.live-play.acgvideo.com/live-qn/165966/live_267605959_3403844.flv?wsSecret=06028657c2cc08ff816df9d9d505a5a3&wsTime=1538217562)
[https://qn.live-play.acgvideo.com/live-qn/165966/live_267605959_3403844.flv?wsSecret=06028657c2cc08ff816df9d9d505a5a3&wsTime=1538217562](https://qn.live-play.acgvideo.com/live-qn/165966/live_267605959_3403844.flv?wsSecret=06028657c2cc08ff816df9d9d505a5a3&wsTime=1538217562)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容