data_nexus/reaggregate_records.py

from app.database import SessionLocal
from app.models import DataVersion
from app.services.storage_service import StorageService
from app.services.gogs_client import GogsClient
import logging

# Setup basic logging to see what's happening
logging.basicConfig(level=logging.INFO)

def reaggregate():
    db = SessionLocal()
    gogs = GogsClient()
    storage = StorageService(db, gogs)
    
    versions = db.query(DataVersion).all()
    print(f"Found {len(versions)} versions to re-aggregate.")
    
    for v in versions:
        print(f"Re-aggregating version {v.id} (Stage: {v.stage}, Commit: {v.commit_id[:8]})")
        try:
            storage.aggregate_version_records(v)
        except Exception as e:
            print(f"Error re-aggregating version {v.id}: {e}")
    
    db.close()
    print("Done!")

if __name__ == "__main__":
    reaggregate()