mirror of
https://github.com/fotoente/MIsskey-ebooks-bot
synced 2024-11-13 23:27:10 +00:00
59 lines
2 KiB
Python
59 lines
2 KiB
Python
import asyncio
|
|
import threading
|
|
|
|
from mi.ext import commands, tasks
|
|
from mi.framework import Note
|
|
from mi.framework.router import Router
|
|
|
|
from roboduck import *
|
|
|
|
# Load Misskey configuration
|
|
config = configparser.ConfigParser()
|
|
config.read(Path(__file__).parent.joinpath('bot.cfg'))
|
|
uri = "https://" + config.get("misskey", "instance_write")
|
|
token = config.get("misskey", "token")
|
|
|
|
|
|
class MyBot(commands.Bot):
|
|
text_model = None # Holds the markov object, so it won't be recreated everytime
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
@tasks.loop(3600)
|
|
async def loop_1h(self):
|
|
text = create_sentence()
|
|
await bot.client.note.send(content=text, visibility="home")
|
|
|
|
@tasks.loop(43200)
|
|
async def loop_12h(self):
|
|
thread_update = threading.Thread(target=update)
|
|
thread_update.daemon = True
|
|
thread_update.start()
|
|
|
|
async def on_ready(self, ws):
|
|
await Router(ws).connect_channel(["global", "main"]) # Connect to global and main channels
|
|
await bot.client.note.send(content=datetime.now().strftime('%Y-%m-%d %H:%M:%S') + " :roboduck: Bot started!",
|
|
visibility="specified")
|
|
self.loop_12h.start() # Launching renew posts every 12 hours
|
|
self.loop_1h.start() #
|
|
print(datetime.now().strftime('%Y-%m-%d %H:%M:%S') + " Roboduck Bot started!")
|
|
|
|
async def on_mention(self, note: Note):
|
|
if not note.author.is_bot:
|
|
text = note.author.action.get_mention() + " "
|
|
text += create_sentence()
|
|
|
|
await note.reply(content=text) # Reply to a note
|
|
|
|
async def on_reconnect(self, ws):
|
|
await Router(ws).connect_channel(["global", "main"]) # Connect to global and main channels
|
|
|
|
if __name__ == "__main__":
|
|
databasepath = Path(__file__).parent.joinpath('roboduck.db')
|
|
|
|
if not (os.path.exists(databasepath) and os.stat(databasepath).st_size != 0):
|
|
init_bot()
|
|
|
|
bot = MyBot()
|
|
asyncio.run(bot.start(uri, token, timeout=600))
|