Put First Google Result in a New CSV Next to Old CSV Full of Search Terms

feeling-lucky.py

import os
import csv
from tqdm import tqdm
from google import search

file_name = "blah.csv"
output_file_name = "out.csv"

input_file = csv.DictReader(open(file_name, 'rU'), dialect='excel')
out_file = csv.DictWriter(open(output_file_name, 'w'), fieldnames=['query', 'result_url'])
num_rows = len(open(file_name, 'rU').readlines())

out_file.writeheader()

for row in tqdm(input_file, total=num_rows):
    search_resp = ''
    for url in search(row['query'], stop=1):
        search_resp = url
    out_file.writerow({'query': row['query'], 'result_url': search_resp})

requirements.txt

beautifulsoup4==4.6.0
google==1.9.3
tqdm==4.15.0