U
    \jA                     @   sz   d Z ddlmZ ddlmZmZ ddlmZ ej	dr@ddini Z
eeje
d	Zedded
Ze Zdd Zdd ZdS )u   数据库连接与初始化。    )create_engine)sessionmakerdeclarative_base   )settingsZsqliteZcheck_same_threadF)connect_args)Z
autocommitZ	autoflushbindc                  c   s    t  } z
| V  W 5 |   X dS )u:   FastAPI 依赖：每请求一个 session，用完关闭。N)SessionLocalclose)db r   "/root/pica-gateway/app/database.pyget_db   s    
r   c                  C   s   ddl m}  tjjtd dS )u	   建表。r   models)r   N) r   BasemetadataZ
create_allenginer   r   r   r   init_db   s    r   N)__doc__Z
sqlalchemyr   Zsqlalchemy.ormr   r   configr   Zdatabase_url
startswithr   r   r	   r   r   r   r   r   r   r   <module>   s   

	