Merge pull request #206 from FunD-StockProject/fix/stock-master-info

Fix 종목 마스터 데이터 업데이트 오류 수정2

Author: MuuiGong

scripts/stocks_info/overseas_stock_code.py

@@ -140,9 +140,16 @@ def get_overseas_master_dataframe(base_dir,val):
         df = pd.read_table(cod_file_path, sep='\t', encoding='cp949')
         df.columns = columns
 
-        # 엑셀 파일 저장 (선택사항)
-        excel_path = os.path.join(base_dir, f'{val}_code.xlsx')
-        df.to_excel(excel_path, index=False)
+        # 엑셀 파일 저장 (선택사항) - openpyxl이 없으면 스킵
+        try:
+            excel_path = os.path.join(base_dir, f'{val}_code.xlsx')
+            df.to_excel(excel_path, index=False)
+        except ImportError:
+            # openpyxl이 없으면 엑셀 저장 스킵
+            pass
+        except Exception as e:
+            # 기타 에러는 무시
+            print(f"Warning: Could not save Excel file: {e}")
 
         return df
 

src/main/java/com/fund/stockProject/stock/service/StockImportService.java

@@ -153,17 +153,40 @@ public void importStocksFromJson(String jsonFilePath) {
                     }
                 }
 
-                // 새 종목 저장
+                // 새 종목 저장 - 개별 처리로 중복 ID 방지
                 if (!newStocks.isEmpty()) {
-                    try {
-                        entityManager.flush();
-                        stockRepository.saveAll(newStocks);
-                        entityManager.flush();
-                        log.info("Saved {} new stocks", newStocks.size());
-                    } catch (Exception e) {
-                        log.error("Error saving new stocks: {}", e.getMessage());
-                        throw e;
+                    int savedCount = 0;
+                    int skippedCount = 0;
+                    for (Stock stock : newStocks) {
+                        try {
+                            // symbol로 이미 존재하는지 다시 확인 (DB에서 직접 확인)
+                            Optional<Stock> existingStock = stockRepository.findBySymbol(stock.getSymbol());
+                            if (existingStock.isPresent()) {
+                                // 이미 존재하면 업데이트로 처리
+                                Stock existing = existingStock.get();
+                                existing.updateSymbolNameIfNull(stock.getSymbolName());
+                                existing.setValid(true);
+                                if (stock.getDomesticSector() != null) {
+                                    existing.setDomesticSector(stock.getDomesticSector());
+                                }
+                                if (stock.getOverseasSector() != null) {
+                                    existing.setOverseasSector(stock.getOverseasSector());
+                                }
+                                stockRepository.save(existing);
+                                skippedCount++;
+                                log.debug("Stock already exists, updated: {}", stock.getSymbol());
+                            } else {
+                                // 새 종목 저장
+                                stockRepository.save(stock);
+                                savedCount++;
+                            }
+                        } catch (Exception e) {
+                            log.warn("Failed to save stock: {} - {}", stock.getSymbol(), e.getMessage());
+                            skippedCount++;
+                        }
                     }
+                    entityManager.flush();
+                    log.info("Saved {} new stocks, {} skipped (already exists or error)", savedCount, skippedCount);
                 }
 
                 // 기존 종목 업데이트 - 개별 처리로 안전하게