research/resources
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modify scripts

Browse Source
This commit is contained in:
sophon
2025-11-10 11:45:01 +08:00
parent 1582c140bf
commit 28124c1bff
2 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
scrapy_proj/my_sqlalchemy/migrations/resources/env.py
View File

@ -56,6 +56,15 @@ from sqlalchemy import create_engine
from models.resources import ResourceBase
target_metadata = ResourceBase.metadata # 关联 MySQL 模型的元数据
def get_mysql_url():
evn = os.environ.get('DB_ENV', 'dev')
if evn.lower() == 'dev':
return "mysql+pymysql://root:mysqlpw@testdb:3306/resources"
elif evn.lower() == 'nas':
return "mysql+pymysql://root:root@mariadb:3306/resources"
else:
return "mysql+pymysql://root:mysqlpw@testdb:3306/resources"
def run_migrations_online() -> None:
"""Run migrations in 'online' mode.
@ -67,7 +76,7 @@ def run_migrations_online() -> None:
#ini_section = config.get_section_option(config.config_ini_section, "ini_section", fallback="dev")
#url = config.get_section_option(ini_section, "sqlalchemy.url")
#connectable = create_engine(url)
url = "mysql+pymysql://root:mysqlpw@testdb:3306/resources"
url = get_mysql_url()
connectable = create_engine(url)
with connectable.connect() as connection: