Update app.py

app.py

@@ -17,7 +17,106 @@ if not os.path.exists(UPLOAD_DIR):
 db_registry = []
 
 # --------------------------
-# 유틸
+# 유틸def display_data(standard, version, selection):
+    if not all([standard, version, selection]):
+        return pd.DataFrame({"Info": ["선택 필요"]})
+
+    try:
+        # 실시간 경로 확인을 위해 직접 경로 생성
+        db_path = os.path.join(UPLOAD_DIR, f"{standard}_{version}.db")
+        if not os.path.exists(db_path):
+            return pd.DataFrame({"Error": [f"파일을 찾을 수 없습니다: {db_path}"]})
+            
+        conn = sqlite3.connect(db_path)
+        conn.text_factory = safe_text_factory 
+
+        tables = pd.read_sql(
+            "SELECT name FROM sqlite_master WHERE type='table';",
+            conn
+        )['name'].tolist()
+
+        main_t = f"{standard}_{version}"
+        if main_t not in tables:
+            main_t = tables[0]
+
+        # --------------------------
+        # 🔹 테이블 명칭 매칭 (selection이 테이블 명칭인지 확인)
+        # --------------------------
+        full_table_name = None
+        pattern = re.compile(f"^{standard}[_\\s-]*{version}[_\\s-]*", re.IGNORECASE)
+
+        for t in tables:
+            if t == selection:
+                full_table_name = t
+                break
+            short = pattern.sub("", t).strip(" _")
+            if short == selection:
+                full_table_name = t
+                break
+
+        # ==========================================
+        # 🌟 부속 테이블 선택 시 (TableA, TableB_C 등)
+        # 원본 열 구조(4열, 5열 등)를 그대로 유지
+        # ==========================================
+        if full_table_name and full_table_name != main_t:
+            df = pd.read_sql(f"SELECT * FROM [{full_table_name}]", conn)
+            
+            # version 열 제외 및 컬럼명 정리
+            df = df.drop(columns=[c for c in df.columns if c.lower() == "version"], errors="ignore")
+            df.columns = [c.replace("_", " ").title() for c in df.columns]
+
+            # 모든 셀에 대해 이미지 렌더링(BLOB 변환) 적용
+            for col in df.columns:
+                df[col] = df[col].apply(blob_to_base64_html)
+
+            conn.close()
+            return df
+
+        # --------------------------
+        # 🔹 메인 테이블 (ALL 또는 특정 Category)
+        # 기존 규격(section, description) 유지
+        # --------------------------
+        cursor = conn.cursor()
+        cursor.execute(f"PRAGMA table_info([{main_t}])")
+        cols_info = cursor.fetchall()
+        cols = [c[1] for c in cols_info]
+        lower = [c.lower().strip() for c in cols]
+
+        real_ch, real_cat = None, None
+        for c, lc in zip(cols, lower):
+            if lc == "chapter": real_ch = c
+            elif lc == "category": real_cat = c
+
+        if selection == "ALL":
+            cursor.execute(f"SELECT * FROM [{main_t}]")
+        elif "." in selection and real_ch and real_cat:
+            ch, ca = selection.split(".", 1)
+            query = f"""
+            SELECT * FROM [{main_t}]
+            WHERE REPLACE(TRIM(CAST([{real_ch}] AS TEXT)), ' ', '') = ?
+            AND REPLACE(TRIM(CAST([{real_cat}] AS TEXT)), ' ', '') = ?
+            """
+            cursor.execute(query, (ch.replace(" ", ""), ca.replace(" ", "")))
+        else:
+            cursor.execute(f"SELECT * FROM [{main_t}]")
+
+        rows = cursor.fetchall()
+        df = pd.DataFrame(rows, columns=cols)
+
+        # 후처리: 메인 테이블은 가독성을 위해 2열(section, description)로 제한
+        df.columns = [c.lower().strip() for c in df.columns]
+        if 'section' in df.columns and 'description' in df.columns:
+            df = df[['section', 'description']]
+
+        for col in df.columns:
+            df[col] = df[col].apply(blob_to_base64_html)
+
+        conn.close()
+        return df
+
+    except Exception as e:
+        traceback.print_exc()
+        return pd.DataFrame({"Error": [str(e)]})
 # --------------------------
 def blob_to_base64_html(blob_data):
     if blob_data is None or pd.isna(blob_data):
@@ -204,9 +303,12 @@ def display_data(standard, version, selection):
         return pd.DataFrame({"Info": ["선택 필요"]})
 
     try:
