Revize 07a6c869
Přidáno uživatelem David Friesecký před téměř 4 roky(ů)
src/db/setup_database.py | ||
---|---|---|
1 |
import os |
|
1 | 2 |
import sqlite3 |
3 |
from pathlib import Path |
|
2 | 4 |
|
3 |
from src.constants import DATABASE_FILE_LOCATION |
|
5 |
from src.constants import DATABASE_FILE_LOCATION, DB_DIR
|
|
4 | 6 |
from src.db.init_queries import SCHEMA_SQL, DEFAULT_VALUES_SQL |
5 | 7 |
from src.utils.logger import Logger |
6 | 8 |
|
... | ... | |
9 | 11 |
Logger.debug(f"Database is setting.") |
10 | 12 |
|
11 | 13 |
try: |
14 |
# TODO improve creating DB dir |
|
15 |
if not os.path.exists(DB_DIR): |
|
16 |
os.makedirs("../../" + DB_DIR) |
|
17 |
|
|
18 |
# TODO improve creating DB file (now via FileAnchor) |
|
12 | 19 |
co = sqlite3.connect(DATABASE_FILE_LOCATION.shortest_relative_path()) |
13 | 20 |
cu = co.cursor() |
14 | 21 |
cu.executescript(SCHEMA_SQL) |
... | ... | |
21 | 28 |
|
22 | 29 |
|
23 | 30 |
if __name__ == '__main__': |
24 |
setup_database() |
|
31 |
setup_database() |
Také k dispozici: Unified diff
Re #8696 - Improved DB creation
- secured start application without lost data