from datetime import datetime, time from peewee import CharField, DateTimeField, Expression, Model, SqliteDatabase from .config import DB_PATH from .utils import blue, highlight_tags, pink # Add regex function to SqliteDatabase OP_REGEXP = 'regexp' def regexp(lhs, rhs): return Expression(lhs, 'regexp', rhs) SqliteDatabase.register_ops({OP_REGEXP: 'REGEXP'}) # Create database reference db = SqliteDatabase(DB_PATH) # Define REGEXP function in sqlite database connection conn = db.get_conn() def regex_matches(regex, string): return bool(re.search(regex, string, flags=re.IGNORECASE)) conn.create_function('REGEXP', 2, regex_matches)