from argparse import _SubParsersAction
from functools import partial
from q2_sdk.core.dynamic_imports import (
api_ExecuteStoredProcedure as ExecuteStoredProcedure,
)
from q2_sdk.tools.decorators import dev_only
from .db_object import DbObject
[docs]
class UserAccountAttribute(DbObject):
GET_BY_NAME_KEY = "ShortName"
NAME = "UserAccountAttribute"
[docs]
def add_arguments(self, parser: _SubParsersAction):
subparser = parser.add_parser("get_user_account_attributes")
subparser.set_defaults(parser="get_user_account_attributes")
subparser.set_defaults(func=partial(self.get, serialize_for_cli=True))
subparser = parser.add_parser("add_user_account_attribute")
subparser.set_defaults(parser="add_user_account_attribute")
subparser.set_defaults(func=partial(self.create))
subparser.add_argument("short_name", help="Q2_UserAccountAttribute.ShortName")
subparser = parser.add_parser("remove_user_account_attribute")
subparser.set_defaults(parser="remove_user_account_attribute")
subparser.set_defaults(func=partial(self.delete))
subparser.add_argument("short_name", help="Q2_UserAccountAttribute.ShortName")
[docs]
async def get(self, serialize_for_cli=False):
response = await self.call_hq("sdk_GetUserAccountAttributes")
if serialize_for_cli:
response = self.serialize_for_cli(
response, ["UserAccountAttributeID", "ShortName"]
)
return response
[docs]
async def create(self, short_name):
return await self.call_hq(
"sdk_AddUserAccountAttribute",
ExecuteStoredProcedure.SqlParameters([
ExecuteStoredProcedure.SqlParam(
ExecuteStoredProcedure.DataType.VarChar,
"short_name",
short_name,
)
]),
)
[docs]
@dev_only
async def delete(self, short_name):
"""Note: this only works in the dev environment"""
return await self.call_hq(
"sdk_RemoveUserAccountAttribute",
ExecuteStoredProcedure.SqlParameters([
ExecuteStoredProcedure.SqlParam(
ExecuteStoredProcedure.DataType.VarChar,
"short_name",
short_name,
)
]),
)