i did something

.gitignore

@@ -1,2 +1,4 @@
-venv
-__pycache__/
+venv/
+env/
+__pycache__/
+bot.db

main.py

@@ -1,10 +1,16 @@
-import sys
 from os import listdir
 from discord.ext.commands import Bot
-from src.utils.config import build_config
+from configparser import ConfigParser
+from peewee import SqliteDatabase
 
 # read config at configured location \\ default to 'config.ini'
-config = build_config()
+config = ConfigParser()
+with open('config.ini', 'r') as configfile:
+    config.read_file(configfile)
+
+# init db
+db = SqliteDatabase(config['DATABASE']['path'])
+db.connect()
 
 # spawn discord bot instance
 # init token from config

src/database/migrations/core.py

@@ -0,0 +1,10 @@
+from src.database.models import Role, User, UserRole
+from main import db
+
+import logging
+
+logger = logging.getLogger('database')
+
+def create_tables():
+    logger.info('Creating core database tables if necessary.')
+    db.create_tables([User, Role, UserRole])

src/database/models.py

@@ -0,0 +1,17 @@
+from peewee import Model, CharField, ForeignKeyField
+from main import config, db
+
+class BaseModel(Model):
+    class Meta:
+        database = db
+
+class User(BaseModel):
+    uuid = CharField(unique=True)
+
+class Role(BaseModel):
+    name = CharField()
+
+class UserRole(BaseModel):
+    user = ForeignKeyField(User, User.uuid)
+    role = ForeignKeyField(Role, Role.name)
+

src/db/__init__.py

@@ -1,2 +0,0 @@
-from .database import BaseModel
-from .models import *

src/db/database.py

@@ -1,15 +0,0 @@
-from peewee import Model, SqliteDatabase
-from main import config
-
-DB_PATH = 'app.db' 
-
-database = SqliteDatabase(DB_PATH, pragmas={
-    'journal_mode': 'wal',
-    'cache_size': 10000,
-    'foreign_keys': 1
-})
-
-class BaseModel(Model):
-
-    class Meta:
-        database = database

src/db/models/__init__.py

@@ -1,14 +0,0 @@
-from .collectible import Collectible
-from .inventory import Inventory
-from .item import Item
-from .recipe import ItemRecipe, ItemRecipeShard
-from .user import User
-
-__all__ = [
-    Collectible
-    , Inventory
-    , Item
-    , ItemRecipe
-    , ItemRecipeShard
-    , User
-]

src/db/models/collectible.py

@@ -1,6 +0,0 @@
-from peewee import CharField
-from src.db import BaseModel
-
-class Collectible(BaseModel):
-    name = CharField()
-    picture = CharField()

src/db/models/inventory.py

@@ -1,7 +0,0 @@
-from peewee import CharField, ForeignKeyField, IntegerField
-from src.db import BaseModel
-
-class Inventory(BaseModel):
-    user_id = CharField()
-    item_id = ForeignKeyField(Item)
-    amount = IntegerField(default=0)

src/db/models/item.py

@@ -1,6 +0,0 @@
-from peewee import CharField
-from src.db.database import BaseModel
-
-class Item(BaseModel):
-    name = CharField()
-    picture = CharField()

src/db/models/recipe.py

@@ -1,10 +0,0 @@
-from peewee import ForeignKeyField, IntegerField
-from . import Collectible, Item
-
-class ItemRecipe(BaseModel):
-    output_item_id = ForeignKeyField(Item)
-
-class ItemRecipeShard(BaseModel):
-    recipe_id = ForeignKeyField(ItemRecipe)
-    collectible_id = ForeignKeyField(Collectible)
-    amount = IntegerField()

src/db/models/user.py

@@ -1,8 +0,0 @@
-from peewee import CharField, IntegerField
-from src.db import BaseModel
-
-class User(BaseModel):
-    user_id = CharField()
-    xp = IntegerField(default=0) # Total xp accumulated
-    xp_mod = IntegerField(default=100) # XP Modifier in Percent // Can be 0 for XP-Lock, but not negative
-    level = IntegerField(default=1)

src/ext/__init__.py

@@ -0,0 +1,2 @@
+def setup(_):
+    pass

src/ext/events.py

@@ -1,6 +1,11 @@
 import discord
 from discord.ext import commands
-from peewee import *
+import logging
+
+import src.database.migrations.core
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
 
 class Events(commands.Cog):
     def __init__(self, bot):
@@ -15,7 +20,9 @@ class Events(commands.Cog):
         - activity
         - presence
         """
-        print(f"[INFO]: Startup complete | {self.bot.user}")
+        logger.info("Running database startup migrations.")
+        src.database.migrations.core.create_tables()
+        logger.info(f"Startup complete | {self.bot.user}")
 
 def setup(bot):
     bot.add_cog(Events(bot))

src/ext/inventory.py

@@ -1,11 +1,10 @@
 from discord.ext import commands
-from src.utils.config import build_config
-
+from main import config
 
 class Inventory(commands.Cog):
     def __init__(self, bot):
         self.bot = bot
-        self.config = build_config()
+        # self.config = config
 
     @commands.group()
     @commands.guild_only()
@@ -31,7 +30,7 @@ class Inventory(commands.Cog):
 
     @inventory.command(name="use")
     @commands.guild_only()
-    async def inventory_use(self, ctx, *, ):
+    async def inventory_use(self, ctx, **kwargs):
         """
         Use an item in user's inventory
 
@@ -40,4 +39,5 @@ class Inventory(commands.Cog):
 
         pass
 
-
+def setup(bot):
+    bot.add_cog(Inventory(bot))

src/utils/config.py

@@ -1,17 +0,0 @@
-from os import getenv
-from sys import exit
-
-from configparser import ConfigParser
-
-def build_config():
-    if (config_path := getenv("CONFIG_PATH")) is None:
-        config_path = "config.ini"
-
-    config = ConfigParser()
-    try:
-        config.read_file(open(config_path))
-    except FileNotFoundError:
-        print("Unable to locate config file. Exiting...")
-        exit(1)
-    
-    return config