mirror of
https://github.com/LorenEteval/Furious.git
synced 2026-09-28 18:08:11 +03:00
141 lines
4.4 KiB
Python
141 lines
4.4 KiB
Python
# Copyright (C) 2024–present Loren Eteval & contributors <loren.eteval@proton.me>
|
||
#
|
||
# This file is part of Furious.
|
||
#
|
||
# This program is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published by
|
||
# the Free Software Foundation, either version 3 of the License, or
|
||
# (at your option) any later version.
|
||
#
|
||
# This program is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
# GNU General Public License for more details.
|
||
#
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
from __future__ import annotations
|
||
|
||
from Furious.QtFramework.QtNetwork import *
|
||
from Furious.QtFramework.QtWidgets import *
|
||
from Furious.QtFramework.DynamicTranslate import gettext as _
|
||
from Furious.Utility import *
|
||
from Furious.Library import *
|
||
|
||
from PySide6 import QtCore
|
||
from PySide6.QtGui import *
|
||
from PySide6.QtNetwork import *
|
||
|
||
from typing import AnyStr, Union, Callable
|
||
|
||
import logging
|
||
import functools
|
||
|
||
__all__ = ['WebGETManager']
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class WebGETManager(AppQNetworkAccessManager):
|
||
def __init__(self, parent=None, actionMessage='web GET', **kwargs):
|
||
super().__init__(parent)
|
||
|
||
self.actionMessage = actionMessage
|
||
|
||
self.mustCallOnce = kwargs.pop('mustCallOnce', True)
|
||
self.mustCalled = False
|
||
|
||
def successCallback(self, networkReply: QNetworkReply, **kwargs):
|
||
pass
|
||
|
||
def hasDataCallback(self, networkReply: QNetworkReply, **kwargs):
|
||
pass
|
||
|
||
def failureCallback(self, networkReply: QNetworkReply, **kwargs):
|
||
pass
|
||
|
||
def mustCall(self, **kwargs):
|
||
pass
|
||
|
||
def must(self, **kwargs):
|
||
def call():
|
||
try:
|
||
self.mustCall(**kwargs)
|
||
except Exception as ex:
|
||
# Any non-exit exceptions
|
||
|
||
logger.error(f'error calling must(): {ex}')
|
||
finally:
|
||
self.mustCalled = True
|
||
|
||
if not self.mustCallOnce:
|
||
call()
|
||
else:
|
||
if not self.mustCalled:
|
||
call()
|
||
|
||
def handleReadyReadByNetworkReply(self, networkReply: QNetworkReply, **kwargs):
|
||
self.hasDataCallback(networkReply, **kwargs)
|
||
|
||
def handleFinishedByNetworkReply(self, networkReply: QNetworkReply, **kwargs):
|
||
try:
|
||
if not isinstance(networkReply, QNetworkReply):
|
||
# Some PySide6 version does not have networkReply as
|
||
# QNetworkReply instance, so assertion is not used here
|
||
logger.error(
|
||
f'QNetworkReply error in PySide6 {PYSIDE6_VERSION} version'
|
||
)
|
||
|
||
return
|
||
|
||
logActionMessage = kwargs.pop('logActionMessage', True)
|
||
|
||
if networkReply.error() != QNetworkReply.NetworkError.NoError:
|
||
if logActionMessage:
|
||
logger.error(
|
||
f'{self.actionMessage} failed. {networkReply.errorString()}'
|
||
)
|
||
|
||
self.failureCallback(networkReply, **kwargs)
|
||
else:
|
||
if logActionMessage:
|
||
logger.info(f'{self.actionMessage} success')
|
||
|
||
self.successCallback(networkReply, **kwargs)
|
||
finally:
|
||
self.must(**kwargs)
|
||
|
||
def configureHttpProxy(self, httpProxy: Union[str, None]) -> bool:
|
||
useProxy = super().configureHttpProxy(httpProxy)
|
||
|
||
if useProxy:
|
||
logger.info(f'{self.actionMessage} uses proxy server {httpProxy}')
|
||
else:
|
||
logger.info(f'{self.actionMessage} uses no proxy')
|
||
|
||
return useProxy
|
||
|
||
def webGET(self, request: Union[QNetworkRequest, str], **kwargs) -> QNetworkReply:
|
||
if isinstance(request, QNetworkRequest):
|
||
networkReply = self.get(request)
|
||
else:
|
||
networkReply = self.get(QNetworkRequest(QtCore.QUrl(request)))
|
||
|
||
networkReply.readyRead.connect(
|
||
functools.partial(
|
||
self.handleReadyReadByNetworkReply,
|
||
networkReply,
|
||
**kwargs,
|
||
)
|
||
)
|
||
networkReply.finished.connect(
|
||
functools.partial(
|
||
self.handleFinishedByNetworkReply,
|
||
networkReply,
|
||
**kwargs,
|
||
)
|
||
)
|
||
|
||
return networkReply
|