76 lines
1.8 KiB
Python
76 lines
1.8 KiB
Python
import sys
|
|
import typing
|
|
import json
|
|
import pathlib
|
|
import datetime
|
|
|
|
import httpx
|
|
|
|
COLORS = {
|
|
"failure": 0xCB2431,
|
|
"success": 0x28A745,
|
|
"info": 0x0000FF,
|
|
}
|
|
|
|
|
|
class EmbedField(typing.TypedDict):
|
|
name: str
|
|
value: str
|
|
inline: bool
|
|
|
|
|
|
class Embed(typing.TypedDict):
|
|
color: str
|
|
title: str
|
|
fields: list[EmbedField]
|
|
|
|
|
|
class MessageState(typing.TypedDict):
|
|
"""
|
|
Partial mapping of Discord message information.
|
|
|
|
See: https://discord.com/developers/docs/resources/channel#message-object
|
|
"""
|
|
|
|
id: int
|
|
timestamp: str
|
|
edited_timestamp: str
|
|
embeds: list[Embed]
|
|
|
|
|
|
class Context(typing.TypedDict):
|
|
message: MessageState
|
|
|
|
|
|
if __name__ == "__main__":
|
|
webhook_url, title, status, variant, context_path = sys.argv[1:6]
|
|
context = None
|
|
if pathlib.Path(context_path).exists():
|
|
with open(context_path, "r", encoding="utf-8") as context_file:
|
|
context = Context(**json.load(context_file))
|
|
|
|
embed_data = Embed(
|
|
color=COLORS[variant],
|
|
title=title,
|
|
fields=[
|
|
EmbedField(name="Status", value=status, inline=True),
|
|
EmbedField(
|
|
name="Updated at",
|
|
value=str(datetime.datetime.now().ctime()),
|
|
inline=True,
|
|
),
|
|
],
|
|
)
|
|
|
|
if context:
|
|
message_id = context["message"]["id"]
|
|
response = httpx.patch(
|
|
f"{webhook_url}/messages/{message_id}?wait=true",
|
|
json={"embeds": [*context["message"]["embeds"], embed_data]},
|
|
)
|
|
else:
|
|
response = httpx.post(f"{webhook_url}?wait=true", json={"embeds": [embed_data]})
|
|
context = Context(message=response.json())
|
|
|
|
with open(context_path, "w", encoding="utf-8") as context_file:
|
|
json.dump(context, context_file)
|