59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
import sqlite3
|
|
from flask import Flask, render_template, request, request, jsonify
|
|
|
|
app = Flask(__name__)
|
|
|
|
SEARCH_DB_NAME = 'search.db'
|
|
|
|
search_db = sqlite3.connect(SEARCH_DB_NAME, check_same_thread=False)
|
|
search_db.row_factory = sqlite3.Row
|
|
|
|
def searchcommand():
|
|
query = request.form.get("query", "").strip()
|
|
include_transcript = request.form.get("includeTranscript") == "true" or request.form.get("includeTranscript") == "on"
|
|
words = [w for w in query.split() if w]
|
|
cursor = search_db.cursor()
|
|
|
|
if not include_transcript:
|
|
# Simple search: all words must be in either relative_path or filename.
|
|
conditions = []
|
|
params = []
|
|
for word in words:
|
|
conditions.append("(relative_path LIKE ? OR filename LIKE ?)")
|
|
params.extend([f"%{word}%", f"%{word}%"])
|
|
sql = "SELECT * FROM files"
|
|
if conditions:
|
|
sql += " WHERE " + " AND ".join(conditions)
|
|
cursor.execute(sql, params)
|
|
raw_results = cursor.fetchall()
|
|
results = [dict(row) for row in raw_results]
|
|
else:
|
|
# Advanced search: include transcript. Count transcript hits.
|
|
conditions = []
|
|
params = []
|
|
for word in words:
|
|
conditions.append("(relative_path LIKE ? OR filename LIKE ? OR transcript LIKE ?)")
|
|
params.extend([f"%{word}%", f"%{word}%", f"%{word}%"])
|
|
sql = "SELECT * FROM files"
|
|
if conditions:
|
|
sql += " WHERE " + " AND ".join(conditions)
|
|
cursor.execute(sql, params)
|
|
raw_results = cursor.fetchall()
|
|
|
|
results = []
|
|
for row in raw_results:
|
|
result = dict(row)
|
|
transcript = result.get("transcript") or ""
|
|
total_hits = sum(transcript.lower().count(word.lower()) for word in words)
|
|
result["transcript_hits"] = total_hits
|
|
result["transcript"] = None # Remove full transcript if needed.
|
|
results.append(result)
|
|
# Sort results so files with more transcript hits are on top.
|
|
results.sort(key=lambda x: x["transcript_hits"], reverse=True)
|
|
|
|
return jsonify(results=results)
|
|
|
|
def search():
|
|
return render_template('search.html')
|
|
|