added ip -> geolocation script

Author: mjhea0

25_ip2geolocation.py

@@ -0,0 +1,56 @@
+importcsv
+importrequests
+
+
+defget_addresses(filename):
+"""
+ Given a CSV file, this function returns a list of lists
+ where each element (list) in the outer list contains the
+ row info from the csv file.
+ """
+all_addresses= []
+withopen(filename, 'rb') asf:
+reader=csv.reader(f)
+forrowinreader:
+all_addresses.append(row)
+returnall_addresses
+
+
+defget_geolocation(all_the_ip_address):
+"""
+ Given a list of lists from `get_addresses()`, this function
+ returns an updated lists of lists containing the geolocation.
+ """
+print("Getting geo information...")
+updated_addresses= []
+counter=1
+# update header
+header_row=all_the_ip_address.pop(0)
+header_row.extend(['Country', 'City'])
+# get geolocation
+forlineinall_the_ip_address:
+print"Grabbing geo info for row #{0}".format(counter)
+r=requests.get('https://freegeoip.net/json/{0}'.format(line[0]))
+line.extend([str(r.json()['country_name']), str(r.json()['city'])])
+updated_addresses.append(line)
+counter+=1
+updated_addresses.insert(0, header_row)
+returnupdated_addresses
+
+
+defcreate_csv(updated_address_list):
+"""
+ Given the updated lists of lists from `get_geolocation()`, this function
+ creates a new CSV.
+ """
+withopen('output.csv', 'wb') asf:
+writer=csv.writer(f)
+writer.writerows(updated_address_list)
+print"All done!"
+
+
+if__name__=='__main__':
+csv_file='25_sample_csv.csv'
+all_the_ip_address=get_addresses(csv_file)
+updated_address_list=get_geolocation(all_the_ip_address)
+create_csv(updated_address_list)

25_sample_csv.csv

@@ -0,0 +1,6 @@
+IP Address,Full Name,Id,Email
+162.252.85.172,Virgie Simonis,0,[email protected]
+208.110.83.202,Tyrese Bartoletti,1,[email protected]
+108.162.199.95,Markus Sanford,2,[email protected]
+169.228.182.227,Anastasia Sawayn,3,[email protected]
+184.72.242.188,Ashly Howe,5,[email protected]

readme.md

@@ -23,4 +23,5 @@
 1.**21_twitter_bot.py**: Twitter Bot
 1.**22_git_tag.py**: Create Git Tag based on a commit
 1.**23_flask_session_test.py**: Just a simple app to see if the sessions are working
-1.**24_sql2csv.py**: SQL to CSV.
+1.**24_sql2csv.py**: SQL to CSV.
+1.**25_ip2geolocation.py**: Given a CSV file with an ip address (see sample - *25_sample_csv.csv*), return the geolocation based on the ip.