Update app.py

app.py

@@ -11,9 +11,32 @@ auth = HTTPBasicAuth()
 
 # Hàm kết nối với SQLite
 def get_db_connection():
-    conn = sqlite3.connect('/app/database.db')
-    conn.row_factory = sqlite3.Row  # Để trả về dữ liệu dưới dạng dictionary
-    return conn
+    db_path = '/app/database.db'
+    db_dir = os.path.dirname(db_path)
+
+    # Tạo thư mục /app/ nếu chưa tồn tại
+    if not os.path.exists(db_dir):
+        try:
+            os.makedirs(db_dir, exist_ok=True)
+        except Exception as e:
+            raise RuntimeError(f"Không thể tạo thư mục {db_dir}: {str(e)}")
+
+    # Kiểm tra quyền ghi
+    if not os.access(db_dir, os.W_OK):
+        # Nếu không có quyền ghi, thử sử dụng /tmp/
+        db_path = '/tmp/database.db'
+        db_dir = os.path.dirname(db_path)
+        if not os.path.exists(db_dir):
+            os.makedirs(db_dir, exist_ok=True)
+        if not os.access(db_dir, os.W_OK):
+            raise RuntimeError(f"Không có quyền ghi tại {db_dir} và /tmp/")
+
+    try:
+        conn = sqlite3.connect(db_path)
+        conn.row_factory = sqlite3.Row  # Để trả về dữ liệu dưới dạng dictionary
+        return conn
+    except sqlite3.OperationalError as e:
+        raise RuntimeError(f"Không thể mở file cơ sở dữ liệu tại {db_path}: {str(e)}")
 
 # Tạo các bảng nếu chưa tồn tại
 def init_db():