Trigent/migrate_database.py at main · QuantStack/Trigent · GitHub

Repository navigation

47 lines (33 loc) · 1.43 KB

#!/usr/bin/env python3

"""Migration script to convert TinyDB JSON files to BetterJSONStorage format."""

from pathlib import Path

from tinydb import TinyDB

from rich_issue_mcp.database import FixedBetterJSONStorage, get_data_directory

def migrate_file(json_file: Path):

"""Migrate a JSON file to TinyDB with BetterJSONStorage."""

print(f"Migrating {json_file.name}...")

# Read data using standard TinyDB

with TinyDB(json_file, access_mode="r") as old_db:

issues_list = old_db.all()

print(f" Read {len(issues_list)} records from original file")

# Create new .db file path

db_file = json_file.with_suffix('.db')

print(f" Creating new database: {db_file.name}")

# Write to new database with BetterJSONStorage using batch insert

with TinyDB(db_file, access_mode="r+", storage=FixedBetterJSONStorage) as new_db:

# Clear any existing data and batch insert all issues

new_db.truncate()

new_db.insert_multiple(issues_list)

print(f" ✅ Migration complete: {len(issues_list)} issues written")

def main():

"""Find and migrate all JSON files in the data directory."""

data_dir = get_data_directory()

json_files = list(data_dir.glob("*.json"))

if not json_files:

print("No JSON files found to migrate")

return

for json_file in json_files:

migrate_file(json_file)

print()

if __name__ == "__main__":

main()