#!/usr/bin/env python
"""Quick seed data generation script"""

import asyncio
import sys
from utils.seed_data import generate_seed_data, clear_all_data


async def main():
    if len(sys.argv) > 1:
        if sys.argv[1] == "clear":
            confirm = input("Delete ALL data? (yes/no): ")
            if confirm.lower() == "yes":
                await clear_all_data()
            else:
                print("Aborted.")
        else:
            count = int(sys.argv[1])
            await generate_seed_data(count)
    else:
        # Default: generate 25 artifacts
        await generate_seed_data(25)


if __name__ == "__main__":
    asyncio.run(main())