owenkaplinsky commited on
Commit
f95198a
Β·
verified Β·
1 Parent(s): 06526b6

Update src/database/candidates/init_db.py

Browse files
Files changed (1) hide show
  1. src/database/candidates/init_db.py +15 -4
src/database/candidates/init_db.py CHANGED
@@ -11,7 +11,7 @@ Usage:
11
 
12
  from src.database.candidates.client import engine
13
  from src.database.candidates.models import Base
14
-
15
 
16
  def init_db():
17
  """
@@ -19,13 +19,24 @@ def init_db():
19
  Intended for dev setup / Docker initialization.
20
  """
21
  try:
 
22
  Base.metadata.create_all(bind=engine)
23
- print("βœ… Database initialized successfully.")
 
 
 
 
 
 
 
 
 
 
 
24
  except Exception as e:
25
  print(f"❌ Failed to initialize database: {e}")
26
  raise
27
 
28
 
29
  if __name__ == "__main__":
30
- init_db()
31
-
 
11
 
12
  from src.database.candidates.client import engine
13
  from src.database.candidates.models import Base
14
+ from sqlalchemy import inspect
15
 
16
  def init_db():
17
  """
 
19
  Intended for dev setup / Docker initialization.
20
  """
21
  try:
22
+ print("πŸš€ Starting database initialization...")
23
  Base.metadata.create_all(bind=engine)
24
+
25
+ # Verify tables
26
+ inspector = inspect(engine)
27
+ tables = inspector.get_table_names()
28
+ print(f"πŸ“Š Found tables: {tables}")
29
+
30
+ if "candidates" in tables:
31
+ print("βœ… Database initialized successfully.")
32
+ return True
33
+ else:
34
+ print("❌ Error: 'candidates' table was not created!")
35
+
36
  except Exception as e:
37
  print(f"❌ Failed to initialize database: {e}")
38
  raise
39
 
40
 
41
  if __name__ == "__main__":
42
+ init_db()