create new project

This commit is contained in:
falsycat 2023-02-11 02:34:34 +09:00
parent c97e0b27ed
commit 6bde533161
2 changed files with 120 additions and 0 deletions

2
.gitignore vendored Normal file
View File

@ -0,0 +1,2 @@
/config.json
/auth.json

118
main.py Normal file
View File

@ -0,0 +1,118 @@
import json
import os
import time
from datetime import datetime, timezone
from html.parser import HTMLParser
from mastodon import Mastodon
class Handler:
def __init__(self, app, st):
self.app = app
self.st = st
self.text = sanitize_html(st["content"])
self.tokens = self.text.split()[1:]
self.handle()
def handle(self):
print(self.tokens)
if self.tokens[0] == "watch":
self.watch()
else:
raise Exception("unknown command")
def watch(self):
date = self.tokens[1]
slots = [int(x) for x in self.tokens[2:]]
elapsed = datetime.now(timezone.utc) - self.st["created_at"]
if elapsed.seconds < self.app.CONFIG["interval"]:
self.app.reply(self.st, "accepted")
print("watching", date, slots)
class App:
def __init__(self):
with open("config.json", "r") as f:
self.CONFIG = json.load(f)
self.M = self.authenticate()
self.run()
def reply(self, st, msg):
try:
print(msg)
#self.M.status_post(
# status = "@{} {}".format(st["account"]["acct"], msg),
# in_reply_to_id = st["id"])
except Exception as e:
print("reply failure:", msg, "(", e, ")")
def run(self):
while True:
try:
items = self.M.notifications(
account_id = self.CONFIG["master"],
types = "mention")
for item in items:
st = item["status"]
# handle the item
try:
Handler(self, st)
except Exception as e:
self.reply(st, "handling aborted: {}".format(e))
self.M.notifications_dismiss(id = item["id"])
print("cycle done")
except Exception as e:
print("cycle aborted:", e)
time.sleep(self.CONFIG["interval"])
return
def authenticate(self):
if os.path.exists("auth.json"):
with open("auth.json", "r") as f:
auth = json.load(f)
M = Mastodon(
api_base_url = self.CONFIG["instance"],
client_id = auth["client_id"],
client_secret = auth["secret"],
access_token = auth["token"])
else:
cid, secret = Mastodon.create_app(
client_name = "maidbot",
api_base_url = self.CONFIG["instance"])
M = Mastodon(
client_id = cid,
client_secret = secret,
api_base_url = self.CONFIG["instance"])
tok = M.log_in(
username = self.CONFIG["username"],
password = self.CONFIG["password"])
dic = {"client_id": cid, "secret": secret, "token": tok}
with open("auth.json", "w") as f:
json.dump(dic, f, indent = 2)
return M
def sanitize_html(html):
class F(HTMLParser):
text = ""
def handle_starttag(self, tag, attrs):
if tag == "br": self.text += " "
def handle_data(self, data):
self.text += data
f = F()
f.feed(html)
return f.text
App()