Differential D7445 Diff 25666 looper/views/tokens.py

Changeset View

Standalone View

looper/views/tokens.py

	from typing import List, NewType, cast			from typing import List, NewType, cast
				import datetime
				import logging

				import dateutil.parser
	from django.http.request import HttpRequest			from django.http.request import HttpRequest

	from looper.models import Customer, Gateway			from looper.models import Customer, Gateway

	BraintreeClientToken = NewType('BraintreeClientToken', str)			BraintreeClientToken = NewType('BraintreeClientToken', str)
	SESSION_KEY_PREFIX: str = 'PAYMENT_GATEWAY_CLIENT_TOKEN_'			SESSION_KEY_PREFIX: str = 'PAYMENT_GATEWAY_CLIENT_TOKEN_'


				log = logging.getLogger(__name__)


	def get_or_set_braintree_client_token(			def get_or_set_braintree_client_token(
	request: HttpRequest, customer: Customer, currency: str			request: HttpRequest, customer: Customer, currency: str
	) -> BraintreeClientToken:			) -> BraintreeClientToken:
	"""Generate a Braintree client token, caching it in the current session."""			"""Generate a Braintree client token, caching it in the current session."""
	key = f'{SESSION_KEY_PREFIX}_braintree_{currency}'			key = f'{SESSION_KEY_PREFIX}_braintree_{currency}'
	client_token: str = request.session.get(key, '')			key_expiry = f'{key}_expire'
	if client_token:
				try:
				client_token = request.session[key]
				token_expires = request.session[key_expiry]
				except KeyError:
				pass
				else:
				expiry_timestamp = dateutil.parser.parse(token_expires)
				if expiry_timestamp > datetime.datetime.now():
				# Not yet expired, use the client token.
	return BraintreeClientToken(client_token)			return BraintreeClientToken(client_token)

	# TODO(Sybren): handle Gateway errors.			# TODO(Sybren): handle Gateway errors.
	braintree_gateway = Gateway.objects.get(name__in={'braintree', 'mock'})			braintree_gateway = Gateway.objects.get(name__in={'braintree', 'mock'})
	gateway_customer_id = customer.gateway_customer_id(braintree_gateway)			gateway_customer_id = customer.gateway_customer_id(braintree_gateway)
				log.info("Generating new client token for %s" % gateway_customer_id)
	client_token = braintree_gateway.provider.generate_client_token(currency, gateway_customer_id)			client_token = braintree_gateway.provider.generate_client_token(currency, gateway_customer_id)

				expiry_timestamp = datetime.datetime.now() + datetime.timedelta(minutes=10)
				sybrenUnsubmitted Done Inline Actions `datetime.timedelta(minutes=10)` should be stored in a 'constant' at the top of the file, so it can get a descriptive name, something like `GATEWAY_TOKEN_EXPIRY`. sybren: `datetime.timedelta(minutes=10)` should be stored in a 'constant' at the top of the file, so it…
				request.session[key_expiry] = expiry_timestamp.isoformat()
	request.session[key] = client_token			request.session[key] = client_token
	return BraintreeClientToken(client_token)			return BraintreeClientToken(client_token)


	def clear_client_tokens(request: HttpRequest) -> None:			def clear_client_tokens(request: HttpRequest) -> None:
	"""Erase the client tokens from the current session."""			"""Erase the client tokens from the current session."""
	keys: List[str] = list(request.session.keys()) # type: ignore[no-untyped-call]			keys: List[str] = list(request.session.keys()) # type: ignore[no-untyped-call]
	for key in keys:			for key in keys:
	if key.startswith(SESSION_KEY_PREFIX):			if key.startswith(SESSION_KEY_PREFIX):
	del request.session[key]			del request.session[key]