50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
import json
|
|
import sys
|
|
from collections import OrderedDict
|
|
import vk_api
|
|
|
|
artistStats = dict()
|
|
|
|
|
|
def save():
|
|
with open(sys.argv[2], "w", encoding="utf-8") as file:
|
|
data = OrderedDict(sorted(artistStats.items(), key=lambda item: item[1]["count"], reverse=True))
|
|
file.write(json.dumps(data, sort_keys=False))
|
|
print("saving")
|
|
|
|
|
|
# getting pazans
|
|
pazanIds = None
|
|
pazansFileName = sys.argv[1]
|
|
with open(pazansFileName) as file:
|
|
pazanIds = [int(line) for line in file]
|
|
|
|
# getting music
|
|
vk = vk_api.VkApi(token=sys.argv[3], app_id=sys.argv[4])
|
|
|
|
for index, pazanId in enumerate(pazanIds, start=(sys.argv[5] if len(sys.argv) > 5 else 0)):
|
|
try:
|
|
print(index)
|
|
jsonData = vk.method("audio.get", {"owner_id": pazanId, "need_user": 0, "count": 100})
|
|
for audio in jsonData["items"]:
|
|
audioName = audio["artist"] + " - " + audio["title"]
|
|
artistStatsItem = artistStats.get(audioName, {
|
|
"url": audio["url"],
|
|
"genre_id": audio.get("genre_id", ""),
|
|
"count": 0
|
|
})
|
|
artistStatsItem["count"] += 1
|
|
artistStats[audioName] = artistStatsItem
|
|
|
|
if index % 100 == 0:
|
|
save()
|
|
except vk_api.ApiError as e:
|
|
if e.code != 201:
|
|
print(e)
|
|
break
|
|
except Exception as e:
|
|
print(e)
|
|
break
|
|
|
|
save()
|