from core.pipeline.step import PipelineStep
from engines.period_comparison_engine import compute_period_comparison

class ComparisonStep(PipelineStep):
    name = "period_comparison"

    def __init__(self, weekly_repo):
        self.weekly_repo = weekly_repo

    async def run(self, context):
        all_snapshots = []
        if self.weekly_repo:
            all_snapshots = self.weekly_repo.get_last_n(20)
        
        context.period_comparison = compute_period_comparison(all_snapshots)