Tùy chỉnh Pipeline

# Thay đổi nguồn cung cấp dữ liệu ưu tiên (hỗ trợ chuyển nguồn dự phòng)
python -m vnstock_pipeline.cli run ohlcv --sources VCI,KBS,VND

# Thay đổi khung thời gian cho dữ liệu lịch sử
python -m vnstock_pipeline.cli run ohlcv --interval 1H

# Tuỳ chỉnh khoảng thời gian lấy dữ liệu cụ thể
python -m vnstock_pipeline.cli run ohlcv --start "2024-01-01" --end "2024-12-31"

# ~/.vnstock/config/pipeline.toml
base_path = "/Users/admin/vnstock_db"
layout_mode = "flat"
default_format = "parquet"

[format_overrides]
financial = "excel"     # Ghi đè: Riêng BCTC tự động xuất ra Excel
trades = "csv"          # Ghi đè: Riêng khớp lệnh tự động xuất ra CSV

from vnstock_pipeline.template.vnstock import VNFetcher
import pandas as pd

class ThuThapDuLieuRieng(VNFetcher):
    def _vn_call(self, ticker: str, **kwargs) -> pd.DataFrame:
        from vnstock_data.ui.market import Market
        # Tải dữ liệu từ Vnstock Data
        market = Market(source="vci").equity(ticker)
        df = market.history(start="2024-01-01", end="2026-06-17", interval="1D")
        
        # Thêm cột đánh dấu nguồn dữ liệu riêng của bạn
        df['nguon_du_lieu'] = "he_thong_noi_bo"
        return df

from vnstock_pipeline.template.vnstock import VNTransformer
import pandas as pd

class TinhToanChiBao(VNTransformer):
    def transform(self, data: pd.DataFrame) -> pd.DataFrame:
        df = super().transform(data)
        # Tính toán đường trung bình động 20 ngày (SMA20)
        if len(df) > 20:
            df['sma_20'] = df['close'].rolling(20).mean()
        return df

from vnstock_pipeline.core.exporter import Exporter
from vnstock_pipeline.core.storage.config import StorageConfig

class AWSS3Exporter(Exporter):
    def __init__(self):
        # Hệ thống vẫn tự động nạp cấu hình thư mục từ pipeline.toml
        self.config = StorageConfig.load()
        
    def export(self, data, ticker: str, category: str, **kwargs):
        # Tại đây, bạn viết mã để đẩy bảng dữ liệu 'data' lên tài khoản S3 của mình
        pass

import pandas as pd
from vnstock_pipeline.core.storage.config import StorageConfig

# Tự động đọc cấu hình từ tệp pipeline.toml
config = StorageConfig.load()

# Tự động giải quyết đường dẫn tuyệt đối dù bạn chạy ở bất kỳ máy tính nào
base_path = config.resolve_base_path()

# Đọc tệp dữ liệu đã tải
file_path = base_path / "ohlcv" / "ACB.parquet"
df = pd.read_parquet(file_path)