src/apiclient/__init__.py

   1 import urllib
   2
   3 import json
   4 import oauth2
   5
   6 from apiclient.settings import WL_CONSUMER_KEY, WL_CONSUMER_SECRET, WL_API_URL, BETA_API_URL
   7
   8
   9 if WL_CONSUMER_KEY and WL_CONSUMER_SECRET:
  10     wl_consumer = oauth2.Consumer(WL_CONSUMER_KEY, WL_CONSUMER_SECRET)
  11 else:
  12     wl_consumer = None
  13
  14
  15 class ApiError(BaseException):
  16     pass
  17
  18
  19 class NotAuthorizedError(BaseException):
  20     pass
  21
  22
  23 def api_call(user, path, data=None, beta=False):
  24     from .models import OAuthConnection
  25     api_url = BETA_API_URL if beta else WL_API_URL
  26     conn = OAuthConnection.get(user=user, beta=beta)
  27     if not conn.access:
  28         raise NotAuthorizedError("No WL authorization for user %s." % user)
  29     token = oauth2.Token(conn.token, conn.token_secret)
  30     client = oauth2.Client(wl_consumer, token)
  31     if data is not None:
  32         data = json.dumps(data)
  33         data = urllib.urlencode({"data": data})
  34         resp, content = client.request(
  35                 "%s%s" % (api_url, path),
  36                 method="POST",
  37                 body=data)
  38     else:
  39         resp, content = client.request(
  40                 "%s%s" % (api_url, path))
  41     status = resp['status']
  42
  43     if status == '200':
  44         return json.loads(content)
  45     elif status.startswith('2'):
  46         return
  47     elif status == '401':
  48         raise ApiError('User not authorized for publishing.')
  49     else:
  50         raise ApiError("WL API call error %s, path: %s" % (status, path))
  51