Module booru.client.yandere

Expand source code
import re
from typing import Union
from ..utils.fetch import request, request_wildcard, roll
from ..utils.constant import Api, better_object, parse_image, get_hostname
from random import shuffle

Booru = Api()


class Yandere(object):
    """Yandere Client

    Methods
    -------
    search : function
        Search method for yandere.

    search_image : function
        Search method for yandere, but only returns image.

    find_tags : function
        Get the proper tags from yandere.

    """

    @staticmethod
    def append_object(raw_object: dict):
        """Extends new object to the raw dict

        Parameters
        ----------
        raw_object : dict
            The raw object returned by yandere.

        Returns
        -------
        str
            The new value of the raw object
        """
        for i in range(len(raw_object)):
            if "id" in raw_object[i]:
                raw_object[i][
                    "post_url"
                ] = f"{get_hostname(Booru.yandere)}/post/show/{raw_object[i]['id']}"

        return raw_object

    def __init__(self):
        self.specs = {}

    async def search(
        self,
        query: str,
        block: str = "",
        limit: int = 100,
        page: int = 1,
        random: bool = True,
        gacha: bool = False,
    ) -> Union[list, str, None]:

        """Search method

        Parameters
        ----------
        query : str
            The query to search for.
        block : str
            The tags you want to block, separated by space.
        limit : int
            Expected number which is from pages
        page : int
            Expected number of page.
        random : bool
            Shuffle the whole dict, default is True.
        gacha : bool
            Get random single object, limit property will be ignored.

        Returns
        -------
        dict
            The json object (as string, you may need booru.resolve())
        """
        if limit > 1000:
            raise ValueError(Booru.error_handling_limit)

        elif block and re.findall(block, query):
            raise ValueError(Booru.error_handling_sameval)

        self.query = query
        self.specs["tags"] = self.query
        self.specs["limit"] = limit
        self.specs["page"] = page

        raw_data = await request(site=Booru.yandere, params_x=self.specs, block=block)
        self.appended = Yandere.append_object(raw_data)

        try:
            if gacha:
                return better_object(roll(self.appended))
            elif random:
                shuffle(self.appended)
                return better_object(self.appended)
            else:
                return better_object(Yandere.append_object(self.appended))
        except Exception as e:
            raise Exception(f"Failed to get data: {e}")

    async def search_image(
        self, query: str, block: str = "", limit: int = 100, page: int = 1
    ) -> Union[list, str, None]:

        """Parses image only

        Parameters
        ----------
        query : str
            The query to search for.
        block : str
            The tags you want to block, separated by space.
        limit : int
            Expected number which is from pages
        page : int
            Expected number of page.

        Returns
        -------
        dict
            The json object (as string, you may need booru.resolve())

        """
        if limit > 1000:
            raise ValueError(Booru.error_handling_limit)

        elif block and re.findall(block, query):
            raise ValueError(Booru.error_handling_sameval)

        self.query = query
        self.specs["tags"] = self.query
        self.specs["limit"] = limit
        self.specs["page"] = page

        raw_data = await request(site=Booru.yandere, params_x=self.specs, block=block)
        self.appended = Yandere.append_object(raw_data)

        try:
            return better_object(parse_image(self.appended))
        except Exception as e:
            raise Exception(f"Failed to get data: {e}")

    async def find_tags(self, query: str) -> Union[list, str, None]:
        """Find tags like a wildcard

        Parameters
        ----------
        query : str
            The tag to search for.

        Returns
        -------
        list
            The list of tags.
        """
        try:
            data = await request_wildcard(site=Booru.yandere_wildcard, query=query)
            return better_object(data)

        except Exception as e:
            raise Exception(f"Failed to get data: {e}")

Classes

class Yandere

Yandere Client

Methods

search : function Search method for yandere.

search_image : function Search method for yandere, but only returns image.

find_tags : function Get the proper tags from yandere.

