mir/microbot
2
0
Fork 0
mirror of https://github.com/Nick80835/microbot synced 2025-08-30 05:58:23 +00:00
Code Issues Releases Wiki Activity

unify booru modules

Browse Source
This commit is contained in:
Nick80835 2020-10-03 18:05:08 -04:00
parent 061fef48e2
commit e5018dc24a
6 changed files with 152 additions and 550 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

152
ubot/modules/booru.py Normal file
View File

@ -0,0 +1,152 @@
# SPDX-License-Identifier: GPL-2.0-or-later
from asyncbooru import Danbooru, Gelbooru, Konachan, Sankaku, Yandere
from telethon import Button
from ubot import ldr
help_str = "Fetches images from Danbooru, Gelbooru, Konachan, Sankaku Complex and Yandere, takes tags as arguments."
dan_api = Danbooru(ldr.aioclient)
gel_api = Gelbooru(ldr.aioclient)
kon_api = Konachan(ldr.aioclient)
san_api = Sankaku(ldr.aioclient)
yan_api = Yandere(ldr.aioclient)
dan_butt = {}
gel_butt = {}
kon_butt = {}
san_butt = {}
yan_butt = {}
commands_sfw = {
"dans": dan_api,
"gels": gel_api,
"kons": kon_api,
"sans": san_api,
"yans": yan_api
}
commands_nsfw = {
("dan", "danx", "danq"): dan_api,
("gel", "gelx", "gelq"): gel_api,
("kon", "konx", "konq"): kon_api,
("san", "sanx", "sanq"): san_api,
("yan", "yanx", "yanq"): yan_api
}
commands_butt = {
"danb": [dan_api, dan_butt, "dan"],
"gelb": [gel_api, gel_butt, "gel"],
"konb": [kon_api, kon_butt, "kon"],
"sanb": [san_api, san_butt, "san"],
"yanb": [yan_api, yan_butt, "yan"]
}
@ldr.add_dict(commands_sfw, pattern_extra="(f|)", help=help_str, userlocking=True)
@ldr.add_dict(commands_nsfw, pattern_extra="(f|)", help=help_str, userlocking=True, nsfw=True)
async def booru(event):
safety_arg = event.command[-1]
as_file = bool(event.other_args[0])
posts = await event.extra.get_random_posts(event.args, 3, safety_arg)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
for image in images:
try:
await event.reply(f"[sauce]({image[1]})", file=image[0], force_document=as_file)
return
except:
pass
await event.reply(f"Failed to fetch media for query: {event.args}")
@ldr.add_inline_photo("dan(s|x|q|)", default="dan", extra=dan_api)
@ldr.add_inline_photo("gel(s|x|q|)", default="gel", extra=gel_api)
@ldr.add_inline_photo("kon(s|x|q|)", default="kon", extra=kon_api)
@ldr.add_inline_photo("san(s|x|q|)", default="san", extra=san_api)
@ldr.add_inline_photo("yan(s|x|q|)", default="yan", extra=yan_api)
async def booru_inline(event):
posts = await event.extra.get_random_posts(event.args, 3, event.other_args[0])
return [[post.file_url, f"[sauce]({post.sauce})"] for post in posts if post.file_url] if posts else None
@ldr.add_dict(commands_butt, help=help_str, userlocking=True, nsfw=True)
async def booru_buttons(event):
posts = await event.extra[0].get_random_posts(event.args, 30)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
event.extra[1][f"{event.chat.id}_{event.id}"] = [0, images]
await event.reply(
f"[sauce]({images[0][1]})",
file=images[0][0],
buttons=[Button.inline('➡️', f'{event.extra[2]}*{event.chat.id}_{event.id}*r')]
)
@ldr.add_callback_query("dan", extra=dan_butt)
@ldr.add_callback_query("gel", extra=gel_butt)
@ldr.add_callback_query("kon", extra=kon_butt)
@ldr.add_callback_query("san", extra=san_butt)
@ldr.add_callback_query("yan", extra=yan_butt)
async def booru_buttons_callback(event):
args_split = event.args.split("*")
dict_id = args_split[0]
direction = args_split[1]
if dict_id in event.extra:
this_dict = event.extra[dict_id]
else:
return
if direction == "r":
this_dict[0] += 1
if this_dict[0] + 1 > len(this_dict[1]):
this_dict[0] = len(this_dict[1]) - 1
this_image = this_dict[1][this_dict[0]]
elif direction == "l":
this_dict[0] -= 1
if this_dict[0] < 0:
this_dict[0] = 0
this_image = this_dict[1][this_dict[0]]
buttons = []
if this_dict[0] > 0:
buttons += [Button.inline('⬅️', f'{event.command}*{dict_id}*l')]
if len(this_dict[1]) - 1 > this_dict[0]:
buttons += [Button.inline('➡️', f'{event.command}*{dict_id}*r')]
try:
await event.edit(
f"[sauce]({this_image[1]})",
file=this_image[0],
buttons=buttons
)
except:
pass

