initial_data.py 633 B

123456789101112131415161718192021222324252627282930
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. # File : initial_data.py
  4. # Author: DaShenHan&道长-----先苦后甜,任凭晚风拂柳颜------
  5. # Author's Blog: https://blog.csdn.net/qq_32394351
  6. # Date : 2023/12/6
  7. from db.session import SessionLocal
  8. from db.init_db import init_db
  9. import logging
  10. logging.basicConfig(level=logging.INFO)
  11. logger = logging.getLogger(__name__)
  12. def init() -> None:
  13. # init_db()
  14. with SessionLocal() as session:
  15. init_db(session)
  16. def main() -> None:
  17. logger.info("Creating initial data")
  18. init()
  19. logger.info("Initial data created")
  20. if __name__ == "__main__":
  21. main()