Spaces:
Running
Running
File size: 2,846 Bytes
351d460 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
#!/usr/bin/env python3
import sys
import os
sys.path.append('.')
from app.database import SessionLocal
from app import crud, models
def fix_image_counts():
"""Update image_count for all existing captions based on their linked images"""
db = SessionLocal()
try:
print("Starting image_count fix for existing multi-uploads...")
# Get all captions with their linked images
captions = crud.get_all_captions_with_images(db)
print(f"Found {len(captions)} captions to process")
updated_count = 0
skipped_count = 0
for caption in captions:
# Skip if image_count is already set correctly
if caption.image_count is not None and caption.image_count > 0:
if caption.image_count == len(caption.images):
skipped_count += 1
continue
# Calculate the correct image count
correct_image_count = len(caption.images)
if correct_image_count == 0:
print(f"Warning: Caption {caption.caption_id} has no linked images")
continue
# Update the image_count
old_count = caption.image_count
caption.image_count = correct_image_count
print(f"Updated caption {caption.caption_id}: {old_count} -> {correct_image_count} (title: '{caption.title}')")
updated_count += 1
# Commit all changes
db.commit()
print(f"\nDatabase update complete!")
print(f"Updated: {updated_count} captions")
print(f"Skipped: {skipped_count} captions (already correct)")
# Verify the changes
print("\nVerifying changes...")
captions_after = crud.get_all_captions_with_images(db)
multi_uploads = [c for c in captions_after if c.image_count and c.image_count > 1]
single_uploads = [c for c in captions_after if c.image_count == 1]
null_counts = [c for c in captions_after if c.image_count is None or c.image_count == 0]
print(f"Multi-uploads (image_count > 1): {len(multi_uploads)}")
print(f"Single uploads (image_count = 1): {len(single_uploads)}")
print(f"Captions with null/zero image_count: {len(null_counts)}")
if null_counts:
print("\nCaptions still with null/zero image_count:")
for c in null_counts[:5]: # Show first 5
print(f" - {c.caption_id}: {len(c.images)} linked images, image_count={c.image_count}")
except Exception as e:
print(f"Error: {e}")
import traceback
traceback.print_exc()
db.rollback()
finally:
db.close()
if __name__ == "__main__":
fix_image_counts()
|