pharos-dashboard/account/jira_util.py

   1 import base64
   2 import os
   3
   4 import oauth2 as oauth
   5 from jira import JIRA
   6 from tlslite.utils import keyfactory
   7
   8 from pharos_dashboard import settings
   9
  10
  11 class SignatureMethod_RSA_SHA1(oauth.SignatureMethod):
  12     name = 'RSA-SHA1'
  13
  14     def signing_base(self, request, consumer, token):
  15         if not hasattr(request, 'normalized_url') or request.normalized_url is None:
  16             raise ValueError("Base URL for request is not set.")
  17
  18         sig = (
  19             oauth.escape(request.method),
  20             oauth.escape(request.normalized_url),
  21             oauth.escape(request.get_normalized_parameters()),
  22         )
  23
  24         key = '%s&' % oauth.escape(consumer.secret)
  25         if token:
  26             key += oauth.escape(token.secret)
  27         raw = '&'.join(sig)
  28         return key, raw
  29
  30     def sign(self, request, consumer, token):
  31         """Builds the base signature string."""
  32         key, raw = self.signing_base(request, consumer, token)
  33
  34         module_dir = os.path.dirname(__file__)  # get current directory
  35         with open(module_dir + '/rsa.pem', 'r') as f:
  36             data = f.read()
  37         privateKeyString = data.strip()
  38         privatekey = keyfactory.parsePrivateKey(privateKeyString)
  39         raw = str.encode(raw)
  40         signature = privatekey.hashAndSign(raw)
  41         return base64.b64encode(signature)
  42
  43
  44 def get_jira(user):
  45     module_dir = os.path.dirname(__file__)  # get current directory
  46     with open(module_dir + '/rsa.pem', 'r') as f:
  47         key_cert = f.read()
  48
  49     oauth_dict = {
  50         'access_token': user.userprofile.oauth_token,
  51         'access_token_secret': user.userprofile.oauth_secret,
  52         'consumer_key': settings.OAUTH_CONSUMER_KEY,
  53         'key_cert': key_cert
  54     }
  55
  56     return JIRA(server=settings.JIRA_URL, oauth=oauth_dict)