X-Git-Url: https://apps.delaporte.us/gitweb/matrix-chore-bot.git/blobdiff_plain/68b00c8ecbd23b7eed4d0942aa8b45f89f2810e0..145745886f1614e292c17163884e15e669dcf224:/matrix_chore_bot/db.py diff --git a/matrix_chore_bot/db.py b/matrix_chore_bot/db.py index 315d279..1a07bec 100644 --- a/matrix_chore_bot/db.py +++ b/matrix_chore_bot/db.py @@ -13,7 +13,6 @@ DATE_FORMAT = '%Y-%m-%d' def today(): return dt.now().strftime(DATE_FORMAT) - logger = logging.getLogger(__name__) logging.basicConfig(filename='chore.log', encoding='utf-8', level=logging.DEBUG) @@ -39,7 +38,7 @@ class Data(): def __init__(self, db_file=None): if not db_file: - db_file = os.path.join(os.environ.get('MATRIX_DATA_DIR'), 'chores.db') + db_file = DB_FILE self.file = db_file self.engine = self.get_engine() @@ -86,12 +85,23 @@ class Data(): cmd = select(Chore) chores = session.exec(cmd).all() if(not chores): - retuls.append('None Found') + results.append('None Found') for chore in chores: results.append(Data.check_chore(session, chore.id)) return results + def filter(self, term): + results = [] + for item in self.list(): + if term in item: + results.append(item) + + return results + + + + def add(self, name): try: with Session(self.engine) as session: