selfprivacy-rest-api/selfprivacy_api/graphql/common_types/user.py

import typing
import strawberry
from selfprivacy_api.graphql.mutations.mutation_interface import (
    MutationReturnInterface,
)
from enum import Enum


@strawberry.enum
class UserType(Enum):
    NORMAL = "NORMAL"
    PRIMARY = "PRIMARY"
    ROOT = "ROOT"


@strawberry.type
class User:
    """Users management"""

    user_type: UserType
    username: str
    # userHomeFolderspace: UserHomeFolderUsage
    ssh_keys: typing.List[str] = []


@strawberry.type
class UserMutationReturn(MutationReturnInterface):
    """Return type for user mutation"""

    user: typing.Optional[User]