111 lines
3.3 KiB
Python
111 lines
3.3 KiB
Python
# Copyright (c) 2022, Ian Burgess
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the GPLv3 license. A copy of this license can be found in the LICENSE file in the root directory of this source tree.
|
|
|
|
import discord
|
|
import random
|
|
import datetime
|
|
import datetime
|
|
import shelve
|
|
import backports.zoneinfo as zoneinfo
|
|
from storage import config
|
|
import version
|
|
from discord import app_commands
|
|
from discord.ext import tasks, commands
|
|
|
|
def run():
|
|
|
|
quotetime = datetime.time(hour=12, tzinfo=zoneinfo.ZoneInfo("MST"))
|
|
avgtimelst = []
|
|
|
|
token = open("TOKEN", "r").read()
|
|
intents = discord.Intents.all()
|
|
bot = commands.Bot(intents=intents, command_prefix=".")
|
|
|
|
|
|
# on ready
|
|
@bot.event
|
|
async def on_ready():
|
|
print('logged in as {0.user}'.format(bot))
|
|
print("discord.py version",discord.__version__)
|
|
print("turdbot version",version.local())
|
|
|
|
if not dailyquote.is_running():
|
|
dailyquote.start()
|
|
|
|
|
|
@bot.hybrid_command(name="eee", with_app_command=True)
|
|
async def foo(ctx):
|
|
for guild in bot.guilds:
|
|
await ctx.send(guild.id)
|
|
|
|
|
|
# main event
|
|
@bot.event
|
|
async def on_message(message):
|
|
# defining variables
|
|
msg = str(message.content)
|
|
channel = str(message.channel.id)
|
|
server = message.guild.id
|
|
author = str(message.author)
|
|
|
|
# reply to bots
|
|
if message.author.bot and config("replytobot") == "false":
|
|
return
|
|
|
|
# start event timer
|
|
print("\n start client.event")
|
|
start = datetime.datetime.now()
|
|
|
|
# triggerbot
|
|
if config("triggerbotenabled") == "true":
|
|
x = 0
|
|
found = "false"
|
|
triggers = config("triggerbottriggers", db="data")
|
|
while x < len(triggers) and found == "false":
|
|
if triggers[x] not in msg:
|
|
x = x+1
|
|
elif triggers[x] in msg:
|
|
found = "true"
|
|
replys = config(triggerbotreplys, db=data)
|
|
rand = random.randint(0, len(replys))-1
|
|
await message.channel.send(replys[rand])
|
|
else:
|
|
print("something happened")
|
|
|
|
# daily quote
|
|
if channel == config("quotequeue") and config("quotebotenabled") == "true":
|
|
file = open("data/quotes.var", "a")
|
|
file.write(msg)
|
|
file.write(":")
|
|
file.close()
|
|
|
|
# bot commands
|
|
await bot.process_commands(message)
|
|
|
|
# end function timer
|
|
end = datetime.datetime.now()
|
|
eventime = (end - start)
|
|
microseconds = eventime.microseconds
|
|
avgtimelst.append(microseconds)
|
|
avgtime = sum(avgtimelst) / len(avgtimelst)
|
|
print("\n end client.event \n microseconds: " +
|
|
str(microseconds) + "\n average time: " + str(avgtime))
|
|
|
|
|
|
@tasks.loop(time=quotetime, reconnect=True)
|
|
async def dailyquote():
|
|
if config("quotebotenabled") == "true":
|
|
file = open("data/quotes.var", "r+")
|
|
list = file.read().split(":")
|
|
rand = random.randint(0, (len(list)-1))
|
|
print(rand)
|
|
channel = bot.get_channel(int(quotebotconf["config"]["dailyquote"]))
|
|
await channel.send(list.pop(rand))
|
|
file.write(":".join(list))
|
|
file.close()
|
|
|
|
|
|
bot.run(token)
|