From 3828f849a4daca1d9ee484638e0ded7430bbe613 Mon Sep 17 00:00:00 2001 From: Damien De Paoli Date: Tue, 29 Apr 2025 15:15:37 +1000 Subject: [PATCH] allow the db to persist in /srv/docker/container, if we are in PROD --- db.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/db.py b/db.py index 30e5228..3531aaa 100644 --- a/db.py +++ b/db.py @@ -1,9 +1,13 @@ # db.py import sqlite3 import pprint +import os def connect_db(as_object): - conn = sqlite3.connect('finance.db') + if 'ENV' in os.environ and os.environ['ENV'] == "production": + conn = sqlite3.connect('/finance.db') + else: + conn = sqlite3.connect('./finance.db') if as_object: conn.row_factory = sqlite3.Row # This allows us to access columns by name return conn