Expand source code
class Yandere(object):
    """Yandere Client

    Methods
    -------
    search : function
        Search method for yandere.

    search_image : function
        Search method for yandere, but only returns image.

    find_tags : function
        Get the proper tags from yandere.

    """

    @staticmethod
    def append_object(raw_object: dict):
        """Extends new object to the raw dict

        Parameters
        ----------
        raw_object : dict
            The raw object returned by yandere.

        Returns
        -------
        str
            The new value of the raw object
        """
        for i in range(len(raw_object)):
            if "id" in raw_object[i]:
                raw_object[i][
                    "post_url"
                ] = f"{get_hostname(Booru.yandere)}/post/show/{raw_object[i]['id']}"

        return raw_object

    def __init__(self):
        self.specs = {}

    async def search(
        self,
        query: str,
        block: str = "",
        limit: int = 100,
        page: int = 1,
        random: bool = True,
        gacha: bool = False,
    ) -> Union[list, str, None]:

        """Search method

        Parameters
        ----------
        query : str
            The query to search for.
        block : str
            The tags you want to block, separated by space.
        limit : int
            Expected number which is from pages
        page : int
            Expected number of page.
        random : bool
            Shuffle the whole dict, default is True.
        gacha : bool
            Get random single object, limit property will be ignored.

        Returns
        -------
        dict
            The json object (as string, you may need booru.resolve())
        """
        if limit > 1000:
            raise ValueError(Booru.error_handling_limit)

        elif block and re.findall(block, query):
            raise ValueError(Booru.error_handling_sameval)

        self.query = query
        self.specs["tags"] = self.query
        self.specs["limit"] = limit
        self.specs["page"] = page

        raw_data = await request(site=Booru.yandere, params_x=self.specs, block=block)
        self.appended = Yandere.append_object(raw_data)

        try:
            if gacha:
                return better_object(roll(self.appended))
            elif random:
                shuffle(self.appended)
                return better_object(self.appended)
            else:
                return better_object(Yandere.append_object(self.appended))
        except Exception as e:
            raise Exception(f"Failed to get data: {e}")

    async def search_image(
        self, query: str, block: str = "", limit: int = 100, page: int = 1
    ) -> Union[list, str, None]:

        """Parses image only

        Parameters
        ----------
        query : str
            The query to search for.
        block : str
            The tags you want to block, separated by space.
        limit : int
            Expected number which is from pages
        page : int
            Expected number of page.

        Returns
        -------
        dict
            The json object (as string, you may need booru.resolve())

        """
        if limit > 1000:
            raise ValueError(Booru.error_handling_limit)

        elif block and re.findall(block, query):
            raise ValueError(Booru.error_handling_sameval)

        self.query = query
        self.specs["tags"] = self.query
        self.specs["limit"] = limit
        self.specs["page"] = page

        raw_data = await request(site=Booru.yandere, params_x=self.specs, block=block)
        self.appended = Yandere.append_object(raw_data)

        try:
            return better_object(parse_image(self.appended))
        except Exception as e:
            raise Exception(f"Failed to get data: {e}")

    async def find_tags(self, query: str) -> Union[list, str, None]:
        """Find tags like a wildcard

        Parameters
        ----------
        query : str
            The tag to search for.

        Returns
        -------
        list
            The list of tags.
        """
        try:
            data = await request_wildcard(site=Booru.yandere_wildcard, query=query)
            return better_object(data)

        except Exception as e:
            raise Exception(f"Failed to get data: {e}")

Static methods

def append_object(raw_object: dict)

Extends new object to the raw dict

Parameters

raw_object : dict
The raw object returned by yandere.

Returns

str
The new value of the raw object
Expand source code
@staticmethod
def append_object(raw_object: dict):
    """Extends new object to the raw dict

    Parameters
    ----------
    raw_object : dict
        The raw object returned by yandere.

    Returns
    -------
    str
        The new value of the raw object
    """
    for i in range(len(raw_object)):
        if "id" in raw_object[i]:
            raw_object[i][
                "post_url"
            ] = f"{get_hostname(Booru.yandere)}/post/show/{raw_object[i]['id']}"

    return raw_object

