[8e6598]: scripts / allura_import_api.py History

Download this file

allura_import_api.py    50 lines (42 with data), 1.6 kB 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import sys

import urllib

import urllib2

import urlparse

import hmac

import hashlib

import json

import logging

from datetime import datetime





log = logging.getLogger('import')



class AlluraImportApiClient(object):



    def __init__(self, base_url, api_key, secret_key, verbose=False, retry=True):

        self.base_url = base_url

        self.api_key = api_key

        self.secret_key = secret_key

        self.verbose = verbose

        self.retry = retry



    def sign(self, path, params):

        params.append(('api_key', self.api_key))

        params.append(('api_timestamp', datetime.utcnow().isoformat()))

        message = path + '?' + urllib.urlencode(sorted(params))

        digest = hmac.new(self.secret_key, message, hashlib.sha256).hexdigest()

        params.append(('api_signature', digest))

        return params



    def call(self, url, **params):

        url = urlparse.urljoin(self.base_url, url)

        params = self.sign(urlparse.urlparse(url).path, params.items())



        while True:

            try:

                result = urllib2.urlopen(url, urllib.urlencode(params))

                resp = result.read()

                return json.loads(resp)

            except urllib2.HTTPError, e:

                if self.verbose:

                    error_content = e.read()

                    e.msg += '. Error response:\n' + error_content

                raise e

            except (urllib2.URLError, IOError):

                if self.retry:

                    log.exception('Error making API request, will retry')

                    continue

                raise