Skip to content

Commit e5b9e13

Browse files
committed
Add database name param
1 parent 83b0e8d commit e5b9e13

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

tfinance/config.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,6 @@
77

88

99
BOT_TOKEN = os.getenv("BOT_TOKEN")
10+
DATABASE_NAME = os.getenv("DATABASE_NAME")
11+
1012
TIMEZONE = pytz.timezone("Europe/Moscow")

tfinance/database.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import sqlite3
22

33
from models import User
4+
from tfinance.config import DATABASE_NAME
45

56

67
def singleton(cls):
@@ -22,7 +23,7 @@ class Database:
2223

2324
def __init__(self):
2425
# Подключение к БД с отключенной проверкой потока.
25-
self.con = sqlite3.connect("data.db", check_same_thread=False)
26+
self.con = sqlite3.connect(f"{DATABASE_NAME}.db", check_same_thread=False)
2627
self.cur = self.con.cursor()
2728
self.setup()
2829

0 commit comments

Comments
 (0)