mirror of
https://github.com/JoeanAmier/XHS-Downloader.git
synced 2025-12-26 04:48:05 +08:00
26 lines
553 B
Python
26 lines
553 B
Python
from rich.text import Text
|
|
|
|
from .static import INFO
|
|
|
|
__all__ = ["retry", "logging"]
|
|
|
|
|
|
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 logging(log, text, style=INFO):
|
|
string = Text(text, style=style)
|
|
if log:
|
|
log.write(string)
|
|
else:
|
|
print(string)
|