From 013a81f454ff7373223eb002b8c54ebea9340a91 Mon Sep 17 00:00:00 2001
From: Seunghoon Shin <liante0904@gmail.com>
Date: Sat, 21 Sep 2024 16:02:57 +0900
Subject: [PATCH] init FirmInfo Class

---
 package/firm_info.py | 59 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/package/firm_info.py b/package/firm_info.py
index 91691434..c8e06110 100644
--- a/package/firm_info.py
+++ b/package/firm_info.py
@@ -232,3 +232,62 @@ def get_firm_info(sec_firm_order, article_board_order):
         "board_name": board_name,
         "label_name": label_name
     }
+
+
+
+class FirmInfo:
+    def __init__(self):
+        self.SEC_FIRM_ORDER = 0
+        self.ARTICLE_BOARD_ORDER = 0
+    
+    def get_firm_name(self):
+        if 0 <= self.SEC_FIRM_ORDER < len(firm_names):
+            return firm_names[self.SEC_FIRM_ORDER]
+        else:
+            return ""
+    
+    def get_board_name(self):
+        if 0 <= self.SEC_FIRM_ORDER < len(board_names) and 0 <= self.ARTICLE_BOARD_ORDER < len(board_names[self.SEC_FIRM_ORDER]):
+            return board_names[self.SEC_FIRM_ORDER][self.ARTICLE_BOARD_ORDER]
+        else:
+            return ""
+    
+    def get_label_name(self):
+        if 0 <= self.SEC_FIRM_ORDER < len(label_names) and 0 <= self.ARTICLE_BOARD_ORDER < len(label_names[self.SEC_FIRM_ORDER]):
+            return label_names[self.SEC_FIRM_ORDER][self.ARTICLE_BOARD_ORDER]
+        else:
+            return ""
+    
+    def set_sec_firm_order(self, sec_firm_order):
+        self.SEC_FIRM_ORDER = sec_firm_order
+    
+    def set_article_board_order(self, article_board_order):
+        self.ARTICLE_BOARD_ORDER = article_board_order
+        
+    def get_state(self):
+        return {
+            "SEC_FIRM_ORDER": self.SEC_FIRM_ORDER,
+            "ARTICLE_BOARD_ORDER": self.ARTICLE_BOARD_ORDER,
+            "FIRM_NAME":firm_names[self.SEC_FIRM_ORDER],
+            "BOARD_NAME":board_names[self.SEC_FIRM_ORDER][self.ARTICLE_BOARD_ORDER],
+            "LABEL_NAME":label_names[self.SEC_FIRM_ORDER][self.ARTICLE_BOARD_ORDER]
+        }
+# 테스트 코드
+if __name__ == "__main__":
+    firm_info = FirmInfo()
+    print(firm_info.get_state())
+    # 증권사 이름 출력
+    print(firm_info.get_firm_name())  # LS증권
+    
+    # 게시판 이름 출력
+    print(firm_info.get_board_name())  # 이슈브리프
+    
+    # 라벨 이름 출력
+    print(firm_info.get_label_name())  # ""
+    
+    # 상태값 변경 후 출력
+    firm_info.set_sec_firm_order(1)
+    firm_info.set_article_board_order(2)
+    print(firm_info.get_firm_name())  # 신한증권
+    print(firm_info.get_board_name())  # 스몰캡
+    print(firm_info.get_label_name())  # ""
\ No newline at end of file