Files
gotohack/audio-fetcher.py
Oleg Morozenkov a3ab86c6e5 Audio fetcher
2016-02-22 09:59:31 +03:00

42 lines
1.0 KiB
Python

import json
import sys
import time
import vk_api
# 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=(int(sys.argv[5]) if len(sys.argv) > 5 else 0)):
done = False
while not done:
try:
print(index)
pazanSongs = []
jsonData = vk.method("audio.get", {"owner_id": pazanId, "need_user": 0, "count": 100})
for audio in jsonData["items"]:
pazanSong = {
"artist": audio["artist"],
"title": audio["title"],
"genre_id": audio.get("genre_id", None),
"url": audio["url"],
}
pazanSongs.append(pazanSong)
with open(sys.argv[2], "a", encoding="utf-8") as file:
file.write(json.dumps({pazanId: pazanSongs}, ensure_ascii=False) + "\n")
done = True
except vk_api.ApiError as e:
if e.code == 9:
print("waiting")
time.sleep(60)
elif e.code == 201 or e.code == 15:
done = True
else:
raise e