feature(tests): add unit tests
This commit is contained in:
0
app/__init__.py
Normal file
0
app/__init__.py
Normal file
20
app/close.py
Normal file
20
app/close.py
Normal file
@ -0,0 +1,20 @@
|
||||
from webex_bot.models.command import Command
|
||||
|
||||
|
||||
class ExitCommand(Command):
|
||||
def __init__(self) -> None:
|
||||
super().__init__(
|
||||
command_keyword="exit",
|
||||
help_message="Exit",
|
||||
delete_previous_message=True,
|
||||
)
|
||||
self.sender: str = ""
|
||||
|
||||
def pre_execute(self, message, attachment_actions, activity) -> None:
|
||||
return
|
||||
|
||||
def execute(self, message, attachment_actions, activity) -> None:
|
||||
return
|
||||
|
||||
def post_execute(self, message, attachment_actions, activity) -> None:
|
||||
return
|
||||
60
app/img.py
Normal file
60
app/img.py
Normal file
@ -0,0 +1,60 @@
|
||||
import requests
|
||||
|
||||
|
||||
CHAR_REPLACEMENTS: list = [
|
||||
["_", "__"],
|
||||
["-", "--"],
|
||||
[" ", "_"],
|
||||
["?", "~q"],
|
||||
["&", "~a"],
|
||||
["%", "~p"],
|
||||
["#", "~h"],
|
||||
["/", "~s"],
|
||||
["\\", "~b"],
|
||||
["<", "~l"],
|
||||
[">", "~g"],
|
||||
['"', "''"],
|
||||
]
|
||||
|
||||
|
||||
def get_templates() -> list[dict]:
|
||||
url: str = "https://api.memegen.link/templates"
|
||||
req: requests.Response = requests.get(url=url, timeout=5)
|
||||
req.raise_for_status()
|
||||
data: dict = req.json()
|
||||
templates: list = []
|
||||
for tmpl in data:
|
||||
if tmpl["lines"] != 2:
|
||||
continue
|
||||
if tmpl["id"] == "oprah":
|
||||
tmpl["name"] = "Oprah You Get A..."
|
||||
tmpl_ext: str = "gif" if "animated" in tmpl["styles"] else "jpg"
|
||||
tmpl_data: dict = {
|
||||
"id": tmpl["id"],
|
||||
"name": tmpl["name"],
|
||||
"ext": tmpl_ext,
|
||||
"choiceval": tmpl["id"] + "." + tmpl_ext,
|
||||
}
|
||||
templates.append(tmpl_data)
|
||||
templates = sorted(templates, key=lambda d: d["name"])
|
||||
return templates
|
||||
|
||||
|
||||
def format_meme_string(input_string: str) -> str:
|
||||
# https://memegen.link/#special-characters
|
||||
out_string: str = input_string
|
||||
for char_replacement in CHAR_REPLACEMENTS:
|
||||
out_string: str = out_string.replace(char_replacement[0], char_replacement[1])
|
||||
return out_string
|
||||
|
||||
|
||||
def generate_api_url(template: str, top_str: str, btm_str: str) -> str:
|
||||
tmpl_name: str
|
||||
tmpl_ext: str
|
||||
tmpl_name, tmpl_ext = template.split(".")
|
||||
|
||||
top_str = format_meme_string(top_str)
|
||||
btm_str = format_meme_string(btm_str)
|
||||
|
||||
url: str = f"https://api.memegen.link/images/{tmpl_name}/{top_str}/{btm_str}.{tmpl_ext}"
|
||||
return url
|
||||
33
app/main.py
Normal file
33
app/main.py
Normal file
@ -0,0 +1,33 @@
|
||||
#!/usr/local/bin/python3
|
||||
|
||||
import os
|
||||
|
||||
from webex_bot.webex_bot import WebexBot
|
||||
|
||||
from app import close, meme
|
||||
|
||||
WBX_API_KEY: str = os.environ["WEBEX_API_KEY"]
|
||||
|
||||
|
||||
def create_bot() -> WebexBot:
|
||||
"""Create a Bot object."""
|
||||
bot = WebexBot(
|
||||
teams_bot_token=WBX_API_KEY,
|
||||
approved_domains=["cisco.com"],
|
||||
bot_name="MemeBot",
|
||||
include_demo_commands=False,
|
||||
)
|
||||
return bot
|
||||
|
||||
|
||||
def main() -> None:
|
||||
bot: WebexBot = create_bot()
|
||||
bot.add_command(meme.MakeMemeCommand())
|
||||
bot.add_command(close.ExitCommand())
|
||||
bot.commands.remove(bot.help_command)
|
||||
bot.help_command = meme.MakeMemeCommand()
|
||||
bot.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
141
app/meme.py
Normal file
141
app/meme.py
Normal file
@ -0,0 +1,141 @@
|
||||
from webex_bot.models.command import Command
|
||||
from webex_bot.models.response import Response, response_from_adaptive_card
|
||||
from webexteamssdk.models.cards import (
|
||||
AdaptiveCard,
|
||||
Column,
|
||||
ColumnSet,
|
||||
FontSize,
|
||||
FontWeight,
|
||||
Text,
|
||||
TextBlock,
|
||||
Choice,
|
||||
Choices,
|
||||
)
|
||||
from webexteamssdk.models.cards.actions import Submit, OpenUrl
|
||||
|
||||
from app import img
|
||||
|
||||
|
||||
TEMPLATES = img.get_templates()
|
||||
|
||||
|
||||
class MakeMemeCommand(Command):
|
||||
def __init__(self) -> None:
|
||||
super().__init__(
|
||||
command_keyword="/meme",
|
||||
help_message="Make a Meme",
|
||||
chained_commands=[MakeMemeCallback()],
|
||||
delete_previous_message=True,
|
||||
)
|
||||
|
||||
def pre_execute(self, message, attachment_actions, activity) -> None:
|
||||
return
|
||||
|
||||
def execute(self, message, attachment_actions, activity) -> Response:
|
||||
card_body: list = [
|
||||
ColumnSet(
|
||||
columns=[
|
||||
Column(
|
||||
width=1,
|
||||
items=[
|
||||
TextBlock(
|
||||
"Make a Meme",
|
||||
weight=FontWeight.BOLDER,
|
||||
size=FontSize.MEDIUM,
|
||||
),
|
||||
TextBlock(
|
||||
"This bot uses memegen.link to generate memes. Click 'View Templates' to view available templates.",
|
||||
weight=FontWeight.LIGHTER,
|
||||
size=FontSize.SMALL,
|
||||
wrap=True,
|
||||
),
|
||||
TextBlock(
|
||||
"Both fields are required. If you do not want to specify a value, please type a space.",
|
||||
weight=FontWeight.LIGHTER,
|
||||
size=FontSize.SMALL,
|
||||
wrap=True,
|
||||
),
|
||||
],
|
||||
),
|
||||
]
|
||||
),
|
||||
ColumnSet(
|
||||
columns=[
|
||||
Column(
|
||||
width=1,
|
||||
items=[
|
||||
Choices(
|
||||
id="meme_type",
|
||||
isMultiSelect=False,
|
||||
choices=[
|
||||
Choice(title=x["name"], value=x["choiceval"]) for x in TEMPLATES
|
||||
],
|
||||
),
|
||||
Text(id="text_top", placeholder="Top Text", maxLength=100),
|
||||
Text(
|
||||
id="text_bottom",
|
||||
placeholder="Bottom Text",
|
||||
maxLength=100,
|
||||
),
|
||||
],
|
||||
),
|
||||
]
|
||||
),
|
||||
]
|
||||
|
||||
card: AdaptiveCard = AdaptiveCard(
|
||||
body=card_body,
|
||||
actions=[
|
||||
Submit(
|
||||
title="Go!",
|
||||
data={"callback_keyword": "make_meme_callback_rbamzfyx"},
|
||||
),
|
||||
OpenUrl(url="https://memegen.link/#templates", title="View Templates"),
|
||||
Submit(title="Cancel", data={"command_keyword": "exit"}),
|
||||
],
|
||||
)
|
||||
return response_from_adaptive_card(card)
|
||||
|
||||
|
||||
class MakeMemeCallback(Command):
|
||||
def __init__(self) -> None:
|
||||
super().__init__(
|
||||
card_callback_keyword="make_meme_callback_rbamzfyx",
|
||||
delete_previous_message=True,
|
||||
)
|
||||
self.error: bool = False
|
||||
self.text_top: str = ""
|
||||
self.text_bottom: str = ""
|
||||
self.meme: str = ""
|
||||
self.meme_filename: str = ""
|
||||
|
||||
def pre_execute(self, message, attachment_actions, activity) -> str:
|
||||
self.meme: str = attachment_actions.inputs.get("meme_type")
|
||||
self.text_top: str = attachment_actions.inputs.get("text_top")
|
||||
self.text_bottom: str = attachment_actions.inputs.get("text_bottom")
|
||||
|
||||
if not self.text_top and not self.text_bottom:
|
||||
self.error = True
|
||||
return "Please provide at least one positional text argument."
|
||||
|
||||
if ".gif" in self.meme:
|
||||
return "Generating your meme. GIF-based memes take a little longer. Please wait..."
|
||||
|
||||
return "Generating your meme..."
|
||||
|
||||
def execute(self, message, attachment_actions, activity) -> str:
|
||||
if not self.error:
|
||||
self.meme_filename: str = img.generate_api_url(
|
||||
self.meme, self.text_top, self.text_bottom
|
||||
)
|
||||
msg: Response = Response(
|
||||
attributes={
|
||||
"roomId": activity["target"]["globalId"],
|
||||
"parentId": "",
|
||||
"files": [self.meme_filename],
|
||||
}
|
||||
)
|
||||
return msg
|
||||
|
||||
def post_execute(self, message, attachment_actions, activity) -> None:
|
||||
return
|
||||
Reference in New Issue
Block a user