conradlan 3 лет назад
Родитель
Сommit
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()