110
ubot/modules/danbooru.py
View File

@ -1,110 +0,0 @@
# SPDX-License-Identifier: GPL-2.0-or-later
from asyncbooru import Danbooru
from telethon import Button
from ubot import ldr
danbooru_api = Danbooru(ldr.aioclient)
dan_button_dict = {}
help_string = "Fetches images from Danbooru, takes tags as arguments."
@ldr.add_list(["dan", "danx", "danq"], pattern_extra="(f|)", nsfw=True, userlocking=True, help=help_string)
@ldr.add("dans", pattern_extra="(f|)", userlocking=True, help=help_string)
async def danbooru(event):
safety_arg = event.command[-1]
as_file = bool(event.other_args[0])
posts = await danbooru_api.get_random_posts(event.args, 3, safety_arg)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
for image in images:
try:
await event.reply(f"[sauce]({image[1]})", file=image[0], force_document=as_file)
return
except:
pass
await event.reply(f"Failed to fetch media for query: {event.args}")
@ldr.add_inline_photo("dan(s|x|q|)", default="dan")
async def danbooru_inline(event):
posts = await danbooru_api.get_random_posts(event.args, 3, event.other_args[0])
return [[post.file_url, f"[sauce]({post.sauce})"] for post in posts if post.file_url] if posts else None
@ldr.add("danb", nsfw=True, userlocking=True)
async def danbooru_buttons(event):
posts = await danbooru_api.get_random_posts(event.args, 30)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
dan_button_dict[f"{event.chat.id}_{event.id}"] = [0, images]
await event.reply(
f"[sauce]({images[0][1]})",
file=images[0][0],
buttons=[Button.inline('➡️', f'dan*{event.chat.id}_{event.id}*r')]
)
@ldr.add_callback_query("dan")
async def danbooru_buttons_callback(event):
args_split = event.args.split("*")
dict_id = args_split[0]
direction = args_split[1]
if dict_id in dan_button_dict:
this_dict = dan_button_dict[dict_id]
else:
return
if direction == "r":
this_dict[0] += 1
if this_dict[0] + 1 > len(this_dict[1]):
this_dict[0] = len(this_dict[1]) - 1
this_image = this_dict[1][this_dict[0]]
elif direction == "l":
this_dict[0] -= 1
if this_dict[0] < 0:
this_dict[0] = 0
this_image = this_dict[1][this_dict[0]]
buttons = []
if this_dict[0] > 0:
buttons += [Button.inline('⬅️', f'dan*{dict_id}*l')]
if len(this_dict[1]) - 1 > this_dict[0]:
buttons += [Button.inline('➡️', f'dan*{dict_id}*r')]
try:
await event.edit(
f"[sauce]({this_image[1]})",
file=this_image[0],
buttons=buttons
)
except:
pass

110
ubot/modules/gelbooru.py
View File

@ -1,110 +0,0 @@
# SPDX-License-Identifier: GPL-2.0-or-later
from asyncbooru import Gelbooru
from telethon import Button
from ubot import ldr
gelbooru_api = Gelbooru(ldr.aioclient)
gel_button_dict = {}
help_string = "Fetches images from Gelbooru, takes tags as arguments."
@ldr.add_list(["gel", "gelx", "gelq"], pattern_extra="(f|)", nsfw=True, userlocking=True, help=help_string)
@ldr.add("gels", pattern_extra="(f|)", userlocking=True, help=help_string)
async def gelbooru(event):
safety_arg = event.command[-1]
as_file = bool(event.other_args[0])
posts = await gelbooru_api.get_random_posts(event.args, 3, safety_arg)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
for image in images:
try:
await event.reply(f"[sauce]({image[1]})", file=image[0], force_document=as_file)
return
except:
pass
await event.reply(f"Failed to fetch media for query: {event.args}")
@ldr.add_inline_photo("gel(s|x|q|)", default="gel")
async def gelbooru_inline(event):
posts = await gelbooru_api.get_random_posts(event.args, 3, event.other_args[0])
return [[post.file_url, f"[sauce]({post.sauce})"] for post in posts if post.file_url] if posts else None
@ldr.add("gelb", nsfw=True, userlocking=True)
async def gelbooru_buttons(event):
posts = await gelbooru_api.get_random_posts(event.args, 30)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
gel_button_dict[f"{event.chat.id}_{event.id}"] = [0, images]
await event.reply(
f"[sauce]({images[0][1]})",
file=images[0][0],
buttons=[Button.inline('➡️', f'gel*{event.chat.id}_{event.id}*r')]
)
@ldr.add_callback_query("gel")
async def gelbooru_buttons_callback(event):
args_split = event.args.split("*")
dict_id = args_split[0]
direction = args_split[1]
if dict_id in gel_button_dict:
this_dict = gel_button_dict[dict_id]
else:
return
if direction == "r":
this_dict[0] += 1
if this_dict[0] + 1 > len(this_dict[1]):
this_dict[0] = len(this_dict[1]) - 1
this_image = this_dict[1][this_dict[0]]
elif direction == "l":
this_dict[0] -= 1
if this_dict[0] < 0:
this_dict[0] = 0
this_image = this_dict[1][this_dict[0]]
buttons = []
if this_dict[0] > 0:
buttons += [Button.inline('⬅️', f'gel*{dict_id}*l')]
if len(this_dict[1]) - 1 > this_dict[0]:
buttons += [Button.inline('➡️', f'gel*{dict_id}*r')]
try:
await event.edit(
f"[sauce]({this_image[1]})",
file=this_image[0],
buttons=buttons
)
except:
pass

