소스 검색

database load env

conradlan 3 년 전
부모
커밋
97a2dacbca
1개의 변경된 파일6개의 추가작업 그리고 3개의 파일을 삭제
  1. 6 3
      sql/database.py

+ 6 - 3
sql/database.py

@@ -1,11 +1,14 @@
+import os
 from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
-
-
+from dotenv import load_dotenv
+from os.path import join, dirname
 ## ENV LOAD
-SQLALCHEMY_DATABASE_URL = 'mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4'
+dotenv_path = dirname(__file__).replace("./sql",'env/database.env') 
+load_dotenv(dotenv_path)
 
+SQLALCHEMY_DATABASE_URL = os.environ.get("SQLALCHEMY_DATABASE_URL")
 engine = create_engine(SQLALCHEMY_DATABASE_URL)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base = declarative_base()