{link}'
- newtext = f'''{a + 1}. {html.escape(i["title"])}
-Link: {link}
-Size: {i["nyaa_size"]}
-Seeders: {i["nyaa_seeders"]}
-Leechers: {i["nyaa_leechers"]}
-Category: {i["nyaa_category"]}\n\n'''
- futtext = text + newtext
- if (a and not a % 10) or len((await parser.parse(futtext))['message']) > 4096:
- results.append(text)
- futtext = newtext
- text = futtext
- a += 1
- results.append(text)
- ttl = time.time()
- used_search_info[query] = results, ttl
- try:
- return results[page], len(results), ttl
- except IndexError:
- return '', len(results), ttl
-
-message_info = dict()
-ignore = set()
-
-@app.on_message(filters.command(['nyaasi', f'nyaasi@{bot.username}']))
-async def nyaa_search(client, message):
- text = message.text.split(' ')
- text.pop(0)
- query = ' '.join(text)
- await init_search(client, message, query, False)
-
-@app.on_message(filters.command(['sukebei', f'sukebei@{bot.username}']))
-async def nyaa_search_sukebei(client, message):
- text = message.text.split(' ')
- text.pop(0)
- query = ' '.join(text)
- await init_search(client, message, query, True)
-
-async def init_search(client, message, query, sukebei):
- result, pages, ttl = await return_search(query, sukebei=sukebei)
- if not result:
- await message.reply_text('No results found')
- else:
- buttons = [InlineKeyboardButton(f'1/{pages}', 'nyaa_nop'), InlineKeyboardButton(f'Next', 'nyaa_next')]
- if pages == 1:
- buttons.pop()
- reply = await message.reply_text(result, reply_markup=InlineKeyboardMarkup([
- buttons
- ]))
- message_info[(reply.chat.id, reply.message_id)] = message.from_user.id, ttl, query, 1, pages, sukebei
-
-@app.on_callback_query(custom_filters.callback_data('nyaa_nop'))
-async def nyaa_nop(client, callback_query):
- await callback_query.answer(cache_time=3600)
-
-callback_lock = asyncio.Lock()
-@app.on_callback_query(custom_filters.callback_data(['nyaa_back', 'nyaa_next']))
-async def nyaa_callback(client, callback_query):
- message = callback_query.message
- message_identifier = (message.chat.id, message.message_id)
- data = callback_query.data
- async with callback_lock:
- if message_identifier in ignore:
- await callback_query.answer()
- return
- user_id, ttl, query, current_page, pages, sukebei = message_info.get(message_identifier, (None, 0, None, 0, 0, None))
- og_current_page = current_page
- if data == 'nyaa_back':
- current_page -= 1
- elif data == 'nyaa_next':
- current_page += 1
- if current_page < 1:
- current_page = 1
- elif current_page > pages:
- current_page = pages
- ttl_ended = (time.time() - ttl) > 3600
- if ttl_ended:
- text = getattr(message.text, 'html', 'Search expired')
- else:
- if callback_query.from_user.id != user_id:
- await callback_query.answer('...no', cache_time=3600)
- return
- text, pages, ttl = await return_search(query, current_page, sukebei)
- buttons = [InlineKeyboardButton(f'Prev', 'nyaa_back'), InlineKeyboardButton(f'{current_page}/{pages}', 'nyaa_nop'), InlineKeyboardButton(f'Next', 'nyaa_next')]
- if ttl_ended:
- buttons = [InlineKeyboardButton('Search Expired', 'nyaa_nop')]
- else:
- if current_page == 1:
- buttons.pop(0)
- if current_page == pages:
- buttons.pop()
- if ttl_ended or current_page != og_current_page:
- await callback_query.edit_message_text(text, reply_markup=InlineKeyboardMarkup([
- buttons
- ]))
- message_info[message_identifier] = user_id, ttl, query, current_page, pages, sukebei
- if ttl_ended:
- ignore.add(message_identifier)
- await callback_query.answer()
-
-# Using upstream API based on: https://github.com/Ryuk-me/Torrents-Api
-# Implemented by https://github.com/jusidama18
-
-class TorrentSearch:
- index = 0
- query = None
- message = None
- response = None
- response_range = None
-
- RESULT_LIMIT = 4
- RESULT_STR = None
-
- def __init__(self, command: str, source: str, result_str: str):
- self.command = command
- self.source = source.rstrip('/')
- self.RESULT_STR = result_str
-
- app.add_handler(MessageHandler(self.find, filters.command([command, f'{self.command}@{bot.username}'])))
- app.add_handler(CallbackQueryHandler(self.previous, filters.regex(f"{self.command}_previous")))
- app.add_handler(CallbackQueryHandler(self.delete, filters.regex(f"{self.command}_delete")))
- app.add_handler(CallbackQueryHandler(self.next, filters.regex(f"{self.command}_next")))
-
- @staticmethod
- def format_magnet(string: str):
- if not string:
- return ""
- return string.split('&tr', 1)[0]
-
- def get_formatted_string(self, values):
- string = self.RESULT_STR.format(**values)
- extra = ""
- if "Files" in values:
- tmp_str = "➲[{Quality} - {Type} ({Size})]({Torrent}): `{magnet}`"
- extra += "\n".join(
- tmp_str.format(**f, magnet=self.format_magnet(f['Magnet']))
- for f in values['Files']
- )
- else:
- magnet = values.get('magnet', values.get('Magnet')) # Avoid updating source dict
- if magnet:
- extra += f"➲Magnet: `{self.format_magnet(magnet)}`"
- if (extra):
- string += "\n" + extra
- return string
-
- async def update_message(self):
- prevBtn = InlineKeyboardButton(f"Prev", callback_data=f"{self.command}_previous")
- delBtn = InlineKeyboardButton(f"{emoji.CROSS_MARK}", callback_data=f"{self.command}_delete")
- nextBtn = InlineKeyboardButton(f"Next", callback_data=f"{self.command}_next")
-
- inline = []
- if (self.index != 0):
- inline.append(prevBtn)
- inline.append(delBtn)
- if (self.index != len(self.response_range) - 1):
- inline.append(nextBtn)
-
- res_lim = min(self.RESULT_LIMIT, len(self.response) - self.RESULT_LIMIT*self.index)
- result = f"**Page - {self.index+1}**\n\n"
- result += "\n\n=======================\n\n".join(
- self.get_formatted_string(self.response[self.response_range[self.index]+i])
- for i in range(res_lim)
- )
-
- await self.message.edit(
- result,
- reply_markup=InlineKeyboardMarkup([inline]),
- parse_mode="markdown",
- )
-
- async def find(self, client, message):
- if len(message.command) < 2:
- await message.reply_text(f"Usage: /{self.command} query")
- return
-
- query = urlencode(message.text.split(None, 1)[1])
- self.message = await message.reply_text("Searching")
- try:
- async with aiohttp.ClientSession() as session:
- async with session.get(f"{self.source}/{query}") as resp:
- if (resp.status != 200):
- raise Exception('unsuccessful request')
- result = await resp.json()
- if (result and isinstance(result[0], list)):
- result = list(itertools.chain(*result))
- self.response = result
- self.response_range = range(0, len(self.response), self.RESULT_LIMIT)
- except:
- await self.message.edit("No Results Found.")
- return
- await self.update_message()
-
- async def delete(self, client, message):
- index = 0
- query = None
- message = None
- response = None
- response_range = None
- await self.message.delete()
-
- async def previous(self, client, message):
- self.index -= 1
- await self.update_message()
-
- async def next(self, client, message):
- self.index += 1
- await self.update_message()
-
-RESULT_STR_1337 = (
- "➲Name: `{Name}`\n"
- "➲Size: {Size}\n"
- "➲Seeders: {Seeders} || ➲Leechers: {Leechers}"
-)
-RESULT_STR_PIRATEBAY = (
- "➲Name: `{Name}`\n"
- "➲Size: {Size}\n"
- "➲Seeders: {Seeders} || ➲Leechers: {Leechers}"
-)
-RESULT_STR_TGX = (
- "➲Name: `{Name}`\n"
- "➲Size: {Size}\n"
- "➲Seeders: {Seeders} || ➲Leechers: {Leechers}"
-)
-RESULT_STR_YTS = (
- "➲Name: `{Name}`\n"
- "➲Released on: {ReleasedDate}\n"
- "➲Genre: {Genre}\n"
- "➲Rating: {Rating}\n"
- "➲Likes: {Likes}\n"
- "➲Duration: {Runtime}\n"
- "➲Language: {Language}"
-)
-RESULT_STR_EZTV = (
- "➲Name: `{Name}`\n"
- "➲Size: {Size}\n"
- "➲Seeders: {Seeders}"
-)
-RESULT_STR_TORLOCK = (
- "➲Name: `{Name}`\n"
- "➲Size: {Size}\n"
- "➲Seeders: {Seeders} || ➲Leechers: {Leechers}"
-)
-RESULT_STR_RARBG = (
- "➲Name: `{Name}`\n"
- "➲Size: {Size}\n"
- "➲Seeders: {Seeders} || ➲Leechers: {Leechers}"
-)
-RESULT_STR_ALL = (
- "➲Name: `{Name}`\n"
- "➲Size: {Size}\n"
- "➲Seeders: {Seeders} || ➲Leechers: {Leechers}"
-)
-
-torrents_dict = {
- '1337x': {'source': "https://torrents--api.herokuapp.com/api/1337x/", 'result_str': RESULT_STR_1337},
- 'piratebay': {'source': "https://torrents--api.herokuapp.com/api/piratebay/", 'result_str': RESULT_STR_PIRATEBAY},
- 'tgx': {'source': "https://torrents--api.herokuapp.com/api/tgx/", 'result_str': RESULT_STR_TGX},
- 'yts': {'source': "https://torrents--api.herokuapp.com/api/yts/", 'result_str': RESULT_STR_YTS},
- 'eztv': {'source': "https://torrents--api.herokuapp.com/api/eztv/", 'result_str': RESULT_STR_EZTV},
- 'torlock': {'source': "https://torrents--api.herokuapp.com/api/torlock/", 'result_str': RESULT_STR_TORLOCK},
- 'rarbg': {'source': "https://torrents--api.herokuapp.com/api/rarbg/", 'result_str': RESULT_STR_RARBG},
- 'ts': {'source': "https://torrents--api.herokuapp.com/api/all/", 'result_str': RESULT_STR_ALL}
-}
-
-torrent_handlers = []
-for command, value in torrents_dict.items():
- torrent_handlers.append(TorrentSearch(command, value['source'], value['result_str']))
-
-def searchhelp(update, context):
- help_string = '''
-Torrent Search
-• /nyaasi [search query]
-• /sukebei [search query]
-• /1337x [search query]
-• /piratebay [search query]
-• /tgx [search query]
-• /yts [search query]
-• /eztv [search query]
-• /torlock [search query]
-• /rarbg [search query]
-• /ts [search query]
-'''
- sendMessage(help_string, context.bot, update)
-
-
-SEARCHHELP_HANDLER = CommandHandler(BotCommands.TsHelpCommand, searchhelp, filters=(CustomFilters.authorized_chat | CustomFilters.authorized_user) & CustomFilters.mirror_owner_filter, run_async=True)
-dispatcher.add_handler(SEARCHHELP_HANDLER)
diff --git a/config_sample.env b/config_sample.env
index f090249..dcbf74e 100644
--- a/config_sample.env
+++ b/config_sample.env
@@ -31,7 +31,7 @@ SHORTENER_API = ""
# qBittorrent
IS_VPS = "" # Don't set this to True even if VPS, unless facing error with web server
SERVER_PORT = "80" # Only if IS_VPS is True
-BASE_URL_OF_BOT = "" # Web Link, Required for Heroku to avoid sleep or use worker if u don't want to use web(selection)
+BASE_URL_OF_BOT = "" # Web Link, Required for Heroku to avoid sleep or use worker if you don't want to use web (selection)
# If you want to use Credentials externally from Index Links, fill these vars with the direct links
# These are optional, if you don't know, simply leave them, don't fill anything in them.
ACCOUNTS_ZIP_URL = ""
diff --git a/docker-compose.yml b/docker-compose.yml
index 863925e..5cc7986 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -7,5 +7,3 @@ services:
restart: on-failure
ports:
- "80:80"
-
-
diff --git a/requirements.txt b/requirements.txt
index b30ae44..3df8cd0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,6 +18,7 @@ psutil
psycopg2-binary
pybase64
pyrogram
+pyshorteners
python-dotenv
python-magic
python-telegram-bot
@@ -30,4 +31,3 @@ TgCrypto
torrentool
urllib3
youtube_dl
-pyshorteners