110
ubot/modules/konachan.py
View File

@ -1,110 +0,0 @@
# SPDX-License-Identifier: GPL-2.0-or-later
from asyncbooru import Konachan
from telethon import Button
from ubot import ldr
konachan_api = Konachan(ldr.aioclient)
kon_button_dict = {}
help_string = "Fetches images from Konachan, takes tags as arguments."
@ldr.add_list(["kon", "konx", "konq"], pattern_extra="(f|)", nsfw=True, userlocking=True, help=help_string)
@ldr.add("kons", pattern_extra="(f|)", userlocking=True, help=help_string)
async def konachan(event):
safety_arg = event.command[-1]
as_file = bool(event.other_args[0])
posts = await konachan_api.get_random_posts(event.args, 3, safety_arg)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
for image in images:
try:
await event.reply(f"[sauce]({image[1]})", file=image[0], force_document=as_file)
return
except:
pass
await event.reply(f"Failed to fetch media for query: {event.args}")
@ldr.add_inline_photo("kon(s|x|q|)", default="kon")
async def konachan_inline(event):
posts = await konachan_api.get_random_posts(event.args, 3, event.other_args[0])
return [[post.file_url, f"[sauce]({post.sauce})"] for post in posts if post.file_url] if posts else None
@ldr.add("konb", nsfw=True, userlocking=True)
async def konachan_buttons(event):
posts = await konachan_api.get_random_posts(event.args, 30)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
kon_button_dict[f"{event.chat.id}_{event.id}"] = [0, images]
await event.reply(
f"[sauce]({images[0][1]})",
file=images[0][0],
buttons=[Button.inline('➡️', f'kon*{event.chat.id}_{event.id}*r')]
)
@ldr.add_callback_query("kon")
async def konachan_buttons_callback(event):
args_split = event.args.split("*")
dict_id = args_split[0]
direction = args_split[1]
if dict_id in kon_button_dict:
this_dict = kon_button_dict[dict_id]
else:
return
if direction == "r":
this_dict[0] += 1
if this_dict[0] + 1 > len(this_dict[1]):
this_dict[0] = len(this_dict[1]) - 1
this_image = this_dict[1][this_dict[0]]
elif direction == "l":
this_dict[0] -= 1
if this_dict[0] < 0:
this_dict[0] = 0
this_image = this_dict[1][this_dict[0]]
buttons = []
if this_dict[0] > 0:
buttons += [Button.inline('⬅️', f'kon*{dict_id}*l')]
if len(this_dict[1]) - 1 > this_dict[0]:
buttons += [Button.inline('➡️', f'kon*{dict_id}*r')]
try:
await event.edit(
f"[sauce]({this_image[1]})",
file=this_image[0],
buttons=buttons
)
except:
pass

110
ubot/modules/sankaku.py
View File

