vnai 0.1.3__py3-none-any.whl → 2.3.7__py3-none-any.whl

vnai/beam/patching.py

@@ -0,0 +1,223 @@
+"""
+Monkey-patch vnstock source methods to apply tier-based period limiting.
+This module patches VCI and KBS Finance methods to automatically apply
+period limits based on user tier, so end users get limited periods
+even when calling vnstock methods directly.
+"""
+from typing import Optional, Callable
+from functools import wraps
+import pandas as pd
+
+def get_max_periods() -> Optional[int]:
+    from vnai.beam.auth import authenticator
+    PERIOD_LIMITS = {
+'guest': 4,
+'free': 8,
+'bronze': None,
+'silver': None,
+'golden': None,
+    }
+    tier = authenticator.get_tier()
+    return PERIOD_LIMITS.get(tier)
+
+def limit_vci_periods(df: pd.DataFrame, max_periods: Optional[int] = None) -> pd.DataFrame:
+    if max_periods is None or df.empty or'yearReport' not in df.columns:
+        return df
+    df_limited = df.head(max_periods).copy()
+    df_limited = df_limited.reset_index(drop=True)
+    return df_limited
+
+def get_period_limit_notice() -> Optional[str]:
+    from vnai.beam.auth import authenticator
+    tier = authenticator.get_tier()
+    if tier =='guest':
+        return (
+"ℹ️  Phiên bản cộng đồng: Báo cáo tài chính được giới hạn tối đa 4 kỳ để minh hoạ thuật toán. "
+"Để truy cập đầy đủ tất cả các kỳ báo cáo, vui lòng tham gia gói thành viên tài trợ dự án: "
+"https://vnstocks.com/insiders-program"
+        )
+    elif tier =='free':
+        return (
+"ℹ️  Phiên bản cộng đồng: Báo cáo tài chính được giới hạn tối đa 8 kỳ để minh hoạ thuật toán. "
+"Để truy cập đầy đủ tất cả các kỳ báo cáo, vui lòng tham gia gói thành viên tài trợ dự án: "
+"https://vnstocks.com/insiders-program"
+        )
+    return None
+
+def get_period_limit_notice_html() -> Optional[str]:
+    from vnai.beam.auth import authenticator
+    tier = authenticator.get_tier()
+    if tier =='guest':
+        return (
+"<div style='background-color: #e3f2fd; border-left: 4px solid #2196f3; "
+"padding: 12px 16px; margin: 12px 0; border-radius: 4px; font-size: 13px;'>"
+"<strong>ℹ️  Phiên bản cộng đồng</strong><br>"
+"Báo cáo tài chính được giới hạn tối đa <strong>4 kỳ</strong> để minh hoạ thuật toán. "
+"Để truy cập đầy đủ tất cả các kỳ báo cáo, vui lòng "
+"<a href='https://vnstocks.com/insiders-program' target='_blank'>tham gia gói thành viên tài trợ dự án</a>."
+"</div>"
+        )
+    elif tier =='free':
+        return (
+"<div style='background-color: #e3f2fd; border-left: 4px solid #2196f3; "
+"padding: 12px 16px; margin: 12px 0; border-radius: 4px; font-size: 13px;'>"
+"<strong>ℹ️  Phiên bản cộng đồng</strong><br>"
+"Báo cáo tài chính được giới hạn tối đa <strong>8 kỳ</strong> để minh hoạ thuật toán. "
+"Để truy cập đầy đủ tất cả các kỳ báo cáo, vui lòng "
+"<a href='https://vnstocks.com/insiders-program' target='_blank'>tham gia gói thành viên tài trợ dự án</a>."
+"</div>"
+        )
+    return None
+
+def display_period_limit_notice_jupyter() -> None:
+    try:
+        from IPython.display import HTML, display
+        notice_html = get_period_limit_notice_html()
+        if notice_html:
+            display(HTML(notice_html))
+    except ImportError:
+        notice = get_period_limit_notice()
+        if notice:
+            print(f"\n{notice}\n")
+
+def should_show_notice() -> bool:
+    from vnai.beam.auth import authenticator
+    tier = authenticator.get_tier()
+    return tier in ('guest','free')
+
+def limit_periods_by_columns(df: pd.DataFrame, max_periods: Optional[int] = None) -> pd.DataFrame:
+    if max_periods is None:
+        return df
+    metadata_cols = [
+'ticker','yearReport','lengthReport',
+'item','item_en','item_id','unit','levels','row_number'
+    ]
+    period_cols = []
+    for col in df.columns:
+        if isinstance(col, str) and col not in metadata_cols:
+            if col.isdigit() or (len(col) > 4 and'-Q' in col):
+                period_cols.append(col)
+    if not period_cols:
+        return df
+
+    def parse_period(p):
+        if'-Q' in str(p):
+            year, quarter = str(p).split('-Q')
+            return (int(year), int(quarter))
+        else:
+            return (int(p), 5)
+    period_cols_sorted = sorted(period_cols, key=parse_period, reverse=True)
+    keep_periods = period_cols_sorted[:max_periods]
+    metadata_cols_present = [col for col in metadata_cols if col in df.columns]
+    financial_cols = [col for col in df.columns if col not in metadata_cols and col not in period_cols]
+    final_cols = metadata_cols_present + financial_cols + keep_periods
+    return df[final_cols]
+
+def patch_vci_finance():
+    try:
+        import sys
+        if'vnstock.explorer.vci.financial' not in sys.modules:
+            return False
+        from vnstock.explorer.vci.financial import Finance as VCI_Finance
+        original_balance_sheet = VCI_Finance.balance_sheet
+        original_income_statement = VCI_Finance.income_statement
+        original_cash_flow = VCI_Finance.cash_flow
+        _notice_shown = {'balance_sheet': False,'income_statement': False,'cash_flow': False}
+        @wraps(original_balance_sheet)
+
+        def balance_sheet_with_limit(self, period: Optional[str] = None, lang: Optional[str] ='en',
+                                     dropna: Optional[bool] = True, show_log: Optional[bool] = False) -> pd.DataFrame:
+            df = original_balance_sheet(self, period=period, lang=lang, dropna=dropna, show_log=show_log)
+            df_limited = limit_vci_periods(df, max_periods=get_max_periods())
+            if should_show_notice() and not _notice_shown['balance_sheet']:
+                _notice_shown['balance_sheet'] = True
+                display_period_limit_notice_jupyter()
+            return df_limited
+        @wraps(original_income_statement)
+
+        def income_statement_with_limit(self, period: Optional[str] = None, lang: Optional[str] ='en',
+                                       dropna: Optional[bool] = True, show_log: Optional[bool] = False) -> pd.DataFrame:
+            df = original_income_statement(self, period=period, lang=lang, dropna=dropna, show_log=show_log)
+            df_limited = limit_vci_periods(df, max_periods=get_max_periods())
+            if should_show_notice() and not _notice_shown['income_statement']:
+                _notice_shown['income_statement'] = True
+                display_period_limit_notice_jupyter()
+            return df_limited
+        @wraps(original_cash_flow)
+
+        def cash_flow_with_limit(self, period: Optional[str] = None, lang: Optional[str] ='en',
+                                dropna: Optional[bool] = True, show_log: Optional[bool] = False) -> pd.DataFrame:
+            df = original_cash_flow(self, period=period, lang=lang, dropna=dropna, show_log=show_log)
+            df_limited = limit_vci_periods(df, max_periods=get_max_periods())
+            if should_show_notice() and not _notice_shown['cash_flow']:
+                _notice_shown['cash_flow'] = True
+                display_period_limit_notice_jupyter()
+            return df_limited
+        VCI_Finance.balance_sheet = balance_sheet_with_limit
+        VCI_Finance.income_statement = income_statement_with_limit
+        VCI_Finance.cash_flow = cash_flow_with_limit
+        return True
+    except Exception as e:
+        print(f"Warning: Could not patch VCI Finance: {e}")
+        return False
+
+def patch_kbs_finance():
+    try:
+        from vnstock.explorer.kbs.financial import Finance as KBS_Finance
+        from vnai.beam.fundamental import limit_periods_by_columns
+        original_balance_sheet = KBS_Finance.balance_sheet
+        original_income_statement = KBS_Finance.income_statement
+        original_cash_flow = KBS_Finance.cash_flow
+        _notice_shown = {'balance_sheet': False,'income_statement': False,'cash_flow': False}
+        @wraps(original_balance_sheet)
+
+        def balance_sheet_with_limit(self, period: Optional[str] = None, show_log: Optional[bool] = False) -> pd.DataFrame:
+            df = original_balance_sheet(self, period=period, show_log=show_log)
+            df_limited = limit_periods_by_columns(df, max_periods=get_max_periods())
+            if should_show_notice() and not _notice_shown['balance_sheet']:
+                _notice_shown['balance_sheet'] = True
+                display_period_limit_notice_jupyter()
+            return df_limited
+        @wraps(original_income_statement)
+
+        def income_statement_with_limit(self, period: Optional[str] = None, show_log: Optional[bool] = False) -> pd.DataFrame:
+            df = original_income_statement(self, period=period, show_log=show_log)
+            df_limited = limit_periods_by_columns(df, max_periods=get_max_periods())
+            if should_show_notice() and not _notice_shown['income_statement']:
+                _notice_shown['income_statement'] = True
+                display_period_limit_notice_jupyter()
+            return df_limited
+        @wraps(original_cash_flow)
+
+        def cash_flow_with_limit(self, period: Optional[str] = None, show_log: Optional[bool] = False) -> pd.DataFrame:
+            df = original_cash_flow(self, period=period, show_log=show_log)
+            df_limited = limit_periods_by_columns(df, max_periods=get_max_periods())
+            if should_show_notice() and not _notice_shown['cash_flow']:
+                _notice_shown['cash_flow'] = True
+                display_period_limit_notice_jupyter()
+            return df_limited
+        KBS_Finance.balance_sheet = balance_sheet_with_limit
+        KBS_Finance.income_statement = income_statement_with_limit
+        KBS_Finance.cash_flow = cash_flow_with_limit
+        return True
+    except Exception as e:
+        print(f"Warning: Could not patch KBS Finance: {e}")
+        return False
+_patches_applied = False
+_patches_lock = __import__('threading').Lock()
+
+def apply_all_patches():
+    global _patches_applied
+    with _patches_lock:
+        if _patches_applied:
+            return {'vci': True,'kbs': True}
+        try:
+            vci_patched = patch_vci_finance()
+            kbs_patched = patch_kbs_finance()
+            _patches_applied = True
+            return {
+'vci': vci_patched,
+'kbs': kbs_patched,
+            }
+        except Exception as e:
+            return {'vci': False,'kbs': False}

