def _chunked_download(
filename: str, response: HTTPResponse, size: int, progress_signal: pyqtSignal
) -> None:
"""
Downloads the zip file and emits progress.
:param response: the HTTP response to download the file from
:param size: the size of the download, in bytes; used for progress
:param progress_signal: the PyQt signal with which to emit download progress
"""
if not size:
size = 10 ** 6 * 44
with open(filename, "wb") as f:
bytes_downloaded = 0
block_size = 1024 * 10
while True:
buffer = response.read(block_size)
if not buffer:
break
f.write(buffer)
bytes_downloaded += block_size
progress_signal.emit(bytes_downloaded / size * 100)
def task(signal: QtCore.pyqtSignal):
"""
多线程任务函数
:param signal: Qt信号
:return: None
"""
while True:
time.sleep(1)
signal.emit(json.dumps(time.time()))
def _mock_download(progress_signal: pyqtSignal) -> None:
"""
Pretends to download a zip file. Used for testing when the zip file is
already stored in the current working directory.
"""
counter = 0
while counter <= 100:
progress_signal.emit(counter)
counter += 1
time.sleep(2 / 100)
def buffered_send(stream: Iterable[object], signal: pyqtSignal, size=2000):
buff = []
i = 0
for data in stream:
buff.append(data)
i += 1
if i == size:
signal.emit(buff)
i = 0
buff = []
if i > 0:
signal.emit(buff)
def download_images(signal: pyqtSignal, core: LegendaryCore):
if not os.path.isdir(IMAGE_DIR):
os.makedirs(IMAGE_DIR)
logger.info("Create Image dir")
# Download Images
for i, game in enumerate(
sorted(core.get_game_list(), key=lambda x: x.app_title)):
try:
download_image(game)
except json.decoder.JSONDecodeError:
shutil.rmtree(f"{IMAGE_DIR}/{game.app_name}")
download_image(game)
signal.emit(i)
def chunked_download(
filename: str,
response: HTTPResponse,
size: int = None,
progress_signal: pyqtSignal = None,
) -> None:
"""
Downloads the zip file and emits progress.
:param response: the HTTP response to download the file from
:param size: the size of the download, in bytes; used for progress
:param progress_signal: the PyQt signal with which to emit download progress
"""
if progress_signal is None:
logging.warning(f"Progress signal is None.")
if not size:
length = response.headers.get("content-length")
if length:
length = int(length)
logging.info(f"Download size is {length / 1024 ** 2:.1f}MB.")
size = length
else:
size = 10**6 * 44
logging.warning(f"Guessing download size.")
with open(filename, "wb") as f:
bytes_downloaded = 0
block_size = 1024 * 10
while True:
buffer = response.read(block_size)
if not buffer:
break
f.write(buffer)
bytes_downloaded += block_size
if progress_signal:
progress_signal.emit(bytes_downloaded / size * 100)
def _notifying_setter(
self,
attr_name: str,
value: Any,
on_set: pyqtSignal,
on_unset: pyqtSignal,
send_value: bool = False
):
""" Utility for a setter that emits Qt signals when the attribute in
question changes state.
Parameters
----------
attr_name :
identifies attribute to be set
value :
set attribute to this value
on_set :
emitted when the new value is not None
on_unset :
emitted when the new value is None
send_value :
if True, the new value will be included in the emitted signal
"""
setattr(self, attr_name, value)
if value is None:
on_unset.emit()
else:
if send_value:
on_set.emit(value)
else:
on_set.emit()
def _obtain_transaction_data(self, format: TxSerialisationFormat,
completion_signal: Optional[pyqtSignal],
done_signal: pyqtSignal,
completion_text: str) -> None:
tx_data = self.tx.to_format(format)
if not isinstance(tx_data, dict):
# This is not ideal, but it covers both bases.
if completion_signal is not None:
completion_signal.emit(format, tx_data)
done_signal.emit(format, tx_data)
return
steps = self._account.estimate_extend_serialised_transaction_steps(
format, self.tx, tx_data)
# The done callbacks should happen in the context of the GUI thread.
def on_done(weakwindow: 'ElectrumWindow',
future: concurrent.futures.Future) -> None:
nonlocal format, done_signal
try:
data = future.result()
except concurrent.futures.CancelledError:
done_signal.emit(format, None)
except Exception as exc:
weakwindow.on_exception(exc)
else:
done_signal.emit(format, data)
title = _("Obtaining transaction data")
func = partial(self._obtain_transaction_data_worker, format, tx_data,
completion_signal, completion_text)
weakwindow = weakref.proxy(self._main_window)
WaitingDialog(self,
title,
func,
on_done=partial(on_done, weakwindow),
progress_steps=steps,
allow_cancel=True,
close_delay=5)
def downloadFile(name, url, signals: pyqtSignal = None):
headers = {'Proxy-Connection': 'keep-alive'}
r = requests.get(url, stream=True, headers=headers)
length = float(r.headers['content-length'])
f = open(name, 'wb')
count = 0
count_tmp = 0
time1 = time.time()
for chunk in r.iter_content(chunk_size=512):
if chunk:
f.write(chunk)
count += len(chunk)
if time.time() - time1 > 1:
p = count / length * 100
speed = (count - count_tmp) / 1024 / 1024 / 2
count_tmp = count
Logger.println(f'已下载 {formatFloat(p)}% (下载速度:{formatFloat(speed)}M/s)')
time1 = time.time()
if signals:
signals.emit(f'已下载 {formatFloat(p)}% (下载速度:{formatFloat(speed)}M/s)')
signals.emit(f'已下载 {formatFloat(100)}%')
f.close()
def download_images(signal: pyqtSignal, core: LegendaryCore):
if not os.path.isdir(IMAGE_DIR):
os.makedirs(IMAGE_DIR)
logger.info("Create Image dir")
# Download Images
for i, game in enumerate(
sorted(core.get_game_list(), key=lambda x: x.app_title)):
if not os.path.isdir(f"{IMAGE_DIR}/" + game.app_name):
os.mkdir(f"{IMAGE_DIR}/" + game.app_name)
if not os.path.isfile(f"{IMAGE_DIR}/{game.app_name}/image.json"):
json_data = {"DieselGameBoxTall": None, "DieselGameBoxLogo": None}
else:
json_data = json.load(
open(f"{IMAGE_DIR}/{game.app_name}/image.json", "r"))
for image in game.metadata["keyImages"]:
if image["type"] == "DieselGameBoxTall" or image[
"type"] == "DieselGameBoxLogo":
if json_data[
image["type"]] != image["md5"] or not os.path.isfile(
f"{IMAGE_DIR}/{game.app_name}/{image['type']}.png"
):
# Download
json_data[image["type"]] = image["md5"]
# os.remove(f"{IMAGE_DIR}/{game.app_name}/{image['type']}.png")
json.dump(
json_data,
open(f"{IMAGE_DIR}/{game.app_name}/image.json", "w"))
logger.info(f"Download Image for Game: {game.app_title}")
url = image["url"]
with open(
f"{IMAGE_DIR}/{game.app_name}/{image['type']}.png",
"wb") as f:
f.write(requests.get(url).content)
f.close()
if not os.path.isfile(f'{IMAGE_DIR}/' + game.app_name +
'/UninstalledArt.png'):
if os.path.isfile(f'{IMAGE_DIR}/' + game.app_name +
'/DieselGameBoxTall.png'):
# finalArt = Image.open(f'{IMAGE_DIR}/' + game.app_name + '/DieselGameBoxTall.png')
# finalArt.save(f'{IMAGE_DIR}/{game.app_name}/FinalArt.png')
# And same with the grayscale one
bg = Image.open(
f"{IMAGE_DIR}/{game.app_name}/DieselGameBoxTall.png")
uninstalledArt = bg.convert('L')
uninstalledArt.save(
f'{IMAGE_DIR}/{game.app_name}/UninstalledArt.png')
elif os.path.isfile(
f"{IMAGE_DIR}/{game.app_name}/DieselGameBoxLogo.png"):
bg: Image.Image = Image.open(
f"{IMAGE_DIR}/{game.app_name}/DieselGameBoxLogo.png")
bg = bg.resize((int(bg.size[1] * 3 / 4), bg.size[1]))
logo = Image.open(
f'{IMAGE_DIR}/{game.app_name}/DieselGameBoxLogo.png'
).convert('RGBA')
wpercent = ((bg.size[0] * (3 / 4)) / float(logo.size[0]))
hsize = int((float(logo.size[1]) * float(wpercent)))
logo = logo.resize((int(bg.size[0] * (3 / 4)), hsize),
Image.ANTIALIAS)
# Calculate where the image has to be placed
pasteX = int((bg.size[0] - logo.size[0]) / 2)
pasteY = int((bg.size[1] - logo.size[1]) / 2)
# And finally copy the background and paste in the image
# finalArt = bg.copy()
# finalArt.paste(logo, (pasteX, pasteY), logo)
# Write out the file
# finalArt.save(f'{IMAGE_DIR}/' + game.app_name + '/FinalArt.png')
logoCopy = logo.copy()
logoCopy.putalpha(int(256 * 3 / 4))
logo.paste(logoCopy, logo)
uninstalledArt = bg.copy()
uninstalledArt.paste(logo, (pasteX, pasteY), logo)
uninstalledArt = uninstalledArt.convert('L')
uninstalledArt.save(f'{IMAGE_DIR}/' + game.app_name +
'/UninstalledArt.png')
else:
logger.warning(
f"File {IMAGE_DIR}/{game.app_name}/DieselGameBoxTall.png dowsn't exist"
)
signal.emit(i)
def worker(widget: QtWidgets.QWidget, trigger: QtCore.pyqtSignal) -> None:
while True:
widget.refresh()
trigger.emit(widget.render)
if widget.delay > 0:
time.sleep(widget.delay)
def __init__(self, text : str, newWindow_signal:QtCore.pyqtSignal, menuIndex : int, parent=None):
super(Qtw.QPushButton, self).__init__(text, parent, cursor=QtCore.Qt.PointingHandCursor, toolTip='Change current editor', clicked=lambda: newWindow_signal.emit(menuIndex))
self.newWindow_signal = newWindow_signal
self.menuIndex = menuIndex
self.setSizePolicy(Qtw.QSizePolicy.Expanding, Qtw.QSizePolicy.Preferred)
self.setObjectName('Custom_Window_Button')
self.setStyleSheet(self.stylesheet)