Run bottleclip generator as subprocess
This commit is contained in:
parent
3a721d9e29
commit
deced2d3a7
4 changed files with 46 additions and 55 deletions
60
main.py
60
main.py
|
@ -5,13 +5,14 @@ import os
|
|||
import sys
|
||||
from time import sleep
|
||||
|
||||
import aiohttp
|
||||
import aiomqtt
|
||||
import pytz
|
||||
from discord import Intents
|
||||
from discord import File, Intents
|
||||
from discord.ext import commands
|
||||
from discord_webhook import DiscordEmbed, DiscordWebhook
|
||||
|
||||
import bottleclip as clip
|
||||
|
||||
mqtt_host = os.getenv("MQTT_HOST")
|
||||
if not mqtt_host:
|
||||
print("MQTT_HOST unset")
|
||||
|
@ -27,11 +28,6 @@ if not webhook_url:
|
|||
print("DISCORD_WEBHOOK_URL unset")
|
||||
sys.exit(1)
|
||||
|
||||
bottleclip_git_token = os.getenv("BOTTLECLIP_GIT_TOKEN")
|
||||
if not bottleclip_git_token:
|
||||
print("BOTTLECLIP_GIT_TOKEN unset")
|
||||
sys.exit(1)
|
||||
|
||||
timezone = pytz.timezone("Europe/Amsterdam")
|
||||
|
||||
# Discord bot stuff
|
||||
|
@ -116,28 +112,7 @@ async def np(ctx):
|
|||
# !bottleclip
|
||||
@HobbyBot.command(description="Generate a bottle-clip STL file suitable for printing")
|
||||
async def bottleclip(ctx, icon: str = "", ears: bool = False):
|
||||
git_api = "https://git.bitlair.nl/api/v1"
|
||||
owner = "bitlair"
|
||||
repo = "bottle-clip"
|
||||
flow = "stl.yaml"
|
||||
auth = {
|
||||
"Authorization": f"Bearer {bottleclip_git_token}",
|
||||
}
|
||||
|
||||
icons = {
|
||||
"thing-logos/glider.dxf",
|
||||
"thing-logos/Club_mate_logo.dxf",
|
||||
"thing-logos/chaosknoten.dxf",
|
||||
"thing-logos/camprocket.dxf",
|
||||
"pixel-cats/laying-left.svg",
|
||||
"pixel-cats/laying-right.svg",
|
||||
"pixel-cats/sitting-left.svg",
|
||||
"pixel-cats/sitting-right.svg",
|
||||
"pixel-cats/spooked-left.svg",
|
||||
"pixel-cats/spooked-right.svg",
|
||||
"pixel-cats/standing-left.svg",
|
||||
"pixel-cats/standing-right.svg",
|
||||
}
|
||||
icons = clip.list_icons()
|
||||
if icon not in icons:
|
||||
await ctx.reply(
|
||||
f"usage: `!bottleclip <icon> [<ears y|n>]`\n* `icon` must be one of {', '.join(icons)}"
|
||||
|
@ -145,30 +120,11 @@ async def bottleclip(ctx, icon: str = "", ears: bool = False):
|
|||
return
|
||||
|
||||
async with ctx.typing():
|
||||
async with aiohttp.ClientSession() as sess:
|
||||
dispatch_body = {
|
||||
"inputs": {
|
||||
"label": ctx.author.nick,
|
||||
"icon": icon,
|
||||
"ears": "true" if ears else "false",
|
||||
},
|
||||
"ref": "main",
|
||||
"return_run_info": True,
|
||||
}
|
||||
async with sess.post(
|
||||
f"{git_api}/repos/{owner}/{repo}/actions/workflows/{flow}/dispatches",
|
||||
headers=auth,
|
||||
json=dispatch_body,
|
||||
) as response:
|
||||
if response.status != 201:
|
||||
await ctx.reply("Meh, stuk")
|
||||
return
|
||||
r = await response.json()
|
||||
run_number = r["run_number"]
|
||||
label = ctx.author.nick or ctx.author.global_name
|
||||
stl_file = clip.create_stl(label, icon, ears)
|
||||
|
||||
await ctx.reply(
|
||||
f"https://git.bitlair.nl/bitlair/bottle-clip/actions/runs/{run_number}"
|
||||
)
|
||||
attach = File(stl_file.name, filename=f"{label}.stl")
|
||||
await ctx.reply("Ok! Hier is je flessenclip", file=attach)
|
||||
|
||||
|
||||
def webhook_message(msg):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue