diff --git a/database.py b/database.py new file mode 100644 index 0000000..237f9a6 --- /dev/null +++ b/database.py @@ -0,0 +1,17 @@ +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker,declarative_base + + +tea_url ='mysql+pymysql://root:123456@127.0.0.1:3306/ai0824_stu?charset=utf8' + +engine = create_engine(tea_url,pool_size=100, echo=False) +Base=declarative_base() +Session = sessionmaker(bind=engine,autoflush=False,autocommit=False) + + +def get_db(): + db = Session() + try: + yield db + finally: + db.close() \ No newline at end of file