def id2note(predictionList): def callback(conn,cursor): cursor.execute("SELECT * from note where status=0") values = cursor.fetchall() predictionNotes = list(filter(lambda x:x[0] in predictionList,values)) return predictionNotes return run_middleware(callback)
def startTrain(): def callback(conn, cursor): cursor.execute('UPDATE sys_state set status = %s where name = %s', ['ing', 'train']) conn.commit() return run_middleware(callback)
def isTraining(): def callback(conn, cursor): cursor.execute("select * from sys_state where name='train' ") values = cursor.fetchall() return values[0][2] == 'ing' return run_middleware(callback)
def mark_training_notes(notes): def callback(conn, cursor): for note in notes: cursor.execute('UPDATE note set training_mark=%s where id = %s', [1, note['id']]) conn.commit() return run_middleware(callback)
def get_ignore_list(): def callback(conn, cursor): cursor.execute("select * from ignore_list") values = cursor.fetchall() values = list(map(lambda x: x[1], values)) return values return run_middleware(callback)
def checkIfNeedTrain(): def callback(conn, cursor): cursor.execute( "select * from note where training_mark=0 and category!=0") values = cursor.fetchall() columns = [ 'id', 'content', 'category', 'create_time', 'modify_time', 'status' ] return [dict(zip(columns, value)) for value in values] return run_middleware(callback)
def get_categorized_notes(): def callback(conn, cursor): cursor.execute( "select * from note where category!=0 order by modify_time desc limit 150" ) values = cursor.fetchall() columns = [ 'id', 'content', 'category', 'create_time', 'modify_time', 'status' ] return [dict(zip(columns, value)) for value in values] return run_middleware(callback)
def get_uncategorized_notes(sec): def callback(conn, cursor): line = time.time() - sec cursor.execute( "select * from note where category=0 and status=0 and create_time > %s", [line]) values = cursor.fetchall() columns = [ 'id', 'content', 'category', 'create_time', 'modify_time', 'status' ] return [dict(zip(columns, value)) for value in values] return run_middleware(callback)
def get_note(noteId): def callback(conn,cursor): cursor.execute("select * from note where id=%s",[noteId]) value = cursor.fetchall()[0] return value[1] return run_middleware(callback)
def get_category(): def callback(conn, cursor): cursor.execute("select * from category order by order_number desc") return cursor.fetchall() return run_middleware(callback)