gistfile1.txt

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

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

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

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

    output_file.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()
output_file = parser.parse_args().output

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

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