def test_register(self):
r = Router()
def func(request):
pass
r.register(r'^test/$', func)
self.assertEqual(r.resolve('test/'), (func, (), {}))
except peewee.IntegrityError:
token = Token.get(device=device, token=apns_token)
try:
with database.transaction():
push = Token.create(device=device, token=push_token, scope=Token.PUSH_SCOPE)
except peewee.IntegrityError:
push = Token.get(device=device, token=push_token)
return RESTResponse(request, {
'device_token': apns_token,
'push_token': push_token,
}, status=status)
router.register(r'^1/devices$', database(RegisterView.as_view()))
class PushView(RESTView):
def get_token(self):
authorization = self.request.headers.get('AUTHORIZATION', None)
if not authorization:
return
if authorization:
_, access_token = authorization.split(' ', 2)
try:
token = Token.get(token=access_token)
except Token.DoesNotExist:
token = None
except peewee.IntegrityError:
token = Token.get(device=device, token=apns_token)
try:
with database.transaction():
push = Token.create(device=device, token=push_token, scope=Token.PUSH_SCOPE)
except peewee.IntegrityError:
push = Token.get(device=device, token=push_token)
return RESTResponse(request, {
'device_token': apns_token,
'push_token': push_token,
}, status=status)
router.register(r'^1/devices$', database(RegisterView.as_view()))
class PermissionRequiredMixin(object):
scope_required = 'all'
def get_token(self):
authorization = self.request.headers.get('AUTHORIZATION', None)
if not authorization:
return
if authorization:
_, access_token = authorization.split(' ', 2)
try:
token = Token.get(token=access_token)
