> For the complete documentation index, see [llms.txt](https://close.gitbook.io/yun-wei-bi-ji/llms.txt). Markdown versions of documentation pages are available by appending `.md` to page URLs; this page is available as [Markdown](https://close.gitbook.io/yun-wei-bi-ji/python/fastapi/fastapi-gen-ju-huan-jing-bu-tong-lian-jie-bu-tong-shu-ju-ku.md).

# FastAPI 根据环境不同连接不同数据库

> 在实际的开发过程中，我们数据库，可以根据连接的环境不一样，我们会拆分成不一样的数据库，根据我们所要用的环境来选择对应的数据库即可，那么我们应该如何去实现根据选择去选择不一样的数据库呢。

* 首先，我们找一个配置的文件，我们这里演示 我们使用config.py去配置环境。配置如下

```python
EVENT="test"
```

* 接下来，我们去创建一个测试用的数据库配置，起名：testDatabse.py

```python
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
SQLALCHEMY_DATABASE_URL = "sqlite:///./test1.db"

engine = create_engine(
    SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
)
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base =   declarative_base()
```

* 然后我们修改models.py。如下

```python
from  config import EVENT
if EVENT=="test":
    from models.testDatabase import Base,engine

else:
    from  models.database import Base,engine
 ....数据库设计
 Base.metadata.create_all(bind=engine)
```

* 接下来我们去根据不同环境去连接不一样的数据库即可。

```python
from models.database import *
from models.testDatabase import TestingSessionLocal
from config import EVENT
def get_test_db():
    db=TestingSessionLocal()
    try:
        yield db
    finally:
        db.close()

def get_db_pro():
    """
    每一个请求处理完毕后会关闭当前连接，不同的请求使用不同的连接
    :return:
    """
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()
if EVENT =="test":
    get_db=get_test_db
else:
    get_db=get_db_pro
```

* 最后在连接数据库的地方，我们把get\_db替换成从

```python
from get_db import get_db
```