import core.logger as logging
from core.statistics.similar_pattern_stocks import SimilarPatternStocks
import os

logger = logging.logger

def main():
    similar_pattern_stocks = SimilarPatternStocks()

    target_stock_list = [
        {
            "symbol": "600111.SH",
            "bar": "1W",
            "start_date": None,
            "end_date": "20250711",
            "compare_record_amount": 100,
        },
        {
            "symbol": "600111.SH",
            "bar": "1M",
            "start_date": None,
            "end_date": "20250630",
            "compare_record_amount": 100,
        },
        {
            "symbol": "601398.SH",
            "bar": "1M",
            "start_date": None,
            "end_date": "20230430",
            "compare_record_amount": 100,
        }
    ]

    for target_stock in target_stock_list:
        similar_pattern_stocks.get_stock_market_data_similar_pattern(
            target_symbol=target_stock["symbol"],
            bar=target_stock["bar"],
            start_date=target_stock["start_date"],
            end_date=target_stock["end_date"],
            compare_record_amount=target_stock["compare_record_amount"],
        )


if __name__ == "__main__":
    main()