repo restructure

examples/demo_asset_tracking.py

@@ -0,0 +1,185 @@
+#!/usr/bin/env python3
+"""
+Demo Asset Tracking
+
+This script demonstrates the asset tracking functionality by showing
+how new and modified assets are detected and downloaded.
+"""
+
+import asyncio
+import logging
+import sys
+import os
+from pathlib import Path
+
+# Add the parent directory to the path so we can import modules
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from src.auth_manager import AuthManager
+from src.asset_tracker import AssetTracker
+from src.image_downloader import ImageDownloader
+
+
+async def demo_asset_tracking():
+    """Demonstrate asset tracking functionality."""
+    print("🎯 ParentZone Asset Tracking Demo")
+    print("=" * 60)
+
+    # Configuration
+    email = "tudor.sitaru@gmail.com"
+    password = "mTVq8uNUvY7R39EPGVAm@"
+    output_dir = "downloaded_images"
+
+    # Setup logging
+    logging.basicConfig(
+        level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+    )
+
+    try:
+        print("\n🔐 Step 1: Testing Authentication")
+        print("-" * 40)
+
+        # Test authentication first
+        auth_manager = AuthManager()
+        success = await auth_manager.login(email, password)
+
+        if not success:
+            print("❌ Authentication failed!")
+            return False
+
+        print(f"✅ Authentication successful!")
+        print(f"   User: {auth_manager.user_name}")
+        print(f"   Provider: {auth_manager.provider_name}")
+
+        print(f"\n📊 Step 2: Check Current Asset Status")
+        print("-" * 40)
+
+        # Check current asset tracker status
+        tracker = AssetTracker(storage_dir=output_dir)
+        stats = tracker.get_stats()
+
+        print(f"Current local assets:")
+        print(f"   Total tracked: {stats['total_tracked_assets']}")
+        print(f"   Existing files: {stats['existing_files']}")
+        print(f"   Missing files: {stats['missing_files']}")
+        print(f"   Total size: {stats['total_size_mb']} MB")
+
+        print(f"\n⬇️  Step 3: Download Assets (First Run)")
+        print("-" * 40)
+        print("This will download new assets and skip existing ones...")
+
+        # Create downloader with asset tracking enabled
+        downloader = ImageDownloader(
+            api_url="https://api.parentzone.me",
+            list_endpoint="/v1/media/list",
+            download_endpoint="/v1/media",
+            output_dir=output_dir,
+            email=email,
+            password=password,
+            track_assets=True,
+            max_concurrent=3,
+        )
+
+        # First download run
+        print("\n🚀 Starting download...")
+        await downloader.download_all_assets()
+
+        # Show results
+        print(f"\n📈 First Run Results:")
+        print(f"   Total assets found: {downloader.stats['total']}")
+        print(f"   Successfully downloaded: {downloader.stats['successful']}")
+        print(f"   Failed downloads: {downloader.stats['failed']}")
+        print(f"   Skipped (existing): {downloader.stats['skipped']}")
+
+        if downloader.asset_tracker:
+            updated_stats = downloader.asset_tracker.get_stats()
+            print(f"\n📊 Updated Asset Status:")
+            print(f"   Total tracked: {updated_stats['total_tracked_assets']}")
+            print(f"   Existing files: {updated_stats['existing_files']}")
+            print(f"   Total size: {updated_stats['total_size_mb']} MB")
+
+        print(f"\n🔄 Step 4: Second Run (Should Skip All)")
+        print("-" * 40)
+        print("Running again - should detect no new assets...")
+
+        # Reset stats for second run
+        downloader.stats = {"total": 0, "successful": 0, "failed": 0, "skipped": 0}
+
+        # Second download run
+        await downloader.download_all_assets()
+
+        print(f"\n📈 Second Run Results:")
+        print(f"   Assets to download: {downloader.stats['total']}")
+        print(f"   New downloads: {downloader.stats['successful']}")
+
+        if downloader.stats["total"] == 0:
+            print("   ✅ Perfect! No new assets found - all are up to date!")
+        else:
+            print(f"   Downloaded: {downloader.stats['successful']}")
+            print(f"   Failed: {downloader.stats['failed']}")
+
+        print(f"\n🧹 Step 5: Cleanup and Final Stats")
+        print("-" * 40)
+
+        if downloader.asset_tracker:
+            # Cleanup any missing files
+            print("Checking for missing files...")
+            downloader.asset_tracker.cleanup_missing_files()
+
+            # Final statistics
+            final_stats = downloader.asset_tracker.get_stats()
+            print(f"\n📊 Final Statistics:")
+            print(f"   Total tracked assets: {final_stats['total_tracked_assets']}")
+            print(f"   Successful downloads: {final_stats['successful_downloads']}")
+            print(f"   Failed downloads: {final_stats['failed_downloads']}")
+            print(f"   Existing files: {final_stats['existing_files']}")
+            print(f"   Total size: {final_stats['total_size_mb']} MB")
+
+        print(f"\n✨ Demo completed successfully!")
+        print("=" * 60)
+
+        return True
+
+    except Exception as e:
+        print(f"❌ Demo failed with error: {e}")
+        import traceback
+
+        traceback.print_exc()
+        return False
+
+
+def show_usage():
+    """Show usage information."""
+    print("Asset Tracking Demo")
+    print("=" * 30)
+    print("This demo shows how the asset tracking system:")
+    print("• Identifies new assets to download")
+    print("• Skips already downloaded assets")
+    print("• Detects modified assets")
+    print("• Maintains local metadata")
+    print()
+    print("Usage:")
+    print("  python3 demo_asset_tracking.py        # Run the demo")
+    print("  python3 demo_asset_tracking.py --help # Show this help")
+    print()
+    print("The demo will:")
+    print("1. Authenticate with ParentZone API")
+    print("2. Check current local asset status")
+    print("3. Download new/modified assets (first run)")
+    print("4. Run again to show efficient skipping")
+    print("5. Display final statistics")
+
+
+async def main():
+    """Main function."""
+    if len(sys.argv) > 1 and sys.argv[1] in ["--help", "-h"]:
+        show_usage()
+        return 0
+
+    print("🚀 Starting ParentZone Asset Tracking Demo...")
+    success = await demo_asset_tracking()
+    return 0 if success else 1
+
+
+if __name__ == "__main__":
+    exit(asyncio.run(main()))