apsniper0673-grsheetmanager 0.1.4__tar.gz

apsniper0673_grsheetmanager-0.1.4/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 sniper0673
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

apsniper0673_grsheetmanager-0.1.4/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: apsniper0673_grsheetmanager
+Version: 0.1.4
+Summary: My grsheetmanager for APSniper0673
+Author: apsniper0673
+Requires-Python: >=3.8
+License-File: LICENSE
+Dynamic: license-file

apsniper0673_grsheetmanager-0.1.4/README.md

@@ -0,0 +1 @@
+# constants

apsniper0673_grsheetmanager-0.1.4/pyproject.toml

@@ -0,0 +1,19 @@
+[project]
+name = "apsniper0673_grsheetmanager"
+version = "0.1.4"
+description = "My grsheetmanager for APSniper0673"
+authors = [
+  { name = "apsniper0673" }
+]
+requires-python = ">=3.8"
+dependencies = [
+    # "pandas",
+    # "numpy",
+]
+
+[build-system]
+requires = ["setuptools>=64", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+where = ["src"]

apsniper0673_grsheetmanager-0.1.4/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

apsniper0673_grsheetmanager-0.1.4/src/apsniper0673_grsheetmanager.egg-info/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: apsniper0673_grsheetmanager
+Version: 0.1.4
+Summary: My grsheetmanager for APSniper0673
+Author: apsniper0673
+Requires-Python: >=3.8
+License-File: LICENSE
+Dynamic: license-file

apsniper0673_grsheetmanager-0.1.4/src/apsniper0673_grsheetmanager.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+README.md
+pyproject.toml
+src/apsniper0673_grsheetmanager.egg-info/PKG-INFO
+src/apsniper0673_grsheetmanager.egg-info/SOURCES.txt
+src/apsniper0673_grsheetmanager.egg-info/dependency_links.txt
+src/apsniper0673_grsheetmanager.egg-info/top_level.txt
+src/mygrsheetmanager/__init__.py
+src/mygrsheetmanager/df_wish_list.py
+src/mygrsheetmanager/gr_sheet_manager.py
+src/mygrsheetmanager/gr_sheet_manager_client.py
+src/mygrsheetmanager/gr_sheet_manager_server.py
+src/mygrsheetmanager/single_trader_wish_list.py

apsniper0673_grsheetmanager-0.1.4/src/apsniper0673_grsheetmanager.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

apsniper0673_grsheetmanager-0.1.4/src/apsniper0673_grsheetmanager.egg-info/top_level.txt

@@ -0,0 +1 @@
+mygrsheetmanager

apsniper0673_grsheetmanager-0.1.4/src/mygrsheetmanager/df_wish_list.py

@@ -0,0 +1,79 @@
+"""
+여러 트레이더들의 wish list를 관리하는 DataFrame을 생성하는 모듈입니다.
+"""
+
+import pandas as pd
+from mygrsheetmanager.single_trader_wish_list import SingleDfWishList
+
+class _DfWithWishList:
+    def __init__(self, trader_names: list):
+        """
+        트레이더 이름 리스트를 받아서 DataFrame을 초기화합니다.
+        :param trader_names: 트레이더 이름의 리스트
+        """
+        self.trader_names = trader_names
+        # self.df_wish_list = pd.DataFrame(index=trader_names)  # index: 트레이더 이름, columns: 종목명, values: 매수 희망 여부
+        self.wish_list_dict = {name: SingleDfWishList(trader_name=name) for name in trader_names}
+    @property
+    def df(self) -> pd.DataFrame:
+        """
+        트레이더들의 매수 희망 종목 리스트를 DataFrame 형태로 반환합니다.
+        :return: DataFrame 
+            index: 트레이더 이름, columns: 종목명, values: 매수 희망 여부
+        """
+        # DataFrame을 생성할 때, 각 트레이더의 wish_list를 합칩니다.
+        dfs = []
+        for trader_name in self.trader_names:
+            s = self.wish_list_dict[trader_name].wish_list
+
+            # 비어 있는 경우라도 빈 DataFrame을 넣어줌
+            if s.empty:
+                dfs.append(pd.DataFrame(index=[trader_name]))  # 빈 row만 있는 DataFrame
+            else:
+                df = s.to_frame().T
+                df.index = [trader_name]
+                dfs.append(df)
+
+        df_merged = pd.concat(dfs, axis=0).fillna("")
+        df_merged.index.name = "trader"
+        return df_merged.reindex(index=self.trader_names).sort_index(axis=1)
+
+class DfWithWishList_manage_single(_DfWithWishList):
+    def add_wish(self, trader_name: str, stock_name: str, wish: str = "매수") -> None:
+        """
+        특정 트레이더의 매수 희망 종목을 추가합니다.
+        :param trader_name: 트레이더 이름
+        :param stock_name: 종목명
+        :param wish: 매수 희망 여부 (예: "매수", "관심")
+        """
+        if trader_name in self.wish_list_dict:
+            self.wish_list_dict[trader_name].add_wish(stock_name, wish)
+    def add_wishes(self, trader_name: str, stock_names: list, wish: str = "매수") -> None:
+        """
+        특정 트레이더의 매수 희망 종목을 일괄적으로 추가합니다.
+        :param trader_name: 트레이더 이름
+        :param stock_names: 종목명 리스트
+        :param wish: 매수 희망 여부 (예: "매수", "관심")
+        """
+        if trader_name in self.wish_list_dict:
+            for stock_name in stock_names:
+                self.wish_list_dict[trader_name].add_wish(stock_name, wish)
+    def remove_wish(self, trader_name: str, stock_name: str) -> None:
+        """
+        특정 트레이더의 매수 희망 종목을 제거합니다.
+        :param trader_name: 트레이더 이름
+        :param stock_name: 제거할 종목명
+        """
+        if trader_name in self.wish_list_dict:
+            self.wish_list_dict[trader_name].remove_wish(stock_name)
+    def replace_wishes(self, trader_name: str, stock_names: list, wish: str = "매수") -> None:
+        """
+        특정 트레이더의 매수 희망 종목을 일괄적으로 교체합니다.
+        :param trader_name: 트레이더 이름
+        :param stock_names: 종목명 리스트
+        :param wish: 매수 희망 여부 (예: "매수", "관심")
+        """
+        if trader_name in self.wish_list_dict:
+            self.wish_list_dict[trader_name].replace_wishes(stock_names, wish)
+class DfWithWishList(DfWithWishList_manage_single):
+    pass

apsniper0673_grsheetmanager-0.1.4/src/mygrsheetmanager/gr_sheet_manager.py

@@ -0,0 +1,109 @@
+"""
+트레이더 GR의 구글시트 매니저
+구현 기능
+1. 여러 GR 트레이더의 매수 희망 종목에 대해서 정리하여 관리
+    wish_list라는 sheet에 트레이더별로 종목을 정리하여 관리한다.
+    트레이더 클래스는 여러개 생성되어도, 이 클래스는 하나만 존재한다.
+"""
+from mygrsheetmanager.df_wish_list import DfWithWishList
+from mystockutil.spreadsheet.spreadsheet_manager import SpreadsheetManager
+from myutil.path import get_credentials_path
+
+CRED_DIR = get_credentials_path()
+CRED_JSON_PATH = CRED_DIR / 'vertical-album-400707-49b25aaf32d5.json'
+
+SHEET_ID = "1iK7UY7g_fSRSAOpIEYrFR41p_qK68LhkcIfqr7NUxug"
+# TRADER_NAMES = [
+#     '황금비0', '황금비1', '황금비2', 
+#     # '황금비3', '황금비4',
+#     '황금비5', '황금비6', 
+#     '황금비_test',
+# ]
+TRADER_NAMES = [
+    '뉴황금비0', '뉴황금비1', 
+    # '황금비2', 
+    # '황금비3', '황금비4',
+    # '황금비5', 
+    '황금비6', 
+    '황금비_test',
+]
+
+class _GRSheetManager:
+    """
+    self._df_wish_list: pd.DataFrame
+        index: 트레이더 이름
+        columns: 종목명
+        values: 매수 희망 여부로, "매수" 등등의 문자열로 표현된다.
+    """
+    def __init__(self):
+        self.sheet_manager = SpreadsheetManager(
+            creds_json_path=CRED_JSON_PATH,
+            sheet_id=SHEET_ID,
+        )
+        self.wish_sheet_name = 'wish_list'
+        self.sell_sheet_name = 'sell_list'
+        self.wish_list = DfWithWishList(trader_names=TRADER_NAMES[0:-1])
+        # 매도리스트도 동일한 클래스로 생성해 준다. 
+        self.sell_list = DfWithWishList(trader_names=TRADER_NAMES[0:-1])
+
+class GRSheetManager_posting(_GRSheetManager):
+    def post_wish_df(self) -> None:
+        """
+        wish_list 시트에 DataFrame을 업로드한다.
+        :param df: 업로드할 DataFrame
+        """
+        self.sheet_manager.upload_df(self.wish_list.df, worksheet_name=self.wish_sheet_name, index=True)
+    def post_sell_df(self) -> None:
+        """
+        sell_list 시트에 DataFrame을 업로드한다.
+        :param df: 업로드할 DataFrame
+        """
+        self.sheet_manager.upload_df(self.sell_list.df, worksheet_name=self.sell_sheet_name, index=True)
+        
+class GRSheetManager(GRSheetManager_posting):
+    pass
+
+"""테스트용 코드"""
+def test_add_single_wish():
+    mgr = GRSheetManager_posting()
+    mgr.wish_list.add_wish("황금비0", "삼성전자", "매수")
+    df = mgr.wish_list.df
+    assert df.loc["황금비0", "삼성전자"] == "매수"
+
+def test_multiple_wishes():
+    mgr = GRSheetManager_posting()
+    mgr.wish_list.add_wish("황금비1", "LG화학", "관심")
+    mgr.wish_list.add_wish("황금비1", "카카오", "매수")
+    df = mgr.wish_list.df
+    assert df.loc["황금비1", "LG화학"] == "관심"
+    assert df.loc["황금비1", "카카오"] == "매수"
+
+def test_replace_wishes():
+    mgr = GRSheetManager_posting()
+    mgr.wish_list.wish_list_dict["황금비2"].replace_wishes(["현대차", "기아"], wish="매수")
+    df = mgr.wish_list.df
+    assert df.loc["황금비2", "현대차"] == "매수"
+    assert df.loc["황금비2", "기아"] == "매수"
+
+def test_remove_wish():
+    mgr = GRSheetManager_posting()
+    mgr.wish_list.add_wish("황금비3", "셀트리온", "매수")
+    mgr.wish_list.wish_list_dict["황금비3"].remove_wish("셀트리온")
+    df = mgr.wish_list.df
+    assert "셀트리온" not in df.columns or df.loc["황금비3", "셀트리온"] == ""
+
+def run_test_case(title: str, func):
+    print(f"\n[TEST] {title}")
+    try:
+        func()
+        print("→ ✅ 성공")
+    except Exception as e:
+        print(f"→ ❌ 실패: {e}")
+
+if __name__ == "__main__":
+    run_test_case("단일 종목 추가", test_add_single_wish)
+    run_test_case("다수 종목 추가", test_multiple_wishes)
+    run_test_case("비어 있는 트레이더 포함", test_empty_wish_trader_included)
+    run_test_case("일괄 종목 교체", test_replace_wishes)
+    run_test_case("종목 삭제", test_remove_wish)
+    print("\n모든 테스트 완료")

apsniper0673_grsheetmanager-0.1.4/src/mygrsheetmanager/gr_sheet_manager_client.py

@@ -0,0 +1,161 @@
+"""
+GR 트레이더의 구글시트 매니저 – 클라이언트용
+"""
+import requests
+import pandas as pd
+
+# Essential imports
+from mystockutil.server.get_available_server import get_available_server
+
+# 로컬 호스트를 먼저 찾는다 > 데이터 절약하기 위해서
+server_candidates = [
+    'http://localhost:6019', # 백업서버
+    'http://localhost:6009',
+    'http://brstk2.iptime.org:6019',
+    'http://brstk2.iptime.org:6009',
+    'http://brstk.com:6019', # 백업서버
+    'http://brstk.com:6009',
+    'http://brstk.iptime.org:6009',
+    'http://brstk.iptime.org:6019',
+    ]
+
+# SERVER_URL = "http://localhost:6009"
+SERVER_URL = get_available_server(server_candidates=server_candidates)
+
+# 안전한 API 호출을 위한 데코레이터
+def safe_api_call(func):
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except Exception as e:
+            print(f"[API 오류] {func.__name__} 실패: {e}")
+            return None
+    return wrapper
+
+
+
+def add_wish(trader_name: str, stock_name: str, wish: str = "매수"):
+    url = f"{SERVER_URL}/wishlist/{trader_name}"
+    data = {
+        "mode": "add",
+        "stock_name": stock_name,
+        "wish": wish,
+    }
+    return requests.post(url, json=data).json()
+
+def replace_wishes(trader_name: str, stock_names: list, wish: str = "매수"):
+    url = f"{SERVER_URL}/wishlist/{trader_name}"
+    data = {
+        "mode": "replace",
+        "stock_names": stock_names,
+        "wish": wish,
+    }
+    return requests.post(url, json=data).json()
+
+def delete_wish(trader_name: str, stock_name: str):
+    url = f"{SERVER_URL}/wishlist/{trader_name}"
+    data = {
+        "mode": "delete",
+        "stock_name": stock_name,
+    }
+    return requests.post(url, json=data).json()
+
+def get_all_wishlist():
+    return requests.get(f"{SERVER_URL}/wishlist").json()
+
+def get_trader_wishlist(trader_name: str):
+    return requests.get(f"{SERVER_URL}/wishlist/{trader_name}").json()
+
+"""
+sell_list에 대해서 동일한 작업 수행
+"""
+def add_sell(trader_name: str, stock_name: str, wish: str = "매도"):
+    url = f"{SERVER_URL}/selllist/{trader_name}"
+    data = {
+        "mode": "add",
+        "stock_name": stock_name,
+        "wish": wish,
+    }
+    return requests.post(url, json=data).json()
+
+def replace_sells(trader_name: str, stock_names: list, wish: str = "매수"):
+    url = f"{SERVER_URL}/selllist/{trader_name}"
+    data = {
+        "mode": "replace",
+        "stock_names": stock_names,
+        "wish": wish,
+    }
+    return requests.post(url, json=data).json()
+
+def delete_sell(trader_name: str, stock_name: str):
+    url = f"{SERVER_URL}/selllist/{trader_name}"
+    data = {
+        "mode": "delete",
+        "stock_name": stock_name,
+    }
+    return requests.post(url, json=data).json()
+
+def get_all_selllist():
+    return requests.get(f"{SERVER_URL}/selllist").json()
+
+def get_trader_selllist(trader_name: str):
+    return requests.get(f"{SERVER_URL}/selllist/{trader_name}").json()
+
+class SingleTraderWishListClient:
+    """싱글 트레이더의 wish list를 관리하는 클라이언트 클래스"""
+
+    def __init__(self, trader_name: str):
+        self.trader_name = trader_name
+
+    @safe_api_call
+    def add_wish(self, stock_name: str, wish: str = "매수"):
+        return add_wish(self.trader_name, stock_name, wish)
+    @safe_api_call
+    def replace_wishes(self, stock_names: list, wish: str = "매수"):
+        return replace_wishes(self.trader_name, stock_names, wish)
+    @safe_api_call
+    def delete_wish(self, stock_name: str):
+        return delete_wish(self.trader_name, stock_name)
+    @safe_api_call
+    def get_wishlist(self):# -> Any:
+        data = get_trader_wishlist(self.trader_name)
+        df = pd.DataFrame.from_records(data['data'])
+        return df
+    @safe_api_call
+    def get_all_wishlist(self):
+        data = get_all_wishlist()
+        df = pd.DataFrame.from_records(data['data'])
+        return df
+    """
+    selllist 코드
+    """
+    @safe_api_call
+    def add_sell(self, stock_name: str, wish: str = "매도"):
+        return add_sell(self.trader_name, stock_name, wish)
+    @safe_api_call
+    def replace_sells(self, stock_names: list, wish: str = "매도"):
+        return replace_sells(self.trader_name, stock_names, wish)
+    @safe_api_call
+    def delete_sell(self, stock_name: str):
+        return delete_sell(self.trader_name, stock_name)
+    @safe_api_call
+    def get_selllist(self):# -> Any:
+        data = get_trader_selllist(self.trader_name)
+        df = pd.DataFrame.from_records(data['data'])
+        return df
+    @safe_api_call
+    def get_all_selllist(self):
+        data = get_all_selllist()
+        df = pd.DataFrame.from_records(data['data'])
+        return df
+
+
+if __name__ == "__main__":
+    # 예시 사용법
+    client = SingleTraderWishListClient("황금비1")
+    print(client.add_wish("삼성전자", "매수"))
+    print(client.get_wishlist())
+    print(client.replace_wishes(["삼성전자", "LG전자"], "추매"))
+    print(client.delete_wish("LG전자"))
+    print(client.replace_wishes(["카카오", "네이버"], "매도"))
+    print(client.get_wishlist())

apsniper0673_grsheetmanager-0.1.4/src/mygrsheetmanager/gr_sheet_manager_server.py

@@ -0,0 +1,142 @@
+"""
+GR Sheet Manager를 구동하는 서버 코드
+
+API 종류:
+------------
+GET    /wishlist
+       전체 wish list를 JSON(table) 형식으로 보내줍니다.
+POST   /wishlist/<trader_name>
+       특정 트레이더의 wish를 추가, 교체 또는 삭제합니다.
+       body 예시 (application/json):
+       {
+           "mode": "add",          # add | replace | delete
+           "stock_name": "삼성전자", # add/delete 모드에서 필요
+           "stock_names": ["삼성전자", "LG화학"], # replace 모드에서 필요
+           "wish": "머야"            # 기본값: "머야"
+       }
+
+실행 방식:
+------------
+$ python wishlist_server.py
+
+* 개발 동안은 Flask 내장 서버로 가능
+* 관리 환경은 Waitress 또는 Gunicorn + systemd 구조로 구성
+
+필요 패키지:
+------------
+flask, pandas, gspread, google-auth, SpreadsheetManager 관련 패키지
+"""
+
+from flask import Flask, jsonify, request
+from mygrsheetmanager.gr_sheet_manager import GRSheetManager
+
+app = Flask(__name__)
+
+grsheet = GRSheetManager()  # 싱글턴으로 사용
+
+def _push_wish_to_sheet() -> None:
+    """Google Sheet에 wish list를 복사합니다."""
+    try:
+        grsheet.post_wish_df()
+    except Exception as exc:
+        app.logger.exception("Google Sheet에 복사 실패")
+        raise exc
+
+@app.route("/wishlist", methods=["GET"])
+def get_wishlist():
+    """GR 트레이더 전체의 wish list를 JSON(table) 형식으로 보내줍니다."""
+    df = grsheet.wish_list.df
+    return df.to_json(orient="table", force_ascii=False), 200, {"Content-Type": "application/json; charset=utf-8"}
+
+@app.route("/wishlist/<trader_name>", methods=["GET"])
+def get_trader_wish(trader_name):
+    df = grsheet.wish_list.df
+    if trader_name not in df.index:
+        return jsonify({"error": f"{trader_name}의 wish list가 없습니다."}), 404
+    return df.loc[[trader_name]].to_json(orient="table", force_ascii=False)
+
+@app.route("/wishlist/<trader>", methods=["POST"])
+def upsert_wish(trader: str):
+    """GR 트레이더의 wish 추가 (add), 교체 (replace), 삭제 (delete)를 지원합니다."""
+    data = request.get_json(force=True)
+    mode: str = data.get("mode", "add")
+    wish: str = data.get("wish", "매수")
+
+    if mode == "add":
+        stock = data.get("stock_name")
+        if not stock:
+            return jsonify({"error": "add 모드에서는 stock_name이 필요합니다."}), 400
+        grsheet.wish_list.add_wish(trader, stock, wish)
+    elif mode == "replace":
+        # 빈 리스트도 허용
+        stocks = data.get("stock_names")
+        grsheet.wish_list.replace_wishes(trader, stocks, wish)
+    elif mode == "delete":
+        stock = data.get("stock_name")
+        if not stock:
+            return jsonify({"error": "delete 모드에서는 stock_name이 필요합니다."}), 400
+        grsheet.wish_list.remove_wish(trader, stock)
+    else:
+        return jsonify({"error": f"알 수 없는 mode: {mode}"}), 400
+
+    _push_wish_to_sheet()
+    return jsonify({"status": "ok"})
+
+"""
+sell_list 코드
+"""
+def _push_sell_to_sheet() -> None:
+    """Google Sheet에 sell list를 복사합니다."""
+    try:
+        grsheet.post_sell_df()
+    except Exception as exc:
+        app.logger.exception("Google Sheet에 복사 실패")
+        raise exc
+
+@app.route("/selllist", methods=["GET"])
+def get_selllist():
+    """GR 트레이더 전체의 sell list를 JSON(table) 형식으로 보내줍니다."""
+    df = grsheet.sell_list.df
+    return df.to_json(orient="table", force_ascii=False), 200, {"Content-Type": "application/json; charset=utf-8"}
+
+@app.route("/selllist/<trader_name>", methods=["GET"])
+def get_trader_sell(trader_name):
+    df = grsheet.sell_list.df
+    if trader_name not in df.index:
+        return jsonify({"error": f"{trader_name}의 sell list가 없습니다."}), 404
+    return df.loc[[trader_name]].to_json(orient="table", force_ascii=False)
+
+@app.route("/selllist/<trader>", methods=["POST"])
+def upsert_sell(trader: str):
+    """GR 트레이더의 sell 추가 (add), 교체 (replace), 삭제 (delete)를 지원합니다."""
+    data = request.get_json(force=True)
+    mode: str = data.get("mode", "add")
+    sell: str = data.get("sell", "매도")
+
+    if mode == "add":
+        stock = data.get("stock_name")
+        if not stock:
+            return jsonify({"error": "add 모드에서는 stock_name이 필요합니다."}), 400
+        grsheet.sell_list.add_wish(trader, stock, sell)
+    elif mode == "replace":
+        # 빈 리스트도 허용
+        stocks = data.get("stock_names")
+        grsheet.sell_list.replace_wishes(trader, stocks, sell)
+    elif mode == "delete":
+        stock = data.get("stock_name")
+        if not stock:
+            return jsonify({"error": "delete 모드에서는 stock_name이 필요합니다."}), 400
+        grsheet.sell_list.remove_wish(trader, stock)
+    else:
+        return jsonify({"error": f"알 수 없는 mode: {mode}"}), 400
+
+    _push_sell_to_sheet()
+    return jsonify({"status": "ok"})
+
+@app.route("/")
+def ping():
+    return "OK"
+
+if __name__ == "__main__":
+    # host='0.0.0.0' 은 개방적인 접속을 위한 파라미터
+    app.run(host="0.0.0.0", port=6009, debug=True)

apsniper0673_grsheetmanager-0.1.4/src/mygrsheetmanager/single_trader_wish_list.py

@@ -0,0 +1,31 @@
+"""
+싱글 트레이더의 매수 희망 종목 리스트를 관리하는 클래스입니다.
+"""
+import pandas as pd
+
+class _SingleDfWishList:
+    def __init__(self, trader_name):
+        self.trader_name = trader_name
+        self.wish_list = pd.Series() # index: 종목명, values: 매수 희망 여부로, "매수" 등등의 문자열로 표현된다.
+class SingleDfWishList(_SingleDfWishList):
+    def add_wish(self, stock_name: str, wish: str="매수") -> None:
+        """
+        매수 희망 종목을 추가한다.
+        :param stock_name: 종목명
+        :param wish: 매수 희망 여부 (예: "매수", "관심")
+        """
+        self.wish_list[stock_name] = wish
+    def remove_wish(self, stock_name: str) -> None:
+        """
+        매수 희망 종목을 제거한다.
+        :param stock_name: 제거할 종목명
+        """
+        if stock_name in self.wish_list:
+            del self.wish_list[stock_name]
+    def replace_wishes(self, stock_names: list, wish: str="매수") -> None:
+        """
+        매수 희망 종목을 일괄적으로 교체한다.
+        :param stock_names: 종목명 리스트
+        :param wish: 매수 희망 여부 (예: "매수", "관심")
+        """
+        self.wish_list = pd.Series(index=stock_names, data=wish)