mirror of
https://github.com/JoeanAmier/XHS-Downloader.git
synced 2025-12-25 20:36:47 +08:00
58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
from asyncio import sleep
|
|
from random import uniform
|
|
from typing import Callable
|
|
|
|
from rich import print
|
|
from rich.text import Text
|
|
|
|
from ..translation import _
|
|
from .static import INFO
|
|
|
|
|
|
def retry(function):
|
|
async def inner(self, *args, **kwargs):
|
|
if result := await function(self, *args, **kwargs):
|
|
return result
|
|
for __ in range(self.retry):
|
|
if result := await function(self, *args, **kwargs):
|
|
return result
|
|
return result
|
|
|
|
return inner
|
|
|
|
|
|
def retry_limited(function):
|
|
# TODO: 不支持 TUI
|
|
def inner(self, *args, **kwargs):
|
|
while True:
|
|
if function(self, *args, **kwargs):
|
|
return
|
|
if self.console.input(
|
|
_(
|
|
"如需重新尝试处理该对象，请关闭所有正在访问该对象的窗口或程序，然后直接按下回车键！\n"
|
|
"如需跳过处理该对象，请输入任意字符后按下回车键！"
|
|
),
|
|
):
|
|
return
|
|
|
|
return inner
|
|
|
|
|
|
def logging(log: Callable, text, style=INFO):
|
|
string = Text(text, style=style)
|
|
func = log()
|
|
if func is print:
|
|
func(string)
|
|
else:
|
|
func.write(
|
|
string,
|
|
scroll_end=True,
|
|
)
|
|
|
|
|
|
async def sleep_time(
|
|
min_time: int | float = 2.0,
|
|
max_time: int | float = 4.0,
|
|
):
|
|
await sleep(uniform(min_time, max_time))
|