# -*- coding: utf-8 -*-
"""Defines plugins for AiiDA users."""
# pylint: disable=too-few-public-methods,redefined-builtin,unused-argument
from typing import Any, Optional
import graphene as gr
from aiida.orm import User
from aiida_restapi.filter_syntax import parse_filter_str
from .nodes import NodesQuery
from .orm_factories import (
ENTITY_DICT_TYPE,
multirow_cls_factory,
resolve_entity,
single_cls_factory,
)
from .plugins import QueryPlugin
from .utils import FilterString
[docs]class UserQuery(single_cls_factory(User)): # type: ignore[misc]
"""Query an AiiDA User"""
nodes = gr.Field(NodesQuery, filters=FilterString())
[docs] @staticmethod
def resolve_nodes(
parent: Any, info: gr.ResolveInfo, filters: Optional[str] = None
) -> dict:
"""Resolution function."""
# pass filter specification to NodesQuery
parsed_filters = parse_filter_str(filters)
parsed_filters["user_id"] = parent["id"]
return {"filters": parsed_filters}
[docs]def resolve_User(
parent: Any,
info: gr.ResolveInfo,
id: Optional[int] = None,
email: Optional[str] = None,
) -> ENTITY_DICT_TYPE:
"""Resolution function."""
return resolve_entity(User, info, id, email, uuid_name="email")
[docs]class UsersQuery(multirow_cls_factory(UserQuery, User, "users")): # type: ignore[misc]
"""Query all AiiDA Users"""
[docs]def resolve_Users(
parent: Any, info: gr.ResolveInfo, filters: Optional[str] = None
) -> dict:
"""Resolution function."""
# pass filter to UsersQuery
return {"filters": parse_filter_str(filters)}
UserQueryPlugin = QueryPlugin(
"user",
gr.Field(
UserQuery, id=gr.Int(), email=gr.String(), description="Query for a single User"
),
resolve_User,
)
UsersQueryPlugin = QueryPlugin(
"users",
gr.Field(
UsersQuery, description="Query for multiple Users", filters=FilterString()
),
resolve_Users,
)
