FilesExpand file tree

 master

/

25_ip2geolocation.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

61 lines (53 loc) · 1.78 KB

 master

/

25_ip2geolocation.py

Copy path

Top

File metadata and controls

• Code
• Blame

61 lines (53 loc) · 1.78 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

import csv

import requests

def get_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 = []

with open(filename, 'rt') as f:

reader = csv.reader(f)

for row in reader:

all_addresses.append(row)

return all_addresses

def get_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

for line in all_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)

return updated_addresses

def create_csv(updated_address_list):

"""

Given the updated lists of lists from `get_geolocation()`, this function

creates a new CSV.

"""

import sys

if sys.version_info >= (3, 0, 0):

f = open('output.csv', 'w', newline='')

else:

f = open('output.csv', 'wb')

with f:

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)