Module booru.client.lolibooru
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 Lolibooru(object):
"""lolibooru Client
Methods
-------
search : function
Search and gets images from lolibooru.
search_image : function
Search and gets images from lolibooru, but only returns image.
find_tags : function
Get the proper tags from lolibooru.
"""
@staticmethod
def append_object(raw_object: dict):
"""Extends new object to the raw dict
Parameters
----------
raw_object : dict
The raw object returned by lolibooru.
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.lolibooru)}/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.lolibooru, params_x=self.specs, block=block)
self.appended = Lolibooru.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(Lolibooru.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.lolibooru, params_x=self.specs, block=block)
self.appended = Lolibooru.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.lolibooru_wildcard, query=query)
return better_object(data)
except Exception as e:
raise Exception(f"Failed to get data: {e}")Classes
class Lolibooru-
lolibooru Client
Methods
search : function Search and gets images from lolibooru.
search_image : function Search and gets images from lolibooru, but only returns image.
find_tags : function Get the proper tags from lolibooru.
Expand source code
class Lolibooru(object): """lolibooru Client Methods ------- search : function Search and gets images from lolibooru. search_image : function Search and gets images from lolibooru, but only returns image. find_tags : function Get the proper tags from lolibooru. """ @staticmethod def append_object(raw_object: dict): """Extends new object to the raw dict Parameters ---------- raw_object : dict The raw object returned by lolibooru. 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.lolibooru)}/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.lolibooru, params_x=self.specs, block=block) self.appended = Lolibooru.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(Lolibooru.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.lolibooru, params_x=self.specs, block=block) self.appended = Lolibooru.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.lolibooru_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 lolibooru.
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 lolibooru. 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.lolibooru)}/post/show/{raw_object[i]['id']}" return raw_object
Methods
-
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.lolibooru_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.lolibooru, params_x=self.specs, block=block) self.appended = Lolibooru.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(Lolibooru.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.lolibooru, params_x=self.specs, block=block) self.appended = Lolibooru.append_object(raw_data) try: return better_object(parse_image(self.appended)) except Exception as e: raise Exception(f"Failed to get data: {e}")