+xdZddlmZddlmZddlmZddlZej dgdZdZGd d e Z y) ICommon code for 'gcloud firebase test * list-device-capacities' commands.)absolute_import)division)unicode_literalsN CapacityEntrymodelnameversioncapacityz table[box]( model:label=MODEL_ID, name:label=MODEL_NAME, version:label=OS_VERSION_ID, capacity.color(red=None,yellow=Low,green=High):label=DEVICE_CAPACITY ) c*eZdZdZdZedZdZy)DeviceCapacitiesrNc|j|jdjj}|jdjj}|j d|j d|jd|jd|jd|j d|j d|jd|jd|jdi |_|jS)z'A map of enum to user-friendly message.testing_messagesNoneHighMediumLow) _capacity_messages_cachecontextPerAndroidVersionInfoDeviceCapacityValueValuesEnumPerIosVersionInfoDEVICE_CAPACITY_UNSPECIFIEDDEVICE_CAPACITY_HIGHDEVICE_CAPACITY_MEDIUMDEVICE_CAPACITY_LOWDEVICE_CAPACITY_NONE)selfdevice_capacity_enum_androiddevice_capacity_enum_ioss =lib/googlecloudsdk/api_lib/firebase/test/device_capacities.pycapacity_messagesz"DeviceCapacities.capacity_messages*s $$,%)\\ &334Q4Q#"& "//0M0M ' B BF & ; ;V & = =x & : :E & ; ;V " > > " 7 7 " 9 98 " 6 6 " 7 7 'd#  ( ((c @g}|jD]}|jD]{}|j|jvr|j)|j t |j|j|j|j|j}|S)aOGenerate a list of devices/OS versions & corresponding capacity info. Args: catalog: Android or iOS catalog Returns: The list of device models, versions, and capacity info we want to have printed later. Obsolete (unsupported) devices, versions, and entries missing capacity info are filtered out. r) modelsperVersionInfo versionIdsupportedVersionIdsdeviceCapacityappendridr r#)rcatalog capacity_datar version_infos r"get_capacity_dataz"DeviceCapacities.get_capacity_dataDsM..,  ! !)B)B B   & & .  hhZZ$..// 0K0KL  N O /  r$)__name__ __module__ __qualname____doc__rpropertyr#r0r$r"rr%s"Q! ) )2r$r) r4 __future__rrr collections namedtuplerDEVICE_CAPACITY_TABLE_FORMATobjectrr6r$r"r<sC P&'& &&'OQ  9v9r$