src/apiclient/__init__.py

   1 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 import json
   5 from urllib.parse import urlencode
   6 import oauth2
   7 from apiclient.settings import WL_CONSUMER_KEY, WL_CONSUMER_SECRET, WL_API_URL, BETA_API_URL
   8
   9
  10 if WL_CONSUMER_KEY and WL_CONSUMER_SECRET:
  11     wl_consumer = oauth2.Consumer(WL_CONSUMER_KEY, WL_CONSUMER_SECRET)
  12 else:
  13     wl_consumer = None
  14
  15
  16 class ApiError(BaseException):
  17     pass
  18
  19
  20 class NotAuthorizedError(BaseException):
  21     pass
  22
  23
  24 def api_call(user, path, data=None, beta=False, method=None, as_json=False):
  25     from .models import OAuthConnection
  26     api_url = BETA_API_URL if beta else WL_API_URL
  27     conn = OAuthConnection.get(user=user, beta=beta)
  28     if not conn.access:
  29         raise NotAuthorizedError("No WL authorization for user %s." % user)
  30     token = oauth2.Token(conn.token, conn.token_secret)
  31     client = oauth2.Client(wl_consumer, token)
  32     if data is not None:
  33         data = json.dumps(data)
  34         headers = {}
  35         if as_json:
  36             headers["Content-Type"] = 'application/json'
  37         else:
  38             data = urlencode({"data": data})
  39         data = data.encode('utf-8')
  40
  41         resp, content = client.request(
  42             "%s%s" % (api_url, path),
  43             method=method or 'POST',
  44             headers=headers,
  45             body=data)
  46     else:
  47         resp, content = client.request(
  48             "%s%s" % (api_url, path),
  49             method=method or 'GET'
  50         )
  51     status = resp['status']
  52
  53     if status in ('200', '201'):
  54         return json.loads(content)
  55     elif status.startswith('2'):
  56         return
  57     elif status == '401':
  58         raise ApiError('User not authorized for publishing.')
  59     else:
  60         raise ApiError("WL API call error %s, path: %s, body: %s" % (status, path, content))