From: gamesguru <mathmuncher11@gmail.com>
Date: Sat, 13 Mar 2021 16:45:48 +0000 (-0500)
Subject: use NT_DB_NAME
X-Git-Tag: 0.0.1~7
X-Git-Url: https://git.nutra.tk/v2?a=commitdiff_plain;h=886680678a5a23f90a495880e5a6d29b1f04fe41;p=nutratech%2Fnt-sqlite.git

use NT_DB_NAME
---

diff --git a/sql/__init__.py b/sql/__init__.py
index 1ef8160..7b435b3 100644
--- a/sql/__init__.py
+++ b/sql/__init__.py
@@ -3,6 +3,8 @@ import csv
 import os
 import sqlite3
 
+NT_DB_NAME = "nt.sqlite"
+
 
 def build_ntsqlite():
     # cd into this script's directory
@@ -10,11 +12,11 @@ def build_ntsqlite():
     os.chdir(SCRIPT_DIR)
 
     print("Cleanup...")
-    if os.path.isfile("nt.sqlite"):
-        os.remove("nt.sqlite")
+    if os.path.isfile(NT_DB_NAME):
+        os.remove(NT_DB_NAME)
 
-    print("\nPack nt.sqlite")
-    con = sqlite3.connect("nt.sqlite")
+    print(f"\nPack {NT_DB_NAME}")
+    con = sqlite3.connect(NT_DB_NAME)
     cur = con.cursor()
 
     print("\n-> Create tables")