#!/usr/bin/python
# -*- coding: utf-8 -*-

from userapi import *
import urllib2
import os
import webbrowser

USER = "@mail.ru"
PASS = "pass"

def getFriendsAvatars(id = None):
    for friend in handler.v_friends(None, id, 0, 60):
        print str(friend.avatar)
        try: os.mkdir("avatars")
        except OSError: pass # directory exists
        if friend.avatar != '0':
            downloadPicture(str(friend.avatar), "avatars", str(friend.id)+".jpg")

def downloadPicture(url, pathToDir, title):
    wf = urllib2.build_opener()
    lf = open("avatars/"+title, "wb")
    lf.write(wf.open(url).read())
    lf.close()

try:
    session = Session()
    session.login(USER, PASS)
    handler = UserAPI(session)
    getFriendsAvatars()
    session.logout()

except UserAPIError as error:
    print "Get code: " + str(error.code) + " " + error.text

except JSONProblemError as error:
    print "JSON data is a bullshit, storing to disk"
    f = file('bs', 'w')
    f.write(error.json_data)
    f.write("\n\n\n" + str(error.supplement))
    f.close()