api/seed.py

22 lines
460 B
Python

#!/usr/bin/env python3
import argparse
from app.seeders.database_seeder import seed_database
def main():
parser = argparse.ArgumentParser(description="Seed the database with sample data")
parser.add_argument(
"--count", "-c",
type=int,
default=5,
help="Number of records to seed for each model (default: 5)",
)
args = parser.parse_args()
seed_database(args.count)
if __name__ == "__main__":
main()