QIDNLF commited on
Commit
08bf22a
·
verified ·
1 Parent(s): ed0e29b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -12
app.py CHANGED
@@ -124,12 +124,9 @@ def update_comp_standard_dropdown(base_std, base_ver):
124
 
125
  try:
126
  conn = sqlite3.connect(os.path.join(UPLOAD_DIR, "mapping.db"))
127
- query = """
128
- SELECT DISTINCT Comp_std FROM Mapping_registry WHERE Base_std=? AND Base_ver=?
129
- UNION
130
- SELECT DISTINCT Base_std FROM Mapping_registry WHERE Comp_std=? AND Comp_ver=?
131
- """
132
- df = pd.read_sql(query, conn, params=[base_std, base_ver, base_std, base_ver])
133
  conn.close()
134
 
135
  mapped_stds = sorted(df['Comp_std'].dropna().unique().tolist()) if not df.empty else []
@@ -137,18 +134,16 @@ def update_comp_standard_dropdown(base_std, base_ver):
137
  except Exception:
138
  return gr.update(choices=[], value=None, interactive=False)
139
 
 
140
  def update_comp_version_dropdown(base_std, base_ver, comp_std):
141
  if not all([base_std, base_ver, comp_std]):
142
  return gr.update(choices=[], value=None, interactive=False)
143
 
144
  try:
145
  conn = sqlite3.connect(os.path.join(UPLOAD_DIR, "mapping.db"))
146
- query = """
147
- SELECT DISTINCT Comp_ver FROM Mapping_registry WHERE Base_std=? AND Base_ver=? AND Comp_std=?
148
- UNION
149
- SELECT DISTINCT Base_ver FROM Mapping_registry WHERE Comp_std=? AND Comp_ver=? AND Base_std=?
150
- """
151
- df = pd.read_sql(query, conn, params=[base_std, base_ver, comp_std, base_std, base_ver, comp_std])
152
  conn.close()
153
 
154
  mapped_vers = sorted(df['Comp_ver'].dropna().unique().tolist()) if not df.empty else []
 
124
 
125
  try:
126
  conn = sqlite3.connect(os.path.join(UPLOAD_DIR, "mapping.db"))
127
+ # [수정] 양방향(UNION) 쿼리 제거, 철저하게 등록된 정방향(Base -> Comp)만 허용
128
+ query = "SELECT DISTINCT Comp_std FROM Mapping_registry WHERE Base_std=? AND Base_ver=?"
129
+ df = pd.read_sql(query, conn, params=[base_std, base_ver])
 
 
 
130
  conn.close()
131
 
132
  mapped_stds = sorted(df['Comp_std'].dropna().unique().tolist()) if not df.empty else []
 
134
  except Exception:
135
  return gr.update(choices=[], value=None, interactive=False)
136
 
137
+
138
  def update_comp_version_dropdown(base_std, base_ver, comp_std):
139
  if not all([base_std, base_ver, comp_std]):
140
  return gr.update(choices=[], value=None, interactive=False)
141
 
142
  try:
143
  conn = sqlite3.connect(os.path.join(UPLOAD_DIR, "mapping.db"))
144
+ # [수정] 양방향(UNION) 쿼리 제거
145
+ query = "SELECT DISTINCT Comp_ver FROM Mapping_registry WHERE Base_std=? AND Base_ver=? AND Comp_std=?"
146
+ df = pd.read_sql(query, conn, params=[base_std, base_ver, comp_std])
 
 
 
147
  conn.close()
148
 
149
  mapped_vers = sorted(df['Comp_ver'].dropna().unique().tolist()) if not df.empty else []