gistfile1.txt

#!/usr/bin/env python
from linklooker.linklooker import Links, Link
import argparse

def output_results(results):
    if not results:
        exit("Couldn't find any results!")

    #TODO: use for -v mode
    #import os
    #for row in results:
    #    for key in row.keys():
    #        print '{} {}'.format(key, row[key])
    #    print os.linesep
    if parser.parse_args().output:
        out_f = parser.parse_args().output

        # Write out header
        header = results[0].keys()

        # Write out dictionary
        import csv
        data = csv.DictWriter(out_f, header,delimiter=',') 
        data.writeheader()
        for r in results:
            print r
            data.writerow(r)

        out_f.close()



parser = argparse.ArgumentParser()
parser.add_argument('--csv', help='Provide a CSV to scan links from with the format: link, link to look for')
parser.add_argument('--prcsv', help='Provide a CSV to scan links in sho\'s format')
parser.add_argument('--output', type=argparse.FileType('w'), help='Path to output CSV')

links = Links()
if parser.parse_args().csv:
    print "Parsing CSV..."
    results = links.status_table_from_csv(parser.parse_args().csv)
    output_results(results) 

if  parser.parse_args().prcsv:
    print "Parsing Extended CSV..."
    results = links.status_table_from_pr_csv(parser.parse_args().prcsv)
    output_results(results)