re
This commit is contained in:
parent
a125c7e98d
commit
229669d557
@ -1,7 +1,7 @@
|
|||||||
FROM python:3.12-slim
|
FROM python:3.12-slim
|
||||||
RUN mkdir -p /app
|
RUN mkdir -p /app
|
||||||
RUN mkdir -p /app/bin
|
RUN mkdir -p /app/bin
|
||||||
COPY bot/. /app
|
COPY . /app
|
||||||
COPY bin/storage.py /app/bin
|
COPY bin/storage.py /app/bin
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
RUN pip install -r requirements.txt
|
RUN pip install -r requirements.txt
|
||||||
@ -1,10 +1,32 @@
|
|||||||
class storage:
|
class storage:
|
||||||
def __init__(self, serverid, db):
|
def __init__(self, server_id, db):
|
||||||
self.serverdb = db[str(serverid)]
|
self.server_id = str(server_id)
|
||||||
|
self.db = db
|
||||||
def db(self, module, key, value=None):
|
self.collection = db['turdbot'] # Replace with your actual collection name
|
||||||
try : r = self.serverdb.find_one({"module": module}).get(key)
|
|
||||||
except AttributeError: r = None
|
async def db(self, module, key=None, value=None):
|
||||||
|
document = await self.collection.find_one({"server_id": self.server_id})
|
||||||
|
|
||||||
|
if not document:
|
||||||
|
document = {
|
||||||
|
"server_id": self.server_id,
|
||||||
|
"modules": {}
|
||||||
|
}
|
||||||
|
|
||||||
|
modules = document.get("modules", {})
|
||||||
|
|
||||||
|
if module not in modules:
|
||||||
|
modules[module] = {}
|
||||||
|
|
||||||
|
if key is None:
|
||||||
|
return modules.get(module)
|
||||||
|
|
||||||
if value is not None:
|
if value is not None:
|
||||||
self.serverdb.update_one({"module": module}, {"$set": {key: value}}, upsert=True)
|
modules[module][key] = value
|
||||||
return r
|
await self.collection.update_one(
|
||||||
|
{"server_id": self.server_id},
|
||||||
|
{"$set": {"modules": modules}},
|
||||||
|
upsert=True
|
||||||
|
)
|
||||||
|
|
||||||
|
return modules[module].get(key)
|
||||||
@ -1,12 +1,9 @@
|
|||||||
from os import environ as env
|
from os import environ as env
|
||||||
import discord
|
import discord
|
||||||
from discord.ext import bridge
|
from discord.ext import bridge
|
||||||
import logging
|
from motor.motor_asyncio import AsyncIOMotorClient as MongoClient
|
||||||
from pymongo import MongoClient
|
|
||||||
from bin.storage import storage
|
from bin.storage import storage
|
||||||
|
|
||||||
logging.basicConfig(filename="turdbot.log",level=logging.INFO)
|
|
||||||
|
|
||||||
bot = bridge.Bot(
|
bot = bridge.Bot(
|
||||||
help_command=None,
|
help_command=None,
|
||||||
command_prefix="!",
|
command_prefix="!",
|
||||||
@ -16,22 +13,20 @@ bot = bridge.Bot(
|
|||||||
name="you")
|
name="you")
|
||||||
)
|
)
|
||||||
|
|
||||||
bot.load_extension("cogs.reply")
|
bot.load_extension("cogs.triggers")
|
||||||
bot.load_extension("cogs.counting")
|
bot.load_extension("cogs.counting")
|
||||||
bot.load_extension("cogs.settings")
|
bot.load_extension("cogs.settings")
|
||||||
|
|
||||||
uri = f"mongodb://{env['DB_USERNAME']}:{env['DB_PASSWORD']}@{env['DB_HOST']}/?authSource=admin"
|
uri = f"mongodb://{env['DB_USERNAME']}:{env['DB_PASSWORD']}@{env['DB_HOST']}/?authSource=admin"
|
||||||
bot.db = MongoClient(uri)["turdbot"]
|
bot.db = MongoClient(uri)["turdbot"]
|
||||||
|
|
||||||
|
bot.version = "4.0.0ALPHA"
|
||||||
|
|
||||||
@bot.event
|
@bot.event
|
||||||
async def on_ready():
|
async def on_ready():
|
||||||
print("Logged in as")
|
print("Logged in as")
|
||||||
print(bot.user.name)
|
print(bot.user.name)
|
||||||
print(bot.user.id)
|
print(bot.user.id)
|
||||||
# logging.info(f"Logged in as {bot.user} (ID: {bot.user.id})")
|
print("------")
|
||||||
# for guild in bot.guilds:
|
|
||||||
# logging.info(f"Added {guild.name} (ID: {guild.id})")
|
|
||||||
# storage(guild.id, bot.db).update_guild()
|
|
||||||
|
|
||||||
|
|
||||||
bot.run(env["BOT_TOKEN"])
|
bot.run(env["BOT_TOKEN"])
|
||||||
@ -1,12 +0,0 @@
|
|||||||
from discord.ext import bridge, commands
|
|
||||||
|
|
||||||
class Reply(commands.Cog):
|
|
||||||
def __init__(self, bot):
|
|
||||||
self.bot = bot
|
|
||||||
|
|
||||||
@bridge.bridge_command()
|
|
||||||
async def reply(self, ctx, args):
|
|
||||||
await ctx.respond(args)
|
|
||||||
|
|
||||||
def setup(bot):
|
|
||||||
bot.add_cog(Reply(bot))
|
|
||||||
14
cogs/misc.py
Normal file
14
cogs/misc.py
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
from discord.ext import bridge, commands
|
||||||
|
from bin.storage import storage
|
||||||
|
|
||||||
|
class Misc(commands.Cog):
|
||||||
|
def __init__(self, bot):
|
||||||
|
self.bot = bot
|
||||||
|
|
||||||
|
@bridge.bridge_command()
|
||||||
|
async def ping(self, ctx):
|
||||||
|
await ctx.send(f"pong! {round(self.bot.latency * 1000)}ms")
|
||||||
|
|
||||||
|
@bridge.bridge_command()
|
||||||
|
async def version(self, ctx):
|
||||||
|
await ctx.send(self.bot.version)
|
||||||
0
cogs/ping.py
Normal file
0
cogs/ping.py
Normal file
19
cogs/triggers.py
Normal file
19
cogs/triggers.py
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
from discord.ext import bridge, commands
|
||||||
|
from bin.storage import storage
|
||||||
|
|
||||||
|
class Triggers(commands.Cog):
|
||||||
|
def __init__(self, bot):
|
||||||
|
self.bot = bot
|
||||||
|
|
||||||
|
@commands.Cog.listener()
|
||||||
|
async def on_message(self, message):
|
||||||
|
if message.author.bot:
|
||||||
|
return
|
||||||
|
db = storage(message.guild.id, self.bot.db).db
|
||||||
|
if db('triggers', 'enabled'):
|
||||||
|
for trigger in db('triggers', 'triggers'):
|
||||||
|
if trigger.lower() in message.content.lower():
|
||||||
|
await message.reply(db('triggers', 'response'))
|
||||||
|
|
||||||
|
def setup(bot):
|
||||||
|
bot.add_cog(Triggers(bot))
|
||||||
@ -1,8 +0,0 @@
|
|||||||
FROM python:3.12-slim
|
|
||||||
RUN mkdir -p /app
|
|
||||||
RUN mkdir -p /app/bin
|
|
||||||
COPY flask/. /app
|
|
||||||
COPY bin/storage.py /app/bin
|
|
||||||
WORKDIR /app
|
|
||||||
RUN pip install -r requirements.txt
|
|
||||||
ENTRYPOINT ["granian", "--port", "5005", "--interface", "wsgi", "app:app"]
|
|
||||||
19
flask/app.py
19
flask/app.py
@ -1,19 +0,0 @@
|
|||||||
from flask import Flask
|
|
||||||
from pymongo import MongoClient
|
|
||||||
from bin.storage import storage
|
|
||||||
from os import environ as env
|
|
||||||
import logging
|
|
||||||
|
|
||||||
logging.basicConfig(level=logging.INFO)
|
|
||||||
|
|
||||||
app = Flask(__name__)
|
|
||||||
|
|
||||||
uri = f"mongodb://{env['DB_USERNAME']}:{env['DB_PASSWORD']}@{env['DB_HOST']}/?authSource=admin"
|
|
||||||
app.db = MongoClient(uri)["turdbot"]
|
|
||||||
|
|
||||||
|
|
||||||
@app.route("/")
|
|
||||||
def hello():
|
|
||||||
db = storage(1004178748205187083,app.db).db
|
|
||||||
count = db('counting', 'count')
|
|
||||||
return f"Hello World! {count}"
|
|
||||||
@ -1,3 +0,0 @@
|
|||||||
granian
|
|
||||||
flask
|
|
||||||
pymongo
|
|
||||||
@ -1,3 +1,4 @@
|
|||||||
py-cord[audio]
|
py-cord[audio]
|
||||||
pymongo
|
pymongo
|
||||||
PyNaCl
|
PyNaCl
|
||||||
|
motor
|
||||||
Loading…
Reference in New Issue
Block a user