#!/usr/bin/env python3

import argparse
import os
import scanner
from scanner import download

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('thread', help='url of the thread')
    parser.add_argument('folder', nargs='?',
                        help='Change the folder name where images are downloaded')
    parser.add_argument("-q", "--quiet", action='store_true', help="quiet")
    parser.add_argument("-o", "--output", help="Specify output folder")
    args = parser.parse_args()

    print(args.thread)

    quiet = False
    if args.quiet:
        quiet = True

    if args.folder is not None:
        folder = ''.join(args.folder)
    else:
        folder = ''.join(args.thread).split('/')[5].split('#')[0]

    if args.output is not None:
        output = args.output
        if not os.path.exists(output):
            print("{0} Does not exist.".format(output))
            exit(1)
    else:
        output = os.getcwd()

    download.download_thread(args.thread, output, folder, quiet)


if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        pass
