1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 03:24:20 +00:00

💥 Implement some more commands and stuff

This commit is contained in:
Steffo 2021-01-08 19:40:57 +01:00
parent e144ca5a59
commit be5c24327f
6 changed files with 167 additions and 12 deletions

14
poetry.lock generated
View file

@ -74,7 +74,7 @@ pyreadline = "*"
[[package]]
name = "idna"
version = "3.0"
version = "3.1"
description = "Internationalized Domain Names in Applications (IDNA)"
category = "main"
optional = false
@ -112,7 +112,7 @@ marker = "sys_platform == \"win32\""
[[package]]
name = "royalnet"
version = "6.0.0a33"
version = "6.0.0a36"
description = "A multipurpose bot and web framework"
category = "main"
optional = false
@ -174,7 +174,7 @@ multidict = ">=4.0"
[metadata]
lock-version = "1.0"
python-versions = "^3.9"
content-hash = "227a77d4af0a59cb4289ac3e29d342788bafe838fbabc5e566383e518f86c5cf"
content-hash = "ef6d764f995f9579e32eef84b9eb802a5b97677893a5870c99b81d4af0d945ea"
[metadata.files]
aiohttp = [
@ -237,8 +237,8 @@ humanfriendly = [
{file = "humanfriendly-9.1.tar.gz", hash = "sha256:066562956639ab21ff2676d1fda0b5987e985c534fc76700a19bd54bcb81121d"},
]
idna = [
{file = "idna-3.0-py2.py3-none-any.whl", hash = "sha256:320229aadbdfc597bc28876748cc0c9d04d476e0fe6caacaaddea146365d9f63"},
{file = "idna-3.0.tar.gz", hash = "sha256:c9a26e10e5558412384fac891eefb41957831d31be55f1e2c98ed97a70abb969"},
{file = "idna-3.1-py3-none-any.whl", hash = "sha256:5205d03e7bcbb919cc9c19885f9920d622ca52448306f2377daede5cf3faac16"},
{file = "idna-3.1.tar.gz", hash = "sha256:c5b02147e01ea9920e6b0a3f1f7bb833612d507592c837a6c49552768f4054e1"},
]
multidict = [
{file = "multidict-5.1.0-cp36-cp36m-macosx_10_14_x86_64.whl", hash = "sha256:b7993704f1a4b204e71debe6095150d43b2ee6150fa4f44d6d966ec356a8d61f"},
@ -309,8 +309,8 @@ pyreadline = [
{file = "pyreadline-2.1.zip", hash = "sha256:4530592fc2e85b25b1a9f79664433da09237c1a270e4d78ea5aa3a2c7229e2d1"},
]
royalnet = [
{file = "royalnet-6.0.0a33-py3-none-any.whl", hash = "sha256:b8455febca99b8e5d2a9260ea4d7694975c65182fbdd4a6159411b6830fa0796"},
{file = "royalnet-6.0.0a33.tar.gz", hash = "sha256:7b9c18c918dae94db437640bf044fb20ff6180b879f45aa067b68abdb598c5d6"},
{file = "royalnet-6.0.0a36-py3-none-any.whl", hash = "sha256:8cfd930b774db4041a4aa17d3e44528ae414e05eae15c3036ceebf856b54ea39"},
{file = "royalnet-6.0.0a36.tar.gz", hash = "sha256:fb2f89e360af1668500b690071aba37d8fbae6ebf813f821171030b1721ce0c3"},
]
sqlalchemy = [
{file = "SQLAlchemy-1.3.22-cp27-cp27m-macosx_10_14_x86_64.whl", hash = "sha256:61628715931f4962e0cdb2a7c87ff39eea320d2aa96bd471a3c293d146f90394"},

View file

@ -7,7 +7,7 @@ license = "AGPL-3.0-or-later"
[tool.poetry.dependencies]
python = "^3.9"
royalnet = "6.0.0a33"
royalnet = "^6.0.0a36"
coloredlogs = "^15.0"
aiohttp = "^3.7.3"

View file

@ -2,11 +2,11 @@ import royalnet.engineer as engi
@engi.PartialCommand.new(syntax="")
def ahnonlosoio(_sentry: engi.Sentry, _msg: engi.Message, **__):
def ahnonlosoio(*, _sentry: engi.Sentry, _msg: engi.Message, **__):
"""
Ah, non lo so io!
"""
await _msg.send_reply(r"¯\_(ツ)_/¯ Ah, non lo so io!")
await _msg.send_reply(text=r"¯\_(ツ)_/¯ Ah, non lo so io!")
__all__ = ("ahnonlosoio",)

96
royalpack/commands/cat.py Normal file
View file

@ -0,0 +1,96 @@
# Special imports
from __future__ import annotations
import royalnet.royaltyping as t
# External imports
import aiohttp
import royalnet.engineer as engi
import logging
import io
# Internal imports
# from . import something
# Special global objects
log = logging.getLogger(__name__)
# Code
@engi.PartialCommand.new(syntax="")
def cat(*, _sentry: engi.Sentry, _msg: engi.Message, **__):
"""
Send a cat in the chat! 🐈
"""
log.debug("Creating a new HTTP session")
async with aiohttp.ClientSession() as session:
log.info("Making a GET request to The Cat API Image Search")
async with session.get("https://api.thecatapi.com/v1/images/search") as response:
log.debug("Ensuring the request was successful")
if response.status >= 400:
log.error(f"The Cat API returned an HTTP error: {response.status}")
await _msg.send_reply(
text="⚠️ Couldn't request a cat from https://thecatapi.com :("
)
return
log.debug("Reading the JSON received from The Cat API")
try:
result = await response.json()
except aiohttp.ContentTypeError:
log.error(f"Couldn't decode received JSON from The Cat API")
await _msg.send_reply(
text="⚠️ Couldn't understand what the cat from https://thecatapi.com was saying :("
)
return
# Example result:
# [
# {
# "breeds": [],
# "id": "MjAzMjY3MQ",
# "url": "https://cdn2.thecatapi.com/images/MjAzMjY3MQ.jpg",
# "width": 557,
# "height": 724
# }
# ]
log.debug("Ensuring at least one image was received")
if len(result) == 0:
log.error("Didn't receive any image from The Cat API")
await _msg.send_reply(
text="⚠️ I couldn't find any cats at https://thecatapi.com :("
)
return
# Select the first image received
selected_cat = result[0]
log.debug(f"Selected {selected_cat!r}")
log.debug("Ensuring an image url is available")
if "url" not in selected_cat:
log.error("Image received from The Cat API did not have any URL")
await _msg.send_reply(
text="⚠️ I found a cat at https://thecatapi.com, but I couldn't find its image :("
)
return
# Download the cat image
log.info("Making a GET request to retrieve a The Cat API image")
async with session.get(selected_cat["url"]) as response:
log.debug("Reading image bytes into memory")
img = io.BytesIO()
while img_data := response.content.read(8192):
img.write(img_data)
log.debug("Sending image in the chat")
await _msg.send_reply(files=[img])
# Objects exported by this module
__all__ = (
"cat",
)

View file

@ -2,11 +2,11 @@ import royalnet.engineer as engi
@engi.PartialCommand.new(syntax="")
def ping(_sentry: engi.Sentry, _msg: engi.Message, **__):
def ping(*, _sentry: engi.Sentry, _msg: engi.Message, **__):
"""
A way to check if the bot is working: it will always reply to this command with "🏓 Pong!".
"""
await _msg.send_reply("🏓 Pong!")
await _msg.send_reply(text="🏓 Pong!")
__all__ = ("ping",)

View file

@ -0,0 +1,59 @@
# Special imports
from __future__ import annotations
import royalnet.royaltyping as t
# External imports
import royalnet.engineer as engi
import logging
import re
# Internal imports
# from . import something
# Special global objects
log = logging.getLogger(__name__)
# Code
@engi.PartialCommand.new(syntax=r"(?P<first>[A-Za-z]+)[\s+&]+(?P<second>[A-Za-z]+)")
def ship(*, _sentry: engi.Sentry, _msg: engi.Message, first: str, second: str, **__):
"""
Ship two names together! 💞
"""
log.info(f"Shipping: {first!r} + {second!r}")
# Convert the names to lowercase
first = first.lower()
second = second.lower()
log.debug(f"Lowercased: {first!r} + {second!r}")
# Get all letters until the first vowel, included
first_match = re.search(r"^[A-Za-z][^aeiouAEIOU]*[aeiouAEIOU]?", first)
# Get all letters from the second to last vowel, excluded
second_match = re.search(r"[^aeiouAEIOU]*[aeiouAEIOU]?[A-Za-z]$", second)
log.debug(f"Matches: {first_match!r} + {second_match!r}")
# Get the matched characters if the matches were successful, or cut the names in half if they weren't
first = first_match.group(0) if first_match else first[:(len(first) // 2)]
second = second_match.group(0) if second_match else second[(len(second) // 2):]
log.debug(f"Cropped: {first!r} + {second!r}")
# Combine the two parts
combined = f"{first}{second}"
log.info(f"Combined: {combined!r}")
# Send the message to the chat
log.debug(f"Sending ship to the chat...")
await _msg.send_reply(
text=f"💞 {first.capitalize()} + {second.capitalize()} = {combined.capitalize()}"
)
# Objects exported by this module
__all__ = (
"ship",
)