Skip to content

Commit 3a0cd26

Browse files
committed
Update database configuration
1 parent c2258a5 commit 3a0cd26

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

app/database/database.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,14 +12,14 @@
1212

1313

1414
class SQLAlchemyClient():
15-
db_url = engine.URL.create(
15+
db_url = environ.get("HEROKU_DATABASE_URL", engine.URL.create(
1616
"postgresql",
1717
database=environ["MEASUREMENTS_DB"],
1818
username=environ["POSTGRES_USER"],
1919
password=environ["POSTGRES_PASSWORD"],
2020
host=environ["POSTGRES_HOST"],
2121
port=environ["POSTGRES_PORT"]
22-
)
22+
))
2323

2424
engine = create_engine(db_url)
2525

app/service/rabbitmq/consumer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,14 +23,14 @@
2323
logger = logging.getLogger("rabbitmq_consumer")
2424
logging.getLogger("pika").setLevel(logging.WARNING)
2525

26-
dbUrl = engine.URL.create(
26+
dbUrl = environ.get("HEROKU_DATABASE_URL", engine.URL.create(
2727
"postgresql",
2828
database=environ["MEASUREMENTS_DB"],
2929
username=environ["POSTGRES_USER"],
3030
password=environ["POSTGRES_PASSWORD"],
3131
host=environ["POSTGRES_HOST"],
3232
port=environ["POSTGRES_PORT"]
33-
)
33+
))
3434
engine = create_engine(dbUrl, echo=True, future=True)
3535
session = Session(engine)
3636

0 commit comments

Comments
 (0)