import csv
import itchat
from pprint import pprint

def analyseLocation(friends):
    headers = ['NickName','Province','City']
    with open('wechat-Location.csv','w',encoding='utf-8',newline='') as csvFile:
        
        writer = csv.DictWriter(csvFile, headers)
        writer.writeheader()   
        
        for friend in friends[1:]:
            row = {}
            row['NickName'] = friend['NickName']
            row['Province'] = friend['Province']
            row['City'] = friend['City']
            pprint(row)
            writer.writerow(row)


if __name__=="__main__":
    itchat.auto_login(hotReload=True)
    friends = itchat.get_friends()
    analyseLocation(friends)