Methods

async def find_tags(self, query: str) ‑> Union[list, str, ForwardRef(None)]

Find tags like a wildcard

Parameters

query : str
The tag to search for.

Returns

list
The list of tags.
Expand source code
async def find_tags(self, query: str) -> Union[list, str, None]:
    """Find tags like a wildcard

    Parameters
    ----------
    query : str
        The tag to search for.

    Returns
    -------
    list
        The list of tags.
    """
    try:
        data = await request_wildcard(site=Booru.yandere_wildcard, query=query)
        return better_object(data)

    except Exception as e:
        raise Exception(f"Failed to get data: {e}")
async def search(self, query: str, block: str = '', limit: int = 100, page: int = 1, random: bool = True, gacha: bool = False) ‑> Union[list, str, ForwardRef(None)]

Search method

Parameters

query : str
The query to search for.
block : str
The tags you want to block, separated by space.
limit : int
Expected number which is from pages
page : int
Expected number of page.
random : bool
Shuffle the whole dict, default is True.
gacha : bool
Get random single object, limit property will be ignored.

Returns

dict
The json object (as string, you may need booru.resolve())
Expand source code
async def search(
    self,
    query: str,
    block: str = "",
    limit: int = 100,
    page: int = 1,
    random: bool = True,
    gacha: bool = False,
) -> Union[list, str, None]:

    """Search method

    Parameters
    ----------
    query : str
        The query to search for.
    block : str
        The tags you want to block, separated by space.
    limit : int
        Expected number which is from pages
    page : int
        Expected number of page.
    random : bool
        Shuffle the whole dict, default is True.
    gacha : bool
        Get random single object, limit property will be ignored.

    Returns
    -------
    dict
        The json object (as string, you may need booru.resolve())
    """
    if limit > 1000:
        raise ValueError(Booru.error_handling_limit)

    elif block and re.findall(block, query):
        raise ValueError(Booru.error_handling_sameval)

    self.query = query
    self.specs["tags"] = self.query
    self.specs["limit"] = limit
    self.specs["page"] = page

    raw_data = await request(site=Booru.yandere, params_x=self.specs, block=block)
    self.appended = Yandere.append_object(raw_data)

    try:
        if gacha:
            return better_object(roll(self.appended))
        elif random:
            shuffle(self.appended)
            return better_object(self.appended)
        else:
            return better_object(Yandere.append_object(self.appended))
    except Exception as e:
        raise Exception(f"Failed to get data: {e}")
async def search_image(self, query: str, block: str = '', limit: int = 100, page: int = 1) ‑> Union[list, str, ForwardRef(None)]

Parses image only

Parameters

query : str
The query to search for.
block : str
The tags you want to block, separated by space.
limit : int
Expected number which is from pages
page : int
Expected number of page.

Returns

dict
The json object (as string, you may need booru.resolve())
Expand source code
async def search_image(
    self, query: str, block: str = "", limit: int = 100, page: int = 1
) -> Union[list, str, None]:

    """Parses image only

    Parameters
    ----------
    query : str
        The query to search for.
    block : str
        The tags you want to block, separated by space.
    limit : int
        Expected number which is from pages
    page : int
        Expected number of page.

    Returns
    -------
    dict
        The json object (as string, you may need booru.resolve())

    """
    if limit > 1000:
        raise ValueError(Booru.error_handling_limit)

    elif block and re.findall(block, query):
        raise ValueError(Booru.error_handling_sameval)

    self.query = query
    self.specs["tags"] = self.query
    self.specs["limit"] = limit
    self.specs["page"] = page

    raw_data = await request(site=Booru.yandere, params_x=self.specs, block=block)
    self.appended = Yandere.append_object(raw_data)

    try:
        return better_object(parse_image(self.appended))
    except Exception as e:
        raise Exception(f"Failed to get data: {e}")