~roignac/gwibber/vkontakte-ru-plugin-cleanup

# -*- coding: utf-8 -*-

'''

Created on 07.12.2010

 

@author: Sergey Prokhorov <root@seriyps.ru>

'''

from gwibber.microblog import network

from gwibber.microblog.util import log

import urllib

import time

 

class vkException(Exception):

 

  def __init__(self, code, msg, url, response):

    self.code=code

    self.msg=msg

    self.response=response

    self.url=url

    Exception.__init__(self, code, msg, url, response)

 

 

class vk_api:

  """Vkontakte.ru (vk.com) API wrapper for Gwibber

  == Usage ==

  Initialization:

    api=vk_api(access_token)

    "access_token" can be retrieved from call to http://api.vkontakte.ru/oauth/authorize

    (see http://vkontakte.ru/developers.php?o=-1&p=%C0%E2%F2%EE%F0%E8%E7%E0%F6%E8%FF%20%EA%EB%E8%E5%ED%F2%F1%EA%E8%F5%20%EF%F0%E8%EB%EE%E6%E5%ED%E8%E9)

  Make calls to simple methods (has no dots in name, like "getProfiles", "isAppUser"):

    res=api.method(arg_name1=arg1, arg_name2=arg2)

    eg res=api.getProfiles(uids="535397,1", fields="uid,first_name,last_name")

  Make calls to namespaced methods (has dots in name, like "wall.post", "newsfeed.get"):

    res=api.namespace.method(arg_name1=arg1, arg_name2=arg2)

    eg res=api.wall.post(message="New wall post from API")

  Alternative call syntax (better performance):

    res=api._load('namespace.method', arg_name1=arg1, arg_name2=arg2)

    eg res=api._get('wall.post', message="New wall post from API")

  """

 

  api_url='https://api.vkontakte.ru/method/%s'

 

  def __init__(self, access_token):

    self._access_token=access_token

    self._prefix=[]

 

  def _load(self, method, **params):

    """Make call by method name and arguments"""

    params["access_token"]=self._access_token

    url=self.api_url%method

    for _i in xrange(3):#this cycle used for API error #6 "Too many requests"

      log.logger.debug("Perform API request to method %s. URL: %s?%s", method, url, urllib.urlencode(params))

      res=network.Download(url, params, False).get_json()

      if not res.has_key("error"):

        #log.logger.debug("%s"%res)

        return res

 

      if res["error"]["error_code"]!=6:

        break

      log.logging.warning('Error #6 "%s". Wait 0.5s and retry...', res["error"]["error_msg"])

      '''if we get error #6 "Too many requests per second" (vk allow

      max 3rps http://vkontakte.ru/developers.php?o=-1&p=%D0%92%D0%B7%D0%B0%D0%B8%D0%BC%D0%BE%D0%B4%D0%B5%D0%B9%D1%81%D1%82%D0%B2%D0%B8%D0%B5+%D0%BF%D1%80%D0%B8%D0%BB%D0%BE%D0%B6%D0%B5%D0%BD%D0%B8%D1%8F+%D1%81+API )

      try to sleep with small delay and retry request not more then 3x times'''

      time.sleep(0.5)#TODO: play with sleep value

    raise vkException(res["error"]["error_code"],

                          res["error"]["error_msg"],

                          "%s?%s"%(url, urllib.urlencode(params)),

                          res)

 

 

  """Support for api.namespace.method(**kwargs) calls below"""

  def __getattr__(self, name):

    self._prefix.append(name)

    return self

 

  def __call__(self, **kwargs):

    if self._prefix:

      method=".".join(self._prefix)

      self._prefix=[]

    else:

      method=kwargs.pop("method")

    return self._load(method, **kwargs)