modified init_db

backend/app/app/db/init_db.py

@@ -26,16 +26,16 @@ def init_db(db: Session) -> None:
     engine = create_engine(f'mysql://{user}:{password}@{host}/{db_name}', pool_pre_ping=True)
     base.Base.metadata.create_all(bind=engine)
 
-    if settings.MEMBERSHIP_TYPES:
-      for TYPE in settings.MEMBERSHIP_TYPES:
-          if not db.query(Membership).filter(Membership.status == TYPE).first():
-                db.add(Membership(status=TYPE))
+    if settings.MEMBERSHIP_STATUS:
+      for STATUS in settings.MEMBERSHIP_STATUS:
+          if not db.query(Membership).filter(Membership.status == STATUS).first():
+                db.add(Membership(status=STATUS))
       db.commit()
 
-    if settings.PROGRESS_TYPES:
-        for TYPE in settings.PROGRESS_TYPES:
-            if not db.query(Progress).filter(Progress.state == TYPE).first():
-                db.add(Progress(state=TYPE))
+    if settings.PROGRESS_STATE:
+        for STATE in settings.PROGRESS_STATE:
+            if not db.query(Progress).filter(Progress.state == STATE).first():
+                db.add(Progress(state=STATE))
         db.commit()
     
     user = crud.user.get_by_email(db, email=settings.FIRST_SUPERUSER)
@@ -47,14 +47,3 @@ def init_db(db: Session) -> None:
         )
         user = crud.user.create(db, obj_in=user_in)  # noqa: F841
     
-    if settings.MEMBERSHIP_TYPES:
-        for TYPE in settings.MEMBERSHIP_TYPES:
-            if not db.query(Membership).filter(Membership.status == TYPE).first():
-                  db.add(Membership(status=TYPE))
-        db.commit()
-
-    if settings.PROGRESS_TYPES:
-        for TYPE in settings.PROGRESS_TYPES:
-            if not db.query(Progress).filter(Progress.state == TYPE).first():
-                db.add(Progress(state=TYPE))
-        db.commit()