This repository has been archived on 2023-10-01. You can view files and clone it, but cannot push or open issues or pull requests.
PyNyaaTa/pynyaata2/filters.py
2023-06-04 17:53:29 +02:00

82 lines
1.8 KiB
Python

from functools import wraps
from os import getenv
from typing import List
from pynyaata2.types import Color, RemoteFile
def duplicate(remotes: List[RemoteFile]) -> List[RemoteFile]:
processed_ids: List[int] = []
dedup_remotes: List[RemoteFile] = []
for remote in remotes:
if remote.id not in processed_ids:
dedup_remotes.append(remote)
processed_ids.append(remote.id)
return dedup_remotes
def inactive(remotes: List[RemoteFile]) -> List[RemoteFile]:
return list(
filter(
lambda remote: remote.seeds != 0 or remote.downloads != 0,
remotes,
)
)
def blacklist(remotes: List[RemoteFile]) -> List[RemoteFile]:
BLACKLIST_WORDS = getenv("BLACKLIST_WORDS", "")
if not BLACKLIST_WORDS:
return remotes
else:
return list(
filter(
lambda remote: not any(
word and word in remote.name.lower()
for word in BLACKLIST_WORDS.split(",")
),
remotes,
)
)
def danger(remotes: List[RemoteFile]) -> List[RemoteFile]:
return list(
filter(
lambda remote: remote.color != Color.DANGER,
remotes,
)
)
def trusted(remotes: List[RemoteFile]) -> List[RemoteFile]:
TRUSTED_WORDS = getenv("TRUSTED_WORDS", "")
for remote in remotes:
if any(
word and word in remote.name.lower() for word in TRUSTED_WORDS.split(",")
):
remote.color = Color.PRIMARY
return remotes
def filter_data(f):
@wraps(f)
def wrapper(*args, **kwargs):
ret = f(*args, **kwargs)
ret = duplicate(ret)
ret = inactive(ret)
ret = blacklist(ret)
ret = danger(ret)
ret = trusted(ret)
return ret
return wrapper