add user index

This commit is contained in:
Сергей Маринкевич
2026-06-05 17:19:12 +07:00
parent 70ee7c6b81
commit b83d4f8b9b
6 changed files with 519 additions and 19 deletions
+7
View File
@@ -6,6 +6,8 @@ import os
import sqlite3
from typing import Any, Dict, Iterable, List, Optional, Tuple
import mm_users
def iter_result_files(root: str) -> Iterable[str]:
for dirpath, _dirnames, filenames in os.walk(root):
@@ -105,6 +107,7 @@ def connect_db(db_path: str) -> sqlite3.Connection:
)
"""
)
mm_users.ensure_users_schema(conn)
return conn
@@ -307,6 +310,10 @@ def main() -> int:
rows_upserted += upserted
rows_skipped += skipped
users_json = mm_users.find_users_json(args.root)
if users_json:
mm_users.upsert_users(conn, mm_users.load_users_json(users_json))
total_posts = conn.execute("SELECT COUNT(*) AS c FROM posts").fetchone()["c"]
total_sources = conn.execute("SELECT COUNT(*) AS c FROM sources").fetchone()["c"]
conn.close()