/
check_concerts.py
157 lines (115 loc) · 4.24 KB
/
check_concerts.py
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
#!/usr/bin/env python
import os
import sys
import configparser
from gmusicapi import Mobileclient
import pprint as pprint
import pickle
from collections import Counter
import requests
import json
pp = pprint.PrettyPrinter(indent=4)
class gmusicConfig:
def __init__(self, account):
self.account = account
self.open_config()
self.api = Mobileclient()
def open_config(self):
config = configparser.ConfigParser()
cf = os.path.join(
os.path.dirname(os.path.realpath(__file__)),
"api.cfg"
)
if not config.read([cf]):
print("No login configuration file found!")
self.email = config.get(self.account, 'email')
self.password = config.get(self.account, 'password')
def login(self):
try:
logged_in = self.api.login(self.email, self.password, self.api.FROM_MAC_ADDRESS)
if logged_in == False:
print("Google Music API login failed")
sys.exit()
if logged_in == True:
return logged_in
except:
print("Google Music API login failed")
class songKickConfig:
def __init__(self, account):
self.account = account
self.open_config()
def open_config(self):
config = configparser.ConfigParser()
cf = os.path.join(
os.path.dirname(os.path.realpath(__file__)),
"api.cfg"
)
if not config.read([cf]):
print("No login configuration file found!")
self.api_key = config.get(self.account, 'api_key')
def count_list(song_list):
play_count_list = []
for song in song_list:
if 'playCount' in song:
song_artist = song['artist']
song_count = song['playCount']
dict = {'artist': song_artist, 'count': song_count}
play_count_list.append(dict)
#Combine values of list
top_list = Counter()
for play_count in play_count_list:
top_list[play_count['artist']] += play_count['count']
return top_list
def get_top_artists(all_artists, top_count):
top_artists = dict(Counter(all_artists).most_common(top_count))
return top_artists
def get_songkick_artist_id(top_artists, songkick):
songkick_search_url = 'http://api.songkick.com/api/3.0/search/artists.json'
artist_ids = []
for artist in top_artists:
# pp.pprint(artist)
if '&' in artist:
artist = artist.replace("&", "and")
parameters = {"apikey": songkick.api_key, "query": artist, 'page': 1}
try:
r = requests.get(songkick_search_url, params=parameters)
# print(r.json()['resultsPage']['results']['artist'][0])
except Exception as e:
print("Artist lookup failed!")
print(e)
try:
artist_ids.append(r.json()['resultsPage']['results']['artist'][0]['id'])
except KeyError as e:
print('KeyError: No artist returned!')
# print(e)
return artist_ids
def main():
top_count = 50
gmusic = gmusicConfig('gmusic')
gmusic.login() #Login to the API
#Grab top songs from gmusic and output to pickle file
# pickle_out = open("songs.pickle","wb")
# song_list = gmusic.api.get_all_songs()
# pickle.dump(song_list, pickle_out)
#Input pickle file of gmusic data
pickle_in = open('songs.pickle','rb')
song_list = pickle.load(pickle_in)
all_artists = count_list(song_list) #Get top artists
top_artists = get_top_artists(all_artists, top_count) #Get top n artists
#Create songkick object
songkick = songKickConfig('songkick')
#Get artist id's from songkick API
artist_ids = get_songkick_artist_id(top_artists, songkick)
for artist_id in artist_ids:
print(artist_id)
url = 'http://api.songkick.com/api/3.0/artists/{artist_id}/calendar.json'
parameters = {"apikey": songkick.api_key, "artist_id": artist_id}# "artist_id": 3184796}
r = requests.get(url, params=parameters)
pp.pprint(r.json()['resultsPage']['results'])
# pp.pprint(vars(r))
# pp.pprint(artist_ids)
# for artist in top_artists:
# pp.pprint(type(artist))
# print(top_artist)
if __name__ == "__main__":
main()