vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

fix: validate token_type, use cstr

Browse source
This commit is contained in:
barredterra 2020-09-28 11:01:16 +02:00
parent 60cb523439
commit ea0ceae42c
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
frappe/integrations/doctype/token_cache/token_cache.py
View file

@ -3,10 +3,11 @@
# For license information, please see license.txt
from __future__ import unicode_literals
import frappe
import requests
from urllib.parse import urlencode
from datetime import datetime, timedelta
import frappe
from frappe import _
from frappe.utils import cstr
from frappe.model.document import Document
class TokenCache(Document):
@ -19,10 +20,16 @@ class TokenCache(Document):
raise frappe.exceptions.DoesNotExistError
def update_data(self, data):
self.access_token = data.get('access_token')
self.refresh_token = data.get('refresh_token')
self.expires_in = data.get('expires_in')
self.token_type = data.get('token_type')
token_type = cstr(data.get('token_type', '')).lower()
if token_type not in ['bearer', 'mac']:
frappe.throw(_('Received an invalid token type.'))
# 'Bearer' or 'MAC'
token_type = token_type.title() if token_type == 'bearer' else token_type.upper()
self.token_type = token_type
self.access_token = cstr(data.get('access_token', ''))
self.refresh_token = cstr(data.get('refresh_token', ''))
self.expires_in = cstr(data.get('expires_in', ''))
new_scopes = data.get('scope')
if new_scopes: