Fix database creation
This commit is contained in:
parent
b0ea7ab935
commit
13ee068358
12
code/bot.py
12
code/bot.py
@ -1,5 +1,5 @@
|
|||||||
import discord
|
import discord
|
||||||
from discord.ext import commands
|
from discord.ext import commands, tasks
|
||||||
import os
|
import os
|
||||||
|
|
||||||
from utils.database import Base, engine
|
from utils.database import Base, engine
|
||||||
@ -14,6 +14,7 @@ class MyBot(commands.Bot):
|
|||||||
)
|
)
|
||||||
|
|
||||||
async def setup_hook(self):
|
async def setup_hook(self):
|
||||||
|
delete_accounts_task.start()
|
||||||
for ext in os.listdir("./code/cogs"):
|
for ext in os.listdir("./code/cogs"):
|
||||||
if ext.endswith(".py"):
|
if ext.endswith(".py"):
|
||||||
await self.load_extension(f"cogs.{ext[:-3]}")
|
await self.load_extension(f"cogs.{ext[:-3]}")
|
||||||
@ -28,7 +29,14 @@ async def on_ready():
|
|||||||
config.LOG.info(f"{bot.user} has connected to Discord.")
|
config.LOG.info(f"{bot.user} has connected to Discord.")
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
@tasks.loop(minutes=1)
|
||||||
|
async def delete_accounts_task():
|
||||||
|
from utils.jellyfin_delete import delete_accounts
|
||||||
|
|
||||||
Base.metadata.create_all(bind=engine)
|
Base.metadata.create_all(bind=engine)
|
||||||
|
delete_accounts()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
config.load_config()
|
config.load_config()
|
||||||
bot.run(config.BOT_TOKEN)
|
bot.run(config.BOT_TOKEN)
|
||||||
|
@ -1,10 +1,9 @@
|
|||||||
import discord
|
import discord
|
||||||
from discord import app_commands
|
from discord import app_commands
|
||||||
from discord.ext import commands, tasks
|
from discord.ext import commands
|
||||||
|
|
||||||
from utils.database import Session
|
from utils.database import Session
|
||||||
from utils.jellyfin_create import create_jellyfin_account
|
from utils.jellyfin_create import create_jellyfin_account
|
||||||
from utils.jellyfin_delete import delete_accounts
|
|
||||||
from utils.models import JellyfinAccounts
|
from utils.models import JellyfinAccounts
|
||||||
from utils.config import (
|
from utils.config import (
|
||||||
JELLYFIN_PUBLIC_URL,
|
JELLYFIN_PUBLIC_URL,
|
||||||
@ -17,9 +16,6 @@ class NewAccount(commands.Cog):
|
|||||||
def __init__(self, bot):
|
def __init__(self, bot):
|
||||||
self.bot = bot
|
self.bot = bot
|
||||||
|
|
||||||
def cog_load(self):
|
|
||||||
self.delete_accounts_loop.start()
|
|
||||||
|
|
||||||
@app_commands.command()
|
@app_commands.command()
|
||||||
@app_commands.check(lambda inter: JELLYFIN_ENABLED)
|
@app_commands.check(lambda inter: JELLYFIN_ENABLED)
|
||||||
async def newaccount(self, interaction: discord.Interaction) -> None:
|
async def newaccount(self, interaction: discord.Interaction) -> None:
|
||||||
@ -91,10 +87,6 @@ class NewAccount(commands.Cog):
|
|||||||
embed=embed, ephemeral=True
|
embed=embed, ephemeral=True
|
||||||
)
|
)
|
||||||
|
|
||||||
@tasks.loop(minutes=1)
|
|
||||||
async def delete_accounts_loop(self):
|
|
||||||
delete_accounts()
|
|
||||||
|
|
||||||
|
|
||||||
async def setup(bot):
|
async def setup(bot):
|
||||||
await bot.add_cog(NewAccount(bot))
|
await bot.add_cog(NewAccount(bot))
|
||||||
|
@ -1,6 +1,10 @@
|
|||||||
from sqlalchemy import create_engine
|
from sqlalchemy import create_engine
|
||||||
from sqlalchemy.ext.declarative import declarative_base
|
from sqlalchemy.ext.declarative import declarative_base
|
||||||
from sqlalchemy.orm import sessionmaker
|
from sqlalchemy.orm import sessionmaker
|
||||||
|
import os
|
||||||
|
|
||||||
|
if not os.path.exists("data"):
|
||||||
|
os.makedirs("data")
|
||||||
|
|
||||||
database_url = "sqlite:///data/cordarr.db"
|
database_url = "sqlite:///data/cordarr.db"
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user