142 lines
4.8 KiB
Python
142 lines
4.8 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
Created on 2018年10月24日
|
||
@author: Irony
|
||
@site: https://github.com/892768447
|
||
@email: 892768447@qq.com
|
||
@file: AsyncioUiClient
|
||
@description:
|
||
"""
|
||
import asyncio
|
||
|
||
from PyQt5.QtCore import Qt
|
||
from PyQt5.QtGui import QPixmap, QMovie
|
||
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QPushButton,\
|
||
QApplication, QListWidget, QListWidgetItem, QLabel, QMessageBox
|
||
import aiohttp
|
||
from quamash import QEventLoop
|
||
|
||
|
||
__Author__ = """By: Irony
|
||
QQ: 892768447
|
||
Email: 892768447@qq.com"""
|
||
__Copyright__ = "Copyright (c) 2018 Irony"
|
||
__Version__ = "Version 1.0"
|
||
|
||
Url = 'https://www.doutula.com/api/search?keyword=%E6%9C%80%E6%96%B0%E8%A1%A8%E6%83%85&mime=0&page={}'
|
||
Headers = {
|
||
':authority': 'www.doutula.com',
|
||
':method': 'GET',
|
||
':scheme': 'https',
|
||
'accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8',
|
||
'accept-language': 'zh-CN,zh;q=0.9',
|
||
'cache-control': 'max-age=0',
|
||
'dnt': '1',
|
||
'upgrade-insecure-requests': '1',
|
||
'user-agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.26 Safari/537.36 Core/1.63.6756.400 QQBrowser/10.2.2498.400'
|
||
}
|
||
|
||
|
||
class Window(QWidget):
|
||
|
||
def __init__(self, *args, **kwargs):
|
||
super(Window, self).__init__(*args, **kwargs)
|
||
layout = QVBoxLayout(self)
|
||
self.listWidget = QListWidget(self)
|
||
self.listWidget.setSpacing(2) # item直接的间隔
|
||
# 隐藏横向滚动条
|
||
self.listWidget.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
|
||
# 让list 从左到右排列
|
||
self.listWidget.setFlow(self.listWidget.LeftToRight)
|
||
# 自动换行
|
||
self.listWidget.setWrapping(True)
|
||
self.listWidget.setResizeMode(self.listWidget.Adjust)
|
||
|
||
self.buttonMsg = QPushButton('弹出提示框', self, clicked=self.showMessage)
|
||
self.buttonDown = QPushButton('下载图片', self, clicked=self.doDownload)
|
||
layout.addWidget(self.listWidget)
|
||
layout.addWidget(self.buttonMsg)
|
||
layout.addWidget(self.buttonDown)
|
||
self.currentPage = 0
|
||
self.initSession() # 其实没必要，session主要用在需要登录的网站。缓存cookie用
|
||
|
||
def initSession(self):
|
||
async def _initSession():
|
||
# 初始化session
|
||
self.session = aiohttp.ClientSession(loop=loop)
|
||
print(self.session)
|
||
asyncio.ensure_future(_initSession(), loop=loop)
|
||
|
||
async def _doDownloadImage(self, url):
|
||
# 下载图片并添加到界面
|
||
async with self.session.get(url) as resp:
|
||
data = await resp.read()
|
||
if not data:
|
||
print('下载失败: ', url)
|
||
return
|
||
path = os.path.join('tmp', os.path.basename(url))
|
||
with open(path, 'wb') as fp:
|
||
fp.write(data)
|
||
item = QListWidgetItem(url, self.listWidget)
|
||
image = QPixmap(path)
|
||
item.setSizeHint(image.size())
|
||
label = QLabel(self.listWidget)
|
||
label.setPixmap(image)
|
||
if path.endswith('.gif'): # 可能是动态图
|
||
label.setMovie(QMovie(path))
|
||
self.listWidget.setItemWidget(item, label)
|
||
self.listWidget.scrollToBottom()
|
||
|
||
async def _doDownload(self):
|
||
# 下载工作
|
||
if self.currentPage == -1:
|
||
QMessageBox.information(self, '提示', '已经没有更多了')
|
||
return
|
||
self.currentPage += 1
|
||
url = Url.format(self.currentPage)
|
||
print('get url: ', url)
|
||
async with self.session.get(url, headers=Headers) as resp:
|
||
data = await resp.json()
|
||
if not data:
|
||
return
|
||
data = data.get('data', None)
|
||
if not data:
|
||
self.currentPage = -1
|
||
print('已经是最后一页了')
|
||
return
|
||
# 解析json并生成item添加到界面中
|
||
for entity in data.get('list', []):
|
||
url = entity.get('image_url', None)
|
||
if not url:
|
||
continue
|
||
await self._doDownloadImage(url) # 下载图片
|
||
|
||
def doDownload(self):
|
||
# 响应按钮点击调用
|
||
asyncio.ensure_future(self._doDownload(), loop=loop)
|
||
|
||
def showMessage(self):
|
||
# 显示对话框
|
||
app.aboutQt()
|
||
|
||
def closeEvent(self, event):
|
||
if not self.session.closed:
|
||
asyncio.ensure_future(self.session.close(), loop=loop)
|
||
super(Window, self).closeEvent(event)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
import sys
|
||
import cgitb
|
||
import os
|
||
os.makedirs('tmp', exist_ok=True)
|
||
sys.excepthook = cgitb.enable(1, None, 5, 'text')
|
||
app = QApplication(sys.argv)
|
||
loop = QEventLoop(app)
|
||
asyncio.set_event_loop(loop)
|
||
w = Window()
|
||
w.show()
|
||
with loop:
|
||
loop.run_forever()
|