feat: 환산계수 모달 구현 전 백업

backend/db/dbsetup.py

@@ -2,6 +2,8 @@
 PIT3000 Database Setup
 SQLAlchemy 기반 데이터베이스 연결 및 스키마 정의
 Windows/Linux 크로스 플랫폼 지원
+
+PostgreSQL 지원 추가: 건조시럽 환산계수 조회 (drysyrup 테이블)
 """
 
 from sqlalchemy import create_engine, MetaData, text
@@ -87,6 +89,9 @@ class DatabaseConfig:
     
     # URL 인코딩된 드라이버
     DRIVER_ENCODED = urllib.parse.quote_plus(DRIVER)
+    
+    # PostgreSQL 연결 정보 (건조시럽 환산계수 DB)
+    POSTGRES_URL = "postgresql+psycopg2://admin:trajet6640@192.168.0.39:5432/label10"
 
     # 데이터베이스별 연결 문자열 (동적 드라이버 사용)
     @classmethod
@@ -135,6 +140,10 @@ class DatabaseManager:
         # SQLite 연결 추가
         self.sqlite_conn = None
         self.sqlite_db_path = Path(__file__).parent / 'mileage.db'
+        
+        # PostgreSQL 연결 (건조시럽 환산계수)
+        self.postgres_engine = None
+        self.postgres_session = None
 
     def get_engine(self, database='PM_BASE'):
         """특정 데이터베이스 엔진 반환"""
@@ -220,6 +229,127 @@ class DatabaseManager:
         # 새 세션 생성
         return self.get_session(database)
 
+    # ─────────────────────────────────────────────────────────────
+    # PostgreSQL 연결 (건조시럽 환산계수)
+    # ─────────────────────────────────────────────────────────────
+    def get_postgres_engine(self):
+        """
+        PostgreSQL 엔진 반환 (건조시럽 환산계수 DB)
+        
+        Returns:
+            Engine 또는 None (연결 실패 시)
+        """
+        if self.postgres_engine is not None:
+            return self.postgres_engine
+        
+        try:
+            self.postgres_engine = create_engine(
+                DatabaseConfig.POSTGRES_URL,
+                pool_size=5,
+                max_overflow=5,
+                pool_timeout=30,
+                pool_recycle=1800,
+                pool_pre_ping=True,
+                echo=False
+            )
+            # 연결 테스트
+            with self.postgres_engine.connect() as conn:
+                conn.execute(text("SELECT 1"))
+            print("[DB Manager] PostgreSQL 연결 성공")
+            return self.postgres_engine
+        except Exception as e:
+            print(f"[DB Manager] PostgreSQL 연결 실패 (무시됨): {e}")
+            self.postgres_engine = None
+            return None
+    
+    def get_postgres_session(self):
+        """
+        PostgreSQL 세션 반환 (건조시럽 환산계수 조회용)
+        
+        Returns:
+            Session 또는 None (연결 실패 시)
+        """
+        engine = self.get_postgres_engine()
+        if engine is None:
+            return None
+        
+        if self.postgres_session is None:
+            try:
+                Session = sessionmaker(bind=engine)
+                self.postgres_session = Session()
+            except Exception as e:
+                print(f"[DB Manager] PostgreSQL 세션 생성 실패: {e}")
+                return None
+        else:
+            # 세션 상태 체크
+            try:
+                self.postgres_session.execute(text("SELECT 1"))
+            except Exception as e:
+                print(f"[DB Manager] PostgreSQL 세션 복구 시도: {e}")
+                try:
+                    self.postgres_session.rollback()
+                except:
+                    pass
+                try:
+                    self.postgres_session.close()
+                except:
+                    pass
+                try:
+                    Session = sessionmaker(bind=engine)
+                    self.postgres_session = Session()
+                except:
+                    self.postgres_session = None
+                    return None
+        
+        return self.postgres_session
+    
+    def get_conversion_factor(self, sung_code):
+        """
+        건조시럽 환산계수 조회
+        
+        Args:
+            sung_code: SUNG_CODE (예: "535000ASY")
+            
+        Returns:
+            dict: {
+                'conversion_factor': float 또는 None,
+                'ingredient_name': str 또는 None,
+                'product_name': str 또는 None
+            }
+        """
+        result = {
+            'conversion_factor': None,
+            'ingredient_name': None,
+            'product_name': None
+        }
+        
+        session = self.get_postgres_session()
+        if session is None:
+            return result
+        
+        try:
+            query = text("""
+                SELECT conversion_factor, ingredient_name, product_name
+                FROM drysyrup
+                WHERE ingredient_code = :sung_code
+                LIMIT 1
+            """)
+            row = session.execute(query, {'sung_code': sung_code}).fetchone()
+            
+            if row:
+                result['conversion_factor'] = float(row[0]) if row[0] is not None else None
+                result['ingredient_name'] = row[1]
+                result['product_name'] = row[2]
+        except Exception as e:
+            print(f"[DB Manager] 환산계수 조회 실패 (SUNG_CODE={sung_code}): {e}")
+            # 세션 롤백
+            try:
+                session.rollback()
+            except:
+                pass
+        
+        return result
+
     def get_sqlite_connection(self, new_connection=False):
         """
         SQLite mileage.db 연결 반환
@@ -442,6 +572,20 @@ class DatabaseManager:
         if self.sqlite_conn:
             self.sqlite_conn.close()
             self.sqlite_conn = None
+        
+        # PostgreSQL 연결 종료
+        if self.postgres_session:
+            try:
+                self.postgres_session.close()
+            except:
+                pass
+            self.postgres_session = None
+        if self.postgres_engine:
+            try:
+                self.postgres_engine.dispose()
+            except:
+                pass
+            self.postgres_engine = None
 
 # 전역 데이터베이스 매니저 인스턴스
 db_manager = DatabaseManager()