Вот код, который я использовал для решения этой проблемы в одном из моих проектов.Это пытается найти данные о месте в нескольких разных местах внутри метаданных твита и возвращает None
, если не может.
states = {
'AL': 'Alabama',
'AK': 'Alaska',
'AZ': 'Arizona',
'AR': 'Arkansas',
'CA': 'California',
'CO': 'Colorado',
'CT': 'Connecticut',
'DE': 'Delaware',
'DC': 'District of Columbia',
'FL': 'Florida',
'GA': 'Georgia',
'HI': 'Hawaii',
'ID': 'Idaho',
'IL': 'Illinois',
'IN': 'Indiana',
'IA': 'Iowa',
'KS': 'Kansas',
'KY': 'Kentucky',
'LA': 'Louisiana',
'ME': 'Maine',
'MD': 'Maryland',
'MA': 'Massachusetts',
'MI': 'Michigan',
'MN': 'Minnesota',
'MS': 'Mississippi',
'MO': 'Missouri',
'MT': 'Montana',
'NE': 'Nebraska',
'NV': 'Nevada',
'NH': 'New Hampshire',
'NJ': 'New Jersey',
'NM': 'New Mexico',
'NY': 'New York',
'NC': 'North Carolina',
'ND': 'North Dakota',
'OH': 'Ohio',
'OK': 'Oklahoma',
'OR': 'Oregon',
'PA': 'Pennsylvania',
'RI': 'Rhode Island',
'SC': 'South Carolina',
'SD': 'South Dakota',
'TN': 'Tennessee',
'TX': 'Texas',
'UT': 'Utah',
'VT': 'Vermont',
'VA': 'Virginia',
'WA': 'Washington',
'WV': 'West Virginia',
'WI': 'Wisconsin',
'WY': 'Wyoming'
}
def extract_place(status):
if type(status) is tweepy.models.Status:
status = status.__dict__
#Try to get the place from the place data inside the status dict
if status['place'] is not None:
place = status['place']
if place['country'] != 'United States':
return place['country']
elif place['place_type'] == 'admin':
return place['name']
elif place['place_type'] == 'city':
return states.get(place['full_name'].split(', ')[-1])
#If the status dict has no place info, get the place from the user data
else:
place = status['user']['location']
try:
place = place.split(', ')[-1].upper()
except AttributeError:
return None
if place in states:
return states[place]
else:
return place