if __name__ == "__main__":
from sql import build_ntsqlite
- build_ntsqlite()
+ build_ntsqlite(verbose=True)
NT_DB_NAME = "nt.sqlite"
-def build_ntsqlite():
+def build_ntsqlite(verbose=False):
# cd into this script's directory
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
os.chdir(SCRIPT_DIR)
- # print("Cleanup...")
+ if verbose:
+ print("Cleanup...")
if os.path.isfile(NT_DB_NAME):
os.remove(NT_DB_NAME)
- # print(f"\nPack {NT_DB_NAME}")
+ if verbose:
+ print(f"\nPack {NT_DB_NAME}")
con = sqlite3.connect(NT_DB_NAME)
cur = con.cursor()
- # print("\n-> Create tables")
+ if verbose:
+ print("\n-> Create tables")
with open("tables.sql") as tables:
cur.executescript(tables.read())
- # print("-> Populate data")
+ if verbose:
+ print("-> Populate data")
for p in os.listdir("data"):
if not p.endswith(".csv"):
continue
cur.close()
con.commit()
con.close()
- # print("\nDone!")
+ if verbose:
+ print("\nDone!")
return True