WhatTheMike/Fox-V3
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix has_table, move age and minutes to trainset

Browse Source
This commit is contained in:
bobloy 2021-07-06 12:01:27 -04:00
parent e0a361b952
commit 47269ba8f4
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
chatter/chat.py
View File

@ -402,7 +402,13 @@ class Chatter(Cog):
) )
@commands.is_owner() @commands.is_owner()
@chatter.command(name="minutes") @chatter.group(name="trainset")
async def chatter_trainset(self, ctx: commands.Context):
"""Commands for configuring training"""
pass
@commands.is_owner()
@chatter_trainset.command(name="minutes")
async def minutes(self, ctx: commands.Context, minutes: int): async def minutes(self, ctx: commands.Context, minutes: int):
""" """
Sets the number of minutes the bot will consider a break in a conversation during training Sets the number of minutes the bot will consider a break in a conversation during training
@ -418,7 +424,7 @@ class Chatter(Cog):
await ctx.tick() await ctx.tick()
@commands.is_owner() @commands.is_owner()
@chatter.command(name="age") @chatter_trainset.command(name="age")
async def age(self, ctx: commands.Context, days: int): async def age(self, ctx: commands.Context, days: int):
""" """
Sets the number of days to look back Sets the number of days to look back

4
chatter/storage_adapters.py
View File

@ -5,7 +5,7 @@ class MyDumbSQLStorageAdapter(SQLStorageAdapter):
def __init__(self, **kwargs): def __init__(self, **kwargs):
super(SQLStorageAdapter, self).__init__(**kwargs) super(SQLStorageAdapter, self).__init__(**kwargs)
from sqlalchemy import create_engine from sqlalchemy import create_engine, inspect
from sqlalchemy.orm import sessionmaker from sqlalchemy.orm import sessionmaker
self.database_uri = kwargs.get("database_uri", False) self.database_uri = kwargs.get("database_uri", False)
@ -31,7 +31,7 @@ class MyDumbSQLStorageAdapter(SQLStorageAdapter):
dbapi_connection.execute("PRAGMA journal_mode=WAL") dbapi_connection.execute("PRAGMA journal_mode=WAL")
dbapi_connection.execute("PRAGMA synchronous=NORMAL") dbapi_connection.execute("PRAGMA synchronous=NORMAL")
if not self.engine.dialect.has_table(self.engine, "Statement"): if not inspect(self.engine).has_table('Statement'):
self.create_database() self.create_database()
self.Session = sessionmaker(bind=self.engine, expire_on_commit=True) self.Session = sessionmaker(bind=self.engine, expire_on_commit=True)