add user index
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user