from argparse import ArgumentParser

from tasks.monitor_tasks.outside_gzh_articles_monitor import OutsideGzhArticlesCollector
from tasks.monitor_tasks.outside_gzh_articles_monitor import OutsideGzhArticlesMonitor


if __name__ == "__main__":
    parser = ArgumentParser()
    parser.add_argument("--task", help="input monitor or collector")
    args = parser.parse_args()
    if args.task:
        task = args.task
        match task:
            case "monitor":
                monitor = OutsideGzhArticlesMonitor()
                monitor.deal()
            case "collector":
                collector = OutsideGzhArticlesCollector()
                collector.deal()
            case _:
                print("task is not support")
    else:
        # first collect data
        collector = OutsideGzhArticlesCollector()
        collector.deal()

        # then monitor each article
        monitor = OutsideGzhArticlesMonitor()
        monitor.deal()