polyjitter/searchbot-discord
1
0
Fork 0
mirror of https://github.com/polyjitter/searchbot-discord.git synced 2024-08-14 22:46:55 +00:00
Code Issues Projects Releases Wiki Activity

Error Handling, Online Util

Browse source
This commit is contained in:
Adriene Hutchins 2020-03-02 18:37:34 -05:00
parent 8adc005f50
commit 830adece17
4 changed files with 97 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

25
extensions/developer.py
View file

@ -19,6 +19,7 @@ import io
import inspect
import textwrap
import subprocess
from extensions.utils import online
class Developer(commands.Cog):
@ -29,6 +30,7 @@ class Developer(commands.Cog):
# Main Stuff
self.bot = bot
self.request = bot.request
self.online = bot.online
self.emoji = "\U0001F3D7"
# Repl/Eval Stuff
@ -54,19 +56,6 @@ class Developer(commands.Cog):
'^',
type(err).__name__)
async def _post_to_hastebin(self, string):
"""Posts a string to hastebin."""
url = "https://hastebin.com/documents"
data = string.encode('utf-8')
async with self.request.post(url=url, data=data) as haste_response:
haste_key = (await haste_response.json())['key']
haste_url = f"http://hastebin.com/{haste_key}"
# data = {'sprunge': ''}
# data['sprunge'] = string
# haste_url = await self.aioclient.post(url='http://sprunge.us',
# data=data)
return haste_url
@commands.group(name='shell',
aliases=['ipython', 'repl', 'longexec'],
invoke_without_command=True)
@ -161,7 +150,7 @@ class Developer(commands.Cog):
item,
history[item])
haste_url = await self._post_to_hastebin(history_string)
haste_url = await self.online.hastebin(history_string)
return_msg = "[`Leaving shell session. "\
"History hosted on hastebin.`]({})".format(
haste_url)
@ -201,7 +190,7 @@ class Developer(commands.Cog):
if len(cleaned) > 800:
cleaned = "<Too big to be printed>"
if len(return_msg) > 800:
haste_url = await self._post_to_hastebin(return_msg)
haste_url = await self.online.hastebin(return_msg)
return_msg = "[`SyntaxError too big to be printed. "\
"Hosted on hastebin.`]({})".format(
haste_url)
@ -253,7 +242,7 @@ class Developer(commands.Cog):
try:
if fmt is not None:
if len(fmt) >= 800:
haste_url = await self._post_to_hastebin(fmt)
haste_url = await self.online.hastebin(fmt)
self.repl_embeds[shell].add_field(
name="`>>> {}`".format(cleaned),
value="[`Content too big to be printed. "
@ -467,7 +456,7 @@ class Developer(commands.Cog):
if (len(result[0]) >= 1024):
stdout = result[0].decode('utf-8')
string = string + f'[[STDOUT]]\n{stdout}'
link = await self._post_to_hastebin(string)
link = await self.online.hastebin(string)
await message.edit(
content=f":x: Content too long. {link}",
embed=None)
@ -476,7 +465,7 @@ class Developer(commands.Cog):
if (len(result[1]) >= 1024):
stdout = result[0].decode('utf-8')
string = string + f'[[STDERR]]\n{stdout}'
link = await self._post_to_hastebin(string)
link = await self.online.hastebin(string)
await message.edit(
content=f":x: Content too long. {link}",
embed=None)

2
extensions/search.py
View file

@ -67,7 +67,7 @@ class Search(commands.Cog):
)
# Create the URL to make an API call to
call = f'{instance}/search?q={query}&format=json&language=en-US'
call = f'{instance}search?q={query}&format=json&language=en-US'
# If a type is provided, add that type to the call URL
if category:

28
extensions/utils/online.py Normal file
View file

@ -0,0 +1,28 @@
# -*- coding: utf-8 -*-
# tacibot online util
# Provides utils for various online tools.
'''Online File'''
class Online():
def __init__(self, bot):
self.bot = bot
self.request = bot.request
async def hastebin(self, string):
"""Posts a string to hastebin."""
url = "https://hastebin.com/documents"
data = string.encode('utf-8')
async with self.request.post(url=url, data=data) as haste_response:
haste_key = (await haste_response.json())['key']
haste_url = f"http://hastebin.com/{haste_key}"
# data = {'sprunge': ''}
# data['sprunge'] = string
# haste_url = await self.aioclient.post(url='http://sprunge.us',
# data=data)
return haste_url
def setup(bot):
bot.online = Online(bot)