PyNyaaTa/pynyaata2/filters.py

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