-        target = next(d for d in db_registry if d["standard"] == standard and d["version"] == version)
-        conn = sqlite3.connect(target["path"])
-        
+        # 실시간 경로 확인을 위해 직접 경로 생성
+        db_path = os.path.join(UPLOAD_DIR, f"{standard}_{version}.db")
+        if not os.path.exists(db_path):
+            return pd.DataFrame({"Error": [f"파일을 찾을 수 없습니다: {db_path}"]})
+            
+        conn = sqlite3.connect(db_path)
         conn.text_factory = safe_text_factory 
 
         tables = pd.read_sql(
@@ -219,7 +321,7 @@ def display_data(standard, version, selection):
             main_t = tables[0]
 
         # --------------------------
-        # 🔹 테이블 reverse lookup
+        # 🔹 ���이블 명칭 매칭 (selection이 테이블 명칭인지 확인)
         # --------------------------
         full_table_name = None
         pattern = re.compile(f"^{standard}[_\\s-]*{version}[_\\s-]*", re.IGNORECASE)
@@ -234,51 +336,26 @@ def display_data(standard, version, selection):
                 break
 
         # ==========================================
-        # 🔹 TableA/B/C 선택 (특수 구조 테이블 처리 추가)
+        # 🌟 부속 테이블 선택 시 (TableA, TableB_C 등)
+        # 원본 열 구조(4열, 5열 등)를 그대로 유지
         # ==========================================
-        if full_table_name:
-            cursor = conn.cursor()
-            cursor.execute(f"SELECT * FROM [{full_table_name}]")
-            rows = cursor.fetchall()
-            cols = [desc[0] for desc in cursor.description]
-            df = pd.DataFrame(rows, columns=cols)
-            df.columns = [c.strip() for c in df.columns]
-            df = df.drop(columns=[c for c in df.columns if c.lower() == "version"], errors="ignore")
+        if full_table_name and full_table_name != main_t:
+            df = pd.read_sql(f"SELECT * FROM [{full_table_name}]", conn)
             
-            # 1. 이미지를 먼저 마크다운/HTML 형식으로 변환
+            # version 열 제외 및 컬럼명 정리
+            df = df.drop(columns=[c for c in df.columns if c.lower() == "version"], errors="ignore")
+            df.columns = [c.replace("_", " ").title() for c in df.columns]
+
+            # 모든 셀에 대해 이미지 렌더링(BLOB 변환) 적용
             for col in df.columns:
                 df[col] = df[col].apply(blob_to_base64_html)
 
-            orig_cols = df.columns.tolist()
-            lower_cols = [c.lower() for c in orig_cols]
-
-            # 2. Section/Description 동적 조립 로직
-            if 'section' in lower_cols and 'description' in lower_cols:
-                # 이미 규격에 맞는 경우 소문자로 이름만 맞춤
-                df.columns = lower_cols
-                df = df[['section', 'description']]
-            else:
-                # TableB_C 처럼 고유 컬럼 구조를 가진 경우의 동적 병합 처리
-                if len(orig_cols) >= 2:
-                    first_col = orig_cols[0] # 첫 번째 열 (예: Threat Code)가 section이 됨
-                    other_cols = orig_cols[1:] # 나머지 열들은 description으로 합쳐짐
-                    
-                    def build_combined_desc(row):
-                        parts = []
-                        for col in other_cols:
-                            val = "" if pd.isna(row[col]) else str(row[col])
-                            parts.append(f"**[{col}]**<br>{val}")
-                        return "<br><br>".join(parts)
-                    
-                    df['description'] = df.apply(build_combined_desc, axis=1)
-                    df = df.rename(columns={first_col: 'section'})
-                    df = df[['section', 'description']]
-
             conn.close()
             return df
 
         # --------------------------
-        # 🔹 메인 테이블 구조
+        # 🔹 메인 테이블 (ALL 또는 특정 Category)
+        # 기존 규격(section, description) 유지
         # --------------------------
         cursor = conn.cursor()
         cursor.execute(f"PRAGMA table_info([{main_t}])")
@@ -288,20 +365,11 @@ def display_data(standard, version, selection):
 
         real_ch, real_cat = None, None
         for c, lc in zip(cols, lower):
-            if lc == "chapter":
-                real_ch = c
-            elif lc == "category":
-                real_cat = c
+            if lc == "chapter": real_ch = c
+            elif lc == "category": real_cat = c
 
-        # --------------------------
-        # 🔹 ALL 선택
-        # --------------------------
         if selection == "ALL":
             cursor.execute(f"SELECT * FROM [{main_t}]")
-
-        # --------------------------
-        # 🔹 chapter.category
-        # --------------------------
         elif "." in selection and real_ch and real_cat:
             ch, ca = selection.split(".", 1)
             query = f"""
@@ -310,21 +378,14 @@ def display_data(standard, version, selection):
             AND REPLACE(TRIM(CAST([{real_cat}] AS TEXT)), ' ', '') = ?
             """
             cursor.execute(query, (ch.replace(" ", ""), ca.replace(" ", "")))
-
-        # --------------------------
-        # 🔹 fallback
-        # --------------------------
         else:
             cursor.execute(f"SELECT * FROM [{main_t}]")
 
         rows = cursor.fetchall()
         df = pd.DataFrame(rows, columns=cols)
 
-        # --------------------------
-        # 🔹 후처리
-        # --------------------------
+        # 후처리: 메인 테이블은 가독성을 위해 2열(section, description)로 제한
         df.columns = [c.lower().strip() for c in df.columns]
-
         if 'section' in df.columns and 'description' in df.columns:
             df = df[['section', 'description']]
 
@@ -335,7 +396,6 @@ def display_data(standard, version, selection):
         return df
 
     except Exception as e:
-        print(f"[Error in display_data]")
         traceback.print_exc()
         return pd.DataFrame({"Error": [str(e)]})