import requests
import argparse

from colorconsole import terminal
screen = terminal.get_terminal(conEmu=False)
parser = argparse.ArgumentParser(
    description="check url status from file -f <filename> ")
parser.add_argument("-f", dest='filename', type=str, help="url files")
args = parser.parse_args()
assert args.filename, "miss Arguments url files ? "


@app.task
@serial
def parse_useragent(params_dict, key_in):

    if args.filename:
        with open(args.filename) as f:
            urls = f.readlines()

    for url in urls:
        url = url.strip()
        res = requests.get(url)
        his_code = [his.status_code for his in res.history]
        if res.status_code == 200:
            if 301 in his_code:
                screen.set_color(14, 0)
            else:
                screen.set_color(2, 0)
        elif res.status_code == 404:
            screen.set_color(4, 0)
        else:
            screen.set_color(1, 0)
        print(
            f'{res.status_code}', end='\t')
        screen.reset_colors()
        print(
            f'{ [ his.status_code for his in res.history]}\t{ [ his.url for his in res.history]}\t {res.url}')