Completed
Push — master ( 98e9e8...6ab974 )
by Fabian
8s
created

Deployment.get_payload()   A

Complexity

Conditions 1

Size

Total Lines 7

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 2
CRAP Score 1

Importance

Changes 1
Bugs 0 Features 0
Metric Value
dl 0
loc 7
ccs 2
cts 2
cp 1
c 1
b 0
f 0
rs 9.4285
cc 1
crap 1
1 1
import requests
2
3
4 1
class NewRelicException(Exception):
5 1
    pass
6
7
8 1
class NewRelicDeploymentException(NewRelicException):
9 1
    pass
10
11
12 1
class Deployment(object):
13 1
    ENDPOINT = 'https://api.newrelic.com/v2/applications/%(app_id)s/deployments.json'
14
15 1
    def __init__(self, api_key, app_id, user):
16 1
        self.__api_key = api_key
17 1
        self.__app_id = app_id
18 1
        self.__user = user
19
20 1
    @property
21
    def endpoint(self):
22 1
        return self.ENDPOINT % dict(app_id=self.__app_id)
23
24 1
    @property
25
    def headers(self):
26 1
        return {
27
            'Content-Type': 'application/json',
28
            'X-Api-Key': self.__api_key
29
        }
30
31 1
    def get_payload(self, revision, changelog, description):
32 1
        return {
33
            "deployment" : {
34
                "revision": str(revision),
35
                "changelog": str(changelog),
36
                "description": str(description),
37
                "user": str(self.__user)
38
            }
39
        }
40
41 1
    def deploy(self, revision, changelog, description):
42 1
        payload = self.get_payload(revision, changelog, description)
43 1
        response = requests.post(self.endpoint, headers=self.headers, json=payload)
44
45 1
        if response.status_code != 201:
46 1
            try:
47 1
                raise NewRelicDeploymentException('Recording deployment failed: %s' %
48
                                                  response.json()['error']['title'])
49 1
            except (AttributeError, KeyError):
50 1
                raise NewRelicDeploymentException('Recording deployment failed')
51
52
        return response
53