Skip to content
Snippets Groups Projects
Commit d6211984 authored by Hangzhi Yu's avatar Hangzhi Yu
Browse files

Fix checking mechanism for session attribute

parent 8f81d7f0
No related branches found
No related tags found
2 merge requests!3Draft: Resolve "Implement shared secret mechanism",!2Frontend
......@@ -25,7 +25,7 @@ class SharedSecretObjectType(DjangoObjectType):
@classmethod
def get_queryset(cls, queryset, info):
if not info.context.session["maka_shared_secret_correct"]:
if not info.context.session.get("maka_shared_secret_correct"):
raise PermissionDenied()
return super().get_queryset(queryset, info)
......@@ -40,7 +40,7 @@ class SharedSecretBatchCreateMixin:
@classmethod
def check_permissions(cls, root, info, input, *args, **kwargs): # noqa: A002
if info.context.session["maka_shared_secret_correct"]:
if info.context.session.get("maka_shared_secret_correct"):
return
raise PermissionDenied()
......@@ -54,7 +54,7 @@ class SharedSecretBatchPatchMixin:
@classmethod
def check_permissions(cls, root, info, input, *args, **kwargs): # noqa: A002
if info.context.session["maka_shared_secret_correct"]:
if info.context.session.get("maka_shared_secret_correct"):
return
raise PermissionDenied()
......@@ -68,7 +68,7 @@ class SharedSecretBatchDeleteMixin:
@classmethod
def check_permissions(cls, root, info, input, *args, **kwargs): # noqa: A002
if info.context.session["maka_shared_secret_correct"]:
if info.context.session.get("maka_shared_secret_correct"):
return
raise PermissionDenied()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment