from argparse import _SubParsersAction
from functools import partial
from typing import List
from q2_sdk.hq.table_row import TableRow
from .db_object import DbObject
[docs]
class ZoneRow(TableRow):
ZoneID: int
Description: str
IsProduction: bool
ZoneHourID: int
TimeZone: str
ShortName: str
ABA: str
ZonePasswordPolicyBundleID: int
[docs]
class Zone(DbObject):
"Q2_Zone table"
NAME = "Zone"
REPRESENTATION_ROW_CLASS = ZoneRow
[docs]
def add_arguments(self, parser: _SubParsersAction):
subparser = parser.add_parser("get_zones")
subparser.set_defaults(parser="get")
subparser.set_defaults(func=partial(self.get, serialize_for_cli=True))
[docs]
async def get(self, serialize_for_cli=False) -> List[ZoneRow]:
response = await self.call_hq("sdk_GetZones")
if serialize_for_cli:
columns = ["ZoneID", "Description", "TimeZone", "ShortName", "ABA"]
response = self.serialize_for_cli(response, columns)
return response