mirror of
https://github.com/Ai-Thinker-Open/TBXX_Flash_Tool.git
synced 2025-05-08 22:37:57 +08:00
208 lines
7.6 KiB
Python
208 lines
7.6 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import argparse
|
|
import sys
|
|
import io
|
|
import os
|
|
import struct
|
|
import time
|
|
import zlib
|
|
import hashlib
|
|
import requests
|
|
from contextlib import closing
|
|
from lxml import etree
|
|
import re
|
|
import markdown2
|
|
from PyQt5.QtWidgets import QPushButton,QLineEdit,QWidget,QTextEdit,QVBoxLayout,QHBoxLayout,QFileDialog,QLabel
|
|
from PyQt5.QtWidgets import QTableWidget,QTableWidgetItem,QAbstractItemView,QFrame,QHeaderView,QMessageBox
|
|
from PyQt5.QtCore import Qt,QThread,pyqtSignal
|
|
from PyQt5.QtGui import QIcon,QPalette,QColor,QFont
|
|
|
|
CMD_SHOW_FORM = 0x00
|
|
CMD_CLOSE_FORM = 0x01
|
|
CMD_UPDATA_OK = 0x02
|
|
CMD_DOWNLOAD_OK = 0x03
|
|
|
|
class DocThread(QThread):
|
|
formSignal = pyqtSignal(int)
|
|
textSignal = pyqtSignal(str)
|
|
presSignal = pyqtSignal(int)
|
|
def __init__(self, action, url, fileName=''):
|
|
super(DocThread, self).__init__()
|
|
|
|
self.action = action
|
|
self.url = url
|
|
self.fileName = fileName
|
|
|
|
self.headers = {'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.77 Safari/537.36'}
|
|
print(action + " : " + url)
|
|
|
|
def run(self):
|
|
|
|
if self.action == "down_doc":#下载Bin文件
|
|
|
|
with closing(requests.get(self.url, headers=self.headers,stream=True)) as response:
|
|
chunkSize = 1024
|
|
dateCount = 0
|
|
with open(self.fileName,"wb") as file:
|
|
for data in response.iter_content(chunk_size=chunkSize):
|
|
file.write(data)
|
|
dateCount = dateCount + len(data)
|
|
self.formSignal.emit(CMD_DOWNLOAD_OK)
|
|
return
|
|
|
|
elif self.action == "get_doc_list":#获取gitee上固件文件列表
|
|
|
|
r = None
|
|
try:
|
|
r = requests.get(self.url, timeout=10, headers=self.headers)
|
|
r.encoding = 'utf-8'#r.apparent_encoding
|
|
except Exception as e:
|
|
self.formSignal.emit(CMD_CLOSE_FORM)
|
|
|
|
if r == None:
|
|
self.formSignal.emit(CMD_CLOSE_FORM)
|
|
return
|
|
|
|
if r.status_code == 200:
|
|
tbodys = re.findall('<div class=\'grid list selection([\w\W]+?)<div class=\'ui tree_progress\'',r.text)
|
|
|
|
if tbodys == None:
|
|
self.formSignal.emit(CMD_CLOSE_FORM)
|
|
return
|
|
|
|
tbody = '<div class=\'grid list selection' + tbodys[0]
|
|
|
|
self.textSignal.emit(tbody)
|
|
|
|
|
|
class Dev_Document(QWidget):
|
|
|
|
def __init__(self,parent=None):
|
|
super().__init__(parent)
|
|
|
|
self.layout=QVBoxLayout(self)
|
|
|
|
palette = QPalette()
|
|
palette.setColor(QPalette.Window, QColor(192,253,123,100))
|
|
|
|
self.waitPage=QLabel(self)
|
|
|
|
self.waitPage.setGeometry(0, 0, 600, 350)
|
|
self.waitPage.setAlignment(Qt.AlignVCenter)
|
|
self.waitPage.setText("<center><font color='red' size='6' line-height='50px';><red>正在获取文档列表......</font></center>")
|
|
self.waitPage.setAutoFillBackground(True)
|
|
self.waitPage.setPalette(palette)
|
|
|
|
self.is_First_Show = True
|
|
|
|
def get_doc_list(self):
|
|
|
|
if not self.is_First_Show: return
|
|
|
|
self.mThread = DocThread(action="get_doc_list", url='https://gitee.com/Ai-Thinker-Open/TB_Dev_Document')
|
|
self.mThread.textSignal.connect(self.show_doc_list)
|
|
self.mThread.formSignal.connect(self.waitPag_State)
|
|
self.mThread.start()
|
|
|
|
self.is_First_Show = False
|
|
|
|
|
|
def show_doc_list(self, tbody):
|
|
|
|
selector = etree.HTML(tbody) # 转换为lxml解析的对象
|
|
|
|
contents = selector.xpath('//div/div[@data-type="file"]/div[@data-type="file"]/a/text()') # 这里返回的是一个列表
|
|
messages = selector.xpath('//div/div[@data-type="file"]/div/div[@class="commit-details"]/a/text()') # 这里返回的是一个列表
|
|
|
|
print(contents)
|
|
print(messages)
|
|
|
|
self.TableWidget=QTableWidget()
|
|
self.TableWidget.setColumnCount(3) # 表格共有四列
|
|
self.TableWidget.verticalHeader().setVisible(False) # 隐藏垂直表头
|
|
self.TableWidget.horizontalHeader().setVisible(True) # 显示水平表头
|
|
|
|
font = QFont('微软雅黑', 10)
|
|
font.setBold(True) #设置字体加粗
|
|
self.TableWidget.horizontalHeader().setFont(font) #设置表头字体
|
|
|
|
# self.TableWidget.setFrameShape(QFrame.NoFrame) ##设置无表格的外框
|
|
self.TableWidget.horizontalHeader().setFixedHeight(25) ##设置表头高度
|
|
|
|
self.TableWidget.horizontalHeader().setSectionResizeMode(0,QHeaderView.Stretch)#设置第一列宽度自动调整，充满屏幕
|
|
# self.TableWidget.horizontalHeader().setStretchLastSection(True) ##设置最后一列拉伸至最大
|
|
|
|
self.TableWidget.setHorizontalHeaderLabels(['文档名称','文档介绍','操作']) #设置表头内容
|
|
self.TableWidget.horizontalHeader().setSectionsClickable(False)
|
|
self.TableWidget.horizontalHeader().setStyleSheet('QHeaderView::section{background:green}')#设置表头的背景色为绿色
|
|
|
|
self.TableWidget.setEditTriggers(QAbstractItemView.NoEditTriggers) # 不可编辑
|
|
self.TableWidget.setSelectionBehavior(QAbstractItemView.SelectRows) #只能选择整行
|
|
|
|
self.TableWidget.setColumnWidth(1,230)
|
|
self.TableWidget.setColumnWidth(2,80)
|
|
|
|
rows = self.TableWidget.rowCount()
|
|
rows_index = 0
|
|
|
|
if not messages :
|
|
for i in range(len(contents)):
|
|
messages.append("获取失败")
|
|
|
|
for content, message in zip(contents, messages):
|
|
content = content.strip() # 去掉字符左右的空格
|
|
message = message.strip()
|
|
self.TableWidget.setRowCount(rows_index + 1)
|
|
self.TableWidget.setItem(rows_index, 0, QTableWidgetItem(content))
|
|
self.TableWidget.setItem(rows_index, 1, QTableWidgetItem(message))
|
|
self.TableWidget.setCellWidget(rows_index, 2, self.buttonForRow(rows_index))
|
|
rows_index += 1
|
|
|
|
self.TableWidget.setGeometry(15, 10,300,300)
|
|
self.layout.setContentsMargins(0, 0, 0, 0)
|
|
self.layout.addWidget(self.TableWidget)
|
|
|
|
return 0
|
|
|
|
|
|
def buttonForRow(self,id):
|
|
widget=QWidget()
|
|
# 修改
|
|
downloadBtn = QPushButton('下载')
|
|
downloadBtn.setStyleSheet(''' text-align : center;
|
|
background-color : NavajoWhite;
|
|
height : 30px;
|
|
border-style: outset;
|
|
font : 13px ''')
|
|
|
|
downloadBtn.clicked.connect(lambda:self.download(id))
|
|
|
|
hLayout = QHBoxLayout()
|
|
hLayout.addWidget(downloadBtn)
|
|
hLayout.setContentsMargins(5,2,5,2)
|
|
widget.setLayout(hLayout)
|
|
return widget
|
|
|
|
def download(self, id):
|
|
|
|
fileName, ok = QFileDialog.getSaveFileName(self, "文件保存", "./combine/" + self.TableWidget.item(id, 0).text(), "All Files (*);;Bin Files (*.pdf)")
|
|
if ok:
|
|
print(fileName)
|
|
|
|
self.mThread = DocThread(action="down_doc", url="https://gitee.com/Ai-Thinker-Open/TB_Dev_Document/raw/master/" + self.TableWidget.item(id, 0).text(), fileName = fileName)
|
|
self.mThread.formSignal.connect(self.waitPag_State)
|
|
self.mThread.start()
|
|
|
|
self.waitPage.setText("<center><font color='red' size='6' line-height='50px';><red>正在下载文档......</font></center>")
|
|
self.waitPage.show()
|
|
self.waitPage.raise_()
|
|
|
|
def waitPag_State(self, state):
|
|
if state == CMD_CLOSE_FORM:
|
|
self.waitPage.hide()
|
|
elif state == CMD_DOWNLOAD_OK:
|
|
self.waitPage.hide()
|
|
QMessageBox.information(self,"温馨提示","下载成功！")
|