aboutsummaryrefslogtreecommitdiffstats
path: root/bot/exts/evergreen/wolfram.py
diff options
context:
space:
mode:
authorGravatar Xithrius <[email protected]>2021-09-03 00:31:12 -0700
committerGravatar GitHub <[email protected]>2021-09-03 00:31:12 -0700
commitea47bc617e558929bcee39e6008a57d6dd814aa1 (patch)
treec40e2f23e55119fb33f83271d227103cb9be7c6f /bot/exts/evergreen/wolfram.py
parentImproved consistency for codeblocks to end with a newline (diff)
parentMerge pull request #802 from python-discord/decorator-factory/typehints-fix (diff)
Merge branch 'main' into android-codeblock-fix
Diffstat (limited to 'bot/exts/evergreen/wolfram.py')
-rw-r--r--bot/exts/evergreen/wolfram.py14
1 files changed, 7 insertions, 7 deletions
diff --git a/bot/exts/evergreen/wolfram.py b/bot/exts/evergreen/wolfram.py
index 26674d37..9a26e545 100644
--- a/bot/exts/evergreen/wolfram.py
+++ b/bot/exts/evergreen/wolfram.py
@@ -1,6 +1,6 @@
import logging
from io import BytesIO
-from typing import Callable, List, Optional, Tuple
+from typing import Callable, Optional
from urllib.parse import urlencode
import arrow
@@ -54,7 +54,7 @@ async def send_embed(
await ctx.send(embed=embed, file=f)
-def custom_cooldown(*ignore: List[int]) -> Callable:
+def custom_cooldown(*ignore: int) -> Callable:
"""
Implement per-user and per-guild cooldowns for requests to the Wolfram API.
@@ -105,7 +105,7 @@ def custom_cooldown(*ignore: List[int]) -> Callable:
return check(predicate)
-async def get_pod_pages(ctx: Context, bot: Bot, query: str) -> Optional[List[Tuple]]:
+async def get_pod_pages(ctx: Context, bot: Bot, query: str) -> Optional[list[tuple[str, str]]]:
"""Get the Wolfram API pod pages for the provided query."""
async with ctx.typing():
params = {
@@ -133,22 +133,22 @@ async def get_pod_pages(ctx: Context, bot: Bot, query: str) -> Optional[List[Tup
f"processing a wolfram request: {log_full_url}, Response: {json}"
)
await send_embed(ctx, message)
- return
+ return None
message = "Something went wrong internally with your request, please notify staff!"
log.warning(f"Something went wrong getting a response from wolfram: {log_full_url}, Response: {json}")
await send_embed(ctx, message)
- return
+ return None
if not result["success"]:
message = f"I couldn't find anything for {query}."
await send_embed(ctx, message)
- return
+ return None
if not result["numpods"]:
message = "Could not find any results."
await send_embed(ctx, message)
- return
+ return None
pods = result["pods"]
pages = []