from django.core.management.base import BaseCommand
import json
from iching.utils.bz import calcYearFate

class Command(BaseCommand):
    help = 'Test the calcYearFate method'

    def add_arguments(self, parser):
        parser.add_argument('--col', type=int, default=1)
        parser.add_argument('--y', type=int, default=None)

    def handle(self, *args, **options):
        self._calcYearFate(options)
    
    def _calcYearFate(self, options):
        result = calcYearFate(options['col'], options['y'])
        self.stdout.write(f"Testing for {options['y']} year fate ({options['col']} columns)")
        self.stdout.write(f"{result}")