万朱浩 / Venue-Ops

Go to a project
Toggle navigation Toggle navigation pinning
Doiiars
53106ab3 1 parent 326d5b8c

MindSpider: DB_DIALECT兼容postgres写法，与其他模块保持一致

Inline Side-by-side
Showing 1 changed file with 2 additions and 2 deletions
@@ -73,7 +73,7 @@ class MindSpider: @@ -73,7 +73,7 @@ class MindSpider:
73 73
74 def build_async_url() -> str: 74 def build_async_url() -> str:
75 dialect = (settings.DB_DIALECT or "mysql").lower() 75 dialect = (settings.DB_DIALECT or "mysql").lower()
76 - if dialect == "postgresql": 76 + if dialect in ("postgresql", "postgres"):
77 return f"postgresql+asyncpg://{settings.DB_USER}:{quote_plus(settings.DB_PASSWORD)}@{settings.DB_HOST}:{settings.DB_PORT}/{settings.DB_NAME}" 77 return f"postgresql+asyncpg://{settings.DB_USER}:{quote_plus(settings.DB_PASSWORD)}@{settings.DB_HOST}:{settings.DB_PORT}/{settings.DB_NAME}"
78 # 默认使用 mysql 异步驱动 asyncmy 78 # 默认使用 mysql 异步驱动 asyncmy
79 return ( 79 return (
@@ -104,7 +104,7 @@ class MindSpider: @@ -104,7 +104,7 @@ class MindSpider:
104 104
105 def build_async_url() -> str: 105 def build_async_url() -> str:
106 dialect = (settings.DB_DIALECT or "mysql").lower() 106 dialect = (settings.DB_DIALECT or "mysql").lower()
107 - if dialect == "postgresql": 107 + if dialect in ("postgresql", "postgres"):
108 return f"postgresql+asyncpg://{settings.DB_USER}:{quote_plus(settings.DB_PASSWORD)}@{settings.DB_HOST}:{settings.DB_PORT}/{settings.DB_NAME}" 108 return f"postgresql+asyncpg://{settings.DB_USER}:{quote_plus(settings.DB_PASSWORD)}@{settings.DB_HOST}:{settings.DB_PORT}/{settings.DB_NAME}"
109 return ( 109 return (
110 f"mysql+asyncmy://{settings.DB_USER}:{quote_plus(settings.DB_PASSWORD)}" 110 f"mysql+asyncmy://{settings.DB_USER}:{quote_plus(settings.DB_PASSWORD)}"