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/pynyaata/filters.py
2023-01-04 22:32:27 +01:00

64 lines
1.3 KiB
Python

from functools import wraps
from os import getenv
from typing import List
from pynyaata.types import Color, RemoteFile
BLACKLIST_WORDS = getenv("BLACKLIST_WORDS", "").split(",")
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]:
return list(
filter(
lambda remote: any(word in remote.name.lower() for word in BLACKLIST_WORDS),
remotes,
)
)
def danger(remotes: List[RemoteFile]) -> List[RemoteFile]:
return list(
filter(
lambda remote: remote.color != Color.DANGER,
remotes,
)
)
def filter_data(f):
@wraps(f)
async def wrapper(*args, **kwargs):
ret = await f(*args, **kwargs)
ret = duplicate(ret)
ret = inactive(ret)
ret = blacklist(ret)
ret = danger(ret)
return ret
return wrapper