@ -1,110 +0,0 @@
# SPDX-License-Identifier: GPL-2.0-or-later
from asyncbooru import Sankaku
from telethon import Button
from ubot import ldr
sankaku_api = Sankaku(ldr.aioclient)
san_button_dict = {}
help_string = "Fetches images from Sankaku Complex, takes tags as arguments."
@ldr.add_list(["san", "sanx", "sanq"], pattern_extra="(f|)", nsfw=True, userlocking=True, help=help_string)
@ldr.add("sans", pattern_extra="(f|)", userlocking=True, help=help_string)
async def sankaku(event):
safety_arg = event.command[-1]
as_file = bool(event.other_args[0])
posts = await sankaku_api.get_random_posts(event.args, 3, safety_arg)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
for image in images:
try:
await event.reply(f"[sauce]({image[1]})", file=image[0], force_document=as_file)
return
except:
pass
await event.reply(f"Failed to fetch media for query: {event.args}")
@ldr.add_inline_photo("san(s|x|q|)", default="san")
async def sankaku_inline(event):
posts = await sankaku_api.get_random_posts(event.args, 3, event.other_args[0])
return [[post.file_url, f"[sauce]({post.sauce})"] for post in posts if post.file_url] if posts else None
@ldr.add("sanb", nsfw=True, userlocking=True)
async def sankaku_buttons(event):
posts = await sankaku_api.get_random_posts(event.args, 30)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
san_button_dict[f"{event.chat.id}_{event.id}"] = [0, images]
await event.reply(
f"[sauce]({images[0][1]})",
file=images[0][0],
buttons=[Button.inline('➡️', f'san*{event.chat.id}_{event.id}*r')]
)
@ldr.add_callback_query("san")
async def sankaku_buttons_callback(event):
args_split = event.args.split("*")
dict_id = args_split[0]
direction = args_split[1]
if dict_id in san_button_dict:
this_dict = san_button_dict[dict_id]
else:
return
if direction == "r":
this_dict[0] += 1
if this_dict[0] + 1 > len(this_dict[1]):
this_dict[0] = len(this_dict[1]) - 1
this_image = this_dict[1][this_dict[0]]
elif direction == "l":
this_dict[0] -= 1
if this_dict[0] < 0:
this_dict[0] = 0
this_image = this_dict[1][this_dict[0]]
buttons = []
if this_dict[0] > 0:
buttons += [Button.inline('⬅️', f'san*{dict_id}*l')]
if len(this_dict[1]) - 1 > this_dict[0]:
buttons += [Button.inline('➡️', f'san*{dict_id}*r')]
try:
await event.edit(
f"[sauce]({this_image[1]})",
file=this_image[0],
buttons=buttons
)
except:
pass

110
ubot/modules/yandere.py
View File

@ -1,110 +0,0 @@
# SPDX-License-Identifier: GPL-2.0-or-later
from asyncbooru import Yandere
from telethon import Button
from ubot import ldr
yandere_api = Yandere(ldr.aioclient)
yan_button_dict = {}
help_string = "Fetches images from Yandere, takes tags as arguments."
@ldr.add_list(["yan", "yanx", "yanq"], pattern_extra="(f|)", nsfw=True, userlocking=True, help=help_string)
@ldr.add("yans", pattern_extra="(f|)", userlocking=True, help=help_string)
async def yandere(event):
safety_arg = event.command[-1]
as_file = bool(event.other_args[0])
posts = await yandere_api.get_random_posts(event.args, 3, safety_arg)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
for image in images:
try:
await event.reply(f"[sauce]({image[1]})", file=image[0], force_document=as_file)
return
except:
pass
await event.reply(f"Failed to fetch media for query: {event.args}")
@ldr.add_inline_photo("yan(s|x|q|)", default="yan")
async def yandere_inline(event):
posts = await yandere_api.get_random_posts(event.args, 3, event.other_args[0])
return [[post.file_url, f"[sauce]({post.sauce})"] for post in posts if post.file_url] if posts else None
@ldr.add("yanb", nsfw=True, userlocking=True)
async def yandere_buttons(event):
posts = await yandere_api.get_random_posts(event.args, 30)
if not posts:
await event.reply(f"No results for query: {event.args}")
return
images = [[post.file_url, post.sauce] for post in posts if post.file_url]
if not images:
await event.reply(f"Failed to find URLs for query: {event.args}")
return
yan_button_dict[f"{event.chat.id}_{event.id}"] = [0, images]
await event.reply(
f"[sauce]({images[0][1]})",
file=images[0][0],
buttons=[Button.inline('➡️', f'yan*{event.chat.id}_{event.id}*r')]
)
@ldr.add_callback_query("yan")
async def yandere_buttons_callback(event):
args_split = event.args.split("*")
dict_id = args_split[0]
direction = args_split[1]
if dict_id in yan_button_dict:
this_dict = yan_button_dict[dict_id]
else:
return
if direction == "r":
this_dict[0] += 1
if this_dict[0] + 1 > len(this_dict[1]):
this_dict[0] = len(this_dict[1]) - 1
this_image = this_dict[1][this_dict[0]]
elif direction == "l":
this_dict[0] -= 1
if this_dict[0] < 0:
this_dict[0] = 0
this_image = this_dict[1][this_dict[0]]
buttons = []
if this_dict[0] > 0:
buttons += [Button.inline('⬅️', f'yan*{dict_id}*l')]
if len(this_dict[1]) - 1 > this_dict[0]:
buttons += [Button.inline('➡️', f'yan*{dict_id}*r')]
try:
await event.edit(
f"[sauce]({this_image[1]})",
file=this_image[0],
buttons=buttons
)
except:
pass