Add auth to service mutations

pull/13/head
Inex Code 2022-08-22 23:32:37 +04:00
parent 8c878ea898
commit ab9e8d81e5
1 changed files with 7 additions and 6 deletions

View File

@ -2,6 +2,7 @@
# pylint: disable=too-few-public-methods # pylint: disable=too-few-public-methods
import typing import typing
import strawberry import strawberry
from selfprivacy_api.graphql import IsAuthenticated
from selfprivacy_api.graphql.common_types.jobs import job_to_api_job from selfprivacy_api.graphql.common_types.jobs import job_to_api_job
from selfprivacy_api.graphql.common_types.service import ( from selfprivacy_api.graphql.common_types.service import (
@ -43,7 +44,7 @@ class ServiceJobMutationReturn(GenericJobButationReturn):
class ServicesMutations: class ServicesMutations:
"""Services mutations.""" """Services mutations."""
@strawberry.mutation @strawberry.mutation(permission_classes=[IsAuthenticated])
def enable_service(self, service_id: str) -> ServiceMutationReturn: def enable_service(self, service_id: str) -> ServiceMutationReturn:
"""Enable service.""" """Enable service."""
service = get_service_by_id(service_id) service = get_service_by_id(service_id)
@ -61,7 +62,7 @@ class ServicesMutations:
service=service_to_graphql_service(service), service=service_to_graphql_service(service),
) )
@strawberry.mutation @strawberry.mutation(permission_classes=[IsAuthenticated])
def disable_service(self, service_id: str) -> ServiceMutationReturn: def disable_service(self, service_id: str) -> ServiceMutationReturn:
"""Disable service.""" """Disable service."""
service = get_service_by_id(service_id) service = get_service_by_id(service_id)
@ -79,7 +80,7 @@ class ServicesMutations:
service=service_to_graphql_service(service), service=service_to_graphql_service(service),
) )
@strawberry.mutation @strawberry.mutation(permission_classes=[IsAuthenticated])
def stop_service(self, service_id: str) -> ServiceMutationReturn: def stop_service(self, service_id: str) -> ServiceMutationReturn:
"""Stop service.""" """Stop service."""
service = get_service_by_id(service_id) service = get_service_by_id(service_id)
@ -97,7 +98,7 @@ class ServicesMutations:
service=service_to_graphql_service(service), service=service_to_graphql_service(service),
) )
@strawberry.mutation @strawberry.mutation(permission_classes=[IsAuthenticated])
def start_service(self, service_id: str) -> ServiceMutationReturn: def start_service(self, service_id: str) -> ServiceMutationReturn:
"""Start service.""" """Start service."""
service = get_service_by_id(service_id) service = get_service_by_id(service_id)
@ -115,7 +116,7 @@ class ServicesMutations:
service=service_to_graphql_service(service), service=service_to_graphql_service(service),
) )
@strawberry.mutation @strawberry.mutation(permission_classes=[IsAuthenticated])
def restart_service(self, service_id: str) -> ServiceMutationReturn: def restart_service(self, service_id: str) -> ServiceMutationReturn:
"""Restart service.""" """Restart service."""
service = get_service_by_id(service_id) service = get_service_by_id(service_id)
@ -133,7 +134,7 @@ class ServicesMutations:
service=service_to_graphql_service(service), service=service_to_graphql_service(service),
) )
@strawberry.mutation @strawberry.mutation(permission_classes=[IsAuthenticated])
def move_service(self, input: MoveServiceInput) -> ServiceJobMutationReturn: def move_service(self, input: MoveServiceInput) -> ServiceJobMutationReturn:
"""Move service.""" """Move service."""
service = get_service_by_id(input.service_id) service = get_service_by_id(input.service_id)