some refactors

This commit is contained in:
collerek
2020-09-17 15:21:11 +02:00
parent 1cfd061e17
commit 12487e1089
5 changed files with 10 additions and 22 deletions

View File

@ -2,10 +2,7 @@ import os
import databases
assert "DATABASE_URL" in os.environ, "DATABASE_URL is not set."
DATABASE_URL = os.environ['DATABASE_URL']
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///test.db")
database_url = databases.DatabaseURL(DATABASE_URL)
if database_url.scheme == "postgresql+aiopg": # pragma no cover
DATABASE_URL = str(database_url.replace(driver=None))
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///test.db")