forked from braintree/braintree_python
-
Notifications
You must be signed in to change notification settings - Fork 1
/
subscription_gateway.py
78 lines (67 loc) · 3.69 KB
/
subscription_gateway.py
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import re
import braintree
from braintree.subscription import Subscription
from braintree.error_result import ErrorResult
from braintree.exceptions.not_found_error import NotFoundError
from braintree.resource import Resource
from braintree.resource_collection import ResourceCollection
from braintree.successful_result import SuccessfulResult
from braintree.transaction import Transaction
class SubscriptionGateway(object):
def __init__(self, gateway):
self.gateway = gateway
self.config = gateway.config
def cancel(self, subscription_id):
response = self.config.http().put("/subscriptions/" + subscription_id + "/cancel")
if "subscription" in response:
return SuccessfulResult({"subscription": Subscription(self.gateway, response["subscription"])})
elif "api_error_response" in response:
return ErrorResult(self.gateway, response["api_error_response"])
def create(self, params={}):
Resource.verify_keys(params, Subscription.create_signature())
response = self.config.http().post("/subscriptions", {"subscription": params})
if "subscription" in response:
return SuccessfulResult({"subscription": Subscription(self.gateway, response["subscription"])})
elif "api_error_response" in response:
return ErrorResult(self.gateway, response["api_error_response"])
def find(self, subscription_id):
try:
response = self.config.http().get("/subscriptions/" + subscription_id)
return Subscription(self.gateway, response["subscription"])
except NotFoundError:
raise NotFoundError("subscription with id " + subscription_id + " not found")
def retry_charge(self, subscription_id, amount=None):
response = self.config.http().post("/transactions", {"transaction": {
"amount": amount,
"subscription_id": subscription_id,
"type": Transaction.Type.Sale
}})
if "transaction" in response:
return SuccessfulResult({"transaction": Transaction(self.gateway, response["transaction"])})
elif "api_error_response" in response:
return ErrorResult(self.gateway, response["api_error_response"])
def search(self, *query):
if isinstance(query[0], list):
query = query[0]
response = self.config.http().post("/subscriptions/advanced_search_ids", {"search": self.__criteria(query)})
return ResourceCollection(query, response, self.__fetch)
def update(self, subscription_id, params={}):
Resource.verify_keys(params, Subscription.update_signature())
response = self.config.http().put("/subscriptions/" + subscription_id, {"subscription": params})
if "subscription" in response:
return SuccessfulResult({"subscription": Subscription(self.gateway, response["subscription"])})
elif "api_error_response" in response:
return ErrorResult(self.gateway, response["api_error_response"])
def __criteria(self, query):
criteria = {}
for term in query:
if criteria.get(term.name):
criteria[term.name] = dict(criteria[term.name].items() + term.to_param().items())
else:
criteria[term.name] = term.to_param()
return criteria
def __fetch(self, query, ids):
criteria = self.__criteria(query)
criteria["ids"] = braintree.subscription_search.SubscriptionSearch.ids.in_list(ids).to_param()
response = self.config.http().post("/subscriptions/advanced_search", {"search": criteria})
return [Subscription(self.gateway, item) for item in ResourceCollection._extract_as_array(response["subscriptions"], "subscription")]