vnai/beam/pulse.py

@@ -0,0 +1,79 @@
+import threading
+import time
+from datetime import datetime
+
+class Monitor:
+    _instance = None
+    _lock = threading.Lock()
+
+    def __new__(cls):
+        with cls._lock:
+            if cls._instance is None:
+                cls._instance = super(Monitor, cls).__new__(cls)
+                cls._instance._initialize()
+            return cls._instance
+
+    def _initialize(self):
+        self.health_status ="healthy"
+        self.last_check = time.time()
+        self.check_interval = 300
+        self.error_count = 0
+        self.warning_count = 0
+        self.status_history = []
+        self._start_background_check()
+
+    def _start_background_check(self):
+        def check_health():
+            while True:
+                try:
+                    self.check_health()
+                except:
+                    pass
+                time.sleep(self.check_interval)
+        thread = threading.Thread(target=check_health, daemon=True)
+        thread.start()
+
+    def check_health(self):
+        from vnai.beam.metrics import collector
+        from vnai.beam.quota import guardian
+        self.last_check = time.time()
+        metrics_summary = collector.get_metrics_summary()
+        has_errors = metrics_summary.get("error", 0) > 0
+        resource_usage = guardian.usage()
+        high_usage = resource_usage > 80
+        if has_errors and high_usage:
+            self.health_status ="critical"
+            self.error_count += 1
+        elif has_errors or high_usage:
+            self.health_status ="warning"
+            self.warning_count += 1
+        else:
+            self.health_status ="healthy"
+        self.status_history.append({
+"timestamp": datetime.now().isoformat(),
+"status": self.health_status,
+"metrics": metrics_summary,
+"resource_usage": resource_usage
+        })
+        if len(self.status_history) > 10:
+            self.status_history = self.status_history[-10:]
+        return self.health_status
+
+    def report(self):
+        if time.time() - self.last_check > self.check_interval:
+            self.check_health()
+        return {
+"status": self.health_status,
+"last_check": datetime.fromtimestamp(self.last_check).isoformat(),
+"error_count": self.error_count,
+"warning_count": self.warning_count,
+"history": self.status_history[-3:],
+        }
+
+    def reset(self):
+        self.health_status ="healthy"
+        self.error_count = 0
+        self.warning_count = 0
+        self.status_history = []
+        self.last_check = time.time()
+monitor = Monitor()