android-env 1.2.2__py3-none-any.whl → 1.2.3__py3-none-any.whl

android_env/proto/snapshot.proto

@@ -1,4 +1,4 @@
-// Copyright 2023 DeepMind Technologies Limited.
+// Copyright 2024 DeepMind Technologies Limited.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.

android_env/proto/snapshot_pb2.py

@@ -1,11 +1,22 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
 # source: android_env/proto/snapshot.proto
+# Protobuf Python Version: 5.29.0
 """Generated protocol buffer code."""
-from google.protobuf.internal import builder as _builder
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'android_env/proto/snapshot.proto'
+)
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -15,22 +26,22 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n android_env/proto/snapshot.proto\x12\x11\x65mulator_snapshot\"\xaf\x03\n\x05Image\x12+\n\x04type\x18\x01 \x01(\x0e\x32\x1d.emulator_snapshot.Image.Type\x12\x0c\n\x04path\x18\x02 \x01(\t\x12\x0f\n\x07present\x18\x03 \x01(\x08\x12\x0c\n\x04size\x18\x04 \x01(\x03\x12\x19\n\x11modification_time\x18\x05 \x01(\x03\"\xb0\x02\n\x04Type\x12\x16\n\x12IMAGE_TYPE_UNKNOWN\x10\x00\x12\x15\n\x11IMAGE_TYPE_KERNEL\x10\x01\x12\x1c\n\x18IMAGE_TYPE_KERNEL_RANCHU\x10\x02\x12\x15\n\x11IMAGE_TYPE_SYSTEM\x10\x03\x12\x1a\n\x16IMAGE_TYPE_SYSTEM_COPY\x10\x04\x12\x13\n\x0fIMAGE_TYPE_DATA\x10\x05\x12\x18\n\x14IMAGE_TYPE_DATA_COPY\x10\x06\x12\x16\n\x12IMAGE_TYPE_RAMDISK\x10\x07\x12\x15\n\x11IMAGE_TYPE_SDCARD\x10\x08\x12\x14\n\x10IMAGE_TYPE_CACHE\x10\t\x12\x15\n\x11IMAGE_TYPE_VENDOR\x10\n\x12\x1d\n\x19IMAGE_TYPE_ENCRYPTION_KEY\x10\x0b\".\n\x04Host\x12\x12\n\ngpu_driver\x18\x04 \x01(\t\x12\x12\n\nhypervisor\x18\x05 \x01(\x05\"m\n\x06\x43onfig\x12\x18\n\x10\x65nabled_features\x18\x01 \x03(\x05\x12\x19\n\x11selected_renderer\x18\x02 \x01(\x05\x12\x16\n\x0e\x63pu_core_count\x18\x03 \x01(\x05\x12\x16\n\x0eram_size_bytes\x18\x04 \x01(\x03\"M\n\tSaveStats\x12\x13\n\x0bincremental\x18\x01 \x01(\r\x12\x10\n\x08\x64uration\x18\x02 \x01(\x04\x12\x19\n\x11ram_changed_bytes\x18\x03 \x01(\x04\"\x8d\x04\n\x08Snapshot\x12\x0f\n\x07version\x18\x01 \x01(\x05\x12\x15\n\rcreation_time\x18\x02 \x01(\x03\x12(\n\x06images\x18\x03 \x03(\x0b\x32\x18.emulator_snapshot.Image\x12%\n\x04host\x18\x04 \x01(\x0b\x32\x17.emulator_snapshot.Host\x12)\n\x06\x63onfig\x18\x05 \x01(\x0b\x32\x19.emulator_snapshot.Config\x12\"\n\x1a\x66\x61iled_to_load_reason_code\x18\x07 \x01(\x03\x12$\n\x1cguest_data_partition_mounted\x18\x08 \x01(\x08\x12\x10\n\x08rotation\x18\t \x01(\x05\x12\x15\n\rinvalid_loads\x18\n \x01(\x05\x12\x18\n\x10successful_loads\x18\x0b \x01(\x05\x12\x14\n\x0clogical_name\x18\x0c \x01(\t\x12\x0e\n\x06parent\x18\r \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x0e \x01(\t\x12\x30\n\nsave_stats\x18\x0f \x03(\x0b\x32\x1c.emulator_snapshot.SaveStats\x12\x0e\n\x06\x66olded\x18\x10 \x01(\x08\x12\x19\n\x11launch_parameters\x18\x11 \x03(\t\x12\x19\n\x11\x65mulator_build_id\x18\x12 \x01(\t\x12\x1d\n\x15system_image_build_id\x18\x13 \x01(\tB\x1f\n\x1d\x63om.android.emulator.snapshot')
 
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'android_env.proto.snapshot_pb2', globals())
-if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\035com.android.emulator.snapshot'
-  _IMAGE._serialized_start=56
-  _IMAGE._serialized_end=487
-  _IMAGE_TYPE._serialized_start=183
-  _IMAGE_TYPE._serialized_end=487
-  _HOST._serialized_start=489
-  _HOST._serialized_end=535
-  _CONFIG._serialized_start=537
-  _CONFIG._serialized_end=646
-  _SAVESTATS._serialized_start=648
-  _SAVESTATS._serialized_end=725
-  _SNAPSHOT._serialized_start=728
-  _SNAPSHOT._serialized_end=1253
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'android_env.proto.snapshot_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\035com.android.emulator.snapshot'
+  _globals['_IMAGE']._serialized_start=56
+  _globals['_IMAGE']._serialized_end=487
+  _globals['_IMAGE_TYPE']._serialized_start=183
+  _globals['_IMAGE_TYPE']._serialized_end=487
+  _globals['_HOST']._serialized_start=489
+  _globals['_HOST']._serialized_end=535
+  _globals['_CONFIG']._serialized_start=537
+  _globals['_CONFIG']._serialized_end=646
+  _globals['_SAVESTATS']._serialized_start=648
+  _globals['_SAVESTATS']._serialized_end=725
+  _globals['_SNAPSHOT']._serialized_start=728
+  _globals['_SNAPSHOT']._serialized_end=1253
 # @@protoc_insertion_point(module_scope)

android_env/proto/snapshot_pb2.pyi

@@ -0,0 +1,117 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class Image(_message.Message):
+    __slots__ = ("type", "path", "present", "size", "modification_time")
+    class Type(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        IMAGE_TYPE_UNKNOWN: _ClassVar[Image.Type]
+        IMAGE_TYPE_KERNEL: _ClassVar[Image.Type]
+        IMAGE_TYPE_KERNEL_RANCHU: _ClassVar[Image.Type]
+        IMAGE_TYPE_SYSTEM: _ClassVar[Image.Type]
+        IMAGE_TYPE_SYSTEM_COPY: _ClassVar[Image.Type]
+        IMAGE_TYPE_DATA: _ClassVar[Image.Type]
+        IMAGE_TYPE_DATA_COPY: _ClassVar[Image.Type]
+        IMAGE_TYPE_RAMDISK: _ClassVar[Image.Type]
+        IMAGE_TYPE_SDCARD: _ClassVar[Image.Type]
+        IMAGE_TYPE_CACHE: _ClassVar[Image.Type]
+        IMAGE_TYPE_VENDOR: _ClassVar[Image.Type]
+        IMAGE_TYPE_ENCRYPTION_KEY: _ClassVar[Image.Type]
+    IMAGE_TYPE_UNKNOWN: Image.Type
+    IMAGE_TYPE_KERNEL: Image.Type
+    IMAGE_TYPE_KERNEL_RANCHU: Image.Type
+    IMAGE_TYPE_SYSTEM: Image.Type
+    IMAGE_TYPE_SYSTEM_COPY: Image.Type
+    IMAGE_TYPE_DATA: Image.Type
+    IMAGE_TYPE_DATA_COPY: Image.Type
+    IMAGE_TYPE_RAMDISK: Image.Type
+    IMAGE_TYPE_SDCARD: Image.Type
+    IMAGE_TYPE_CACHE: Image.Type
+    IMAGE_TYPE_VENDOR: Image.Type
+    IMAGE_TYPE_ENCRYPTION_KEY: Image.Type
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    PATH_FIELD_NUMBER: _ClassVar[int]
+    PRESENT_FIELD_NUMBER: _ClassVar[int]
+    SIZE_FIELD_NUMBER: _ClassVar[int]
+    MODIFICATION_TIME_FIELD_NUMBER: _ClassVar[int]
+    type: Image.Type
+    path: str
+    present: bool
+    size: int
+    modification_time: int
+    def __init__(self, type: _Optional[_Union[Image.Type, str]] = ..., path: _Optional[str] = ..., present: bool = ..., size: _Optional[int] = ..., modification_time: _Optional[int] = ...) -> None: ...
+
+class Host(_message.Message):
+    __slots__ = ("gpu_driver", "hypervisor")
+    GPU_DRIVER_FIELD_NUMBER: _ClassVar[int]
+    HYPERVISOR_FIELD_NUMBER: _ClassVar[int]
+    gpu_driver: str
+    hypervisor: int
+    def __init__(self, gpu_driver: _Optional[str] = ..., hypervisor: _Optional[int] = ...) -> None: ...
+
+class Config(_message.Message):
+    __slots__ = ("enabled_features", "selected_renderer", "cpu_core_count", "ram_size_bytes")
+    ENABLED_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    SELECTED_RENDERER_FIELD_NUMBER: _ClassVar[int]
+    CPU_CORE_COUNT_FIELD_NUMBER: _ClassVar[int]
+    RAM_SIZE_BYTES_FIELD_NUMBER: _ClassVar[int]
+    enabled_features: _containers.RepeatedScalarFieldContainer[int]
+    selected_renderer: int
+    cpu_core_count: int
+    ram_size_bytes: int
+    def __init__(self, enabled_features: _Optional[_Iterable[int]] = ..., selected_renderer: _Optional[int] = ..., cpu_core_count: _Optional[int] = ..., ram_size_bytes: _Optional[int] = ...) -> None: ...
+
+class SaveStats(_message.Message):
+    __slots__ = ("incremental", "duration", "ram_changed_bytes")
+    INCREMENTAL_FIELD_NUMBER: _ClassVar[int]
+    DURATION_FIELD_NUMBER: _ClassVar[int]
+    RAM_CHANGED_BYTES_FIELD_NUMBER: _ClassVar[int]
+    incremental: int
+    duration: int
+    ram_changed_bytes: int
+    def __init__(self, incremental: _Optional[int] = ..., duration: _Optional[int] = ..., ram_changed_bytes: _Optional[int] = ...) -> None: ...
+
+class Snapshot(_message.Message):
+    __slots__ = ("version", "creation_time", "images", "host", "config", "failed_to_load_reason_code", "guest_data_partition_mounted", "rotation", "invalid_loads", "successful_loads", "logical_name", "parent", "description", "save_stats", "folded", "launch_parameters", "emulator_build_id", "system_image_build_id")
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    CREATION_TIME_FIELD_NUMBER: _ClassVar[int]
+    IMAGES_FIELD_NUMBER: _ClassVar[int]
+    HOST_FIELD_NUMBER: _ClassVar[int]
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    FAILED_TO_LOAD_REASON_CODE_FIELD_NUMBER: _ClassVar[int]
+    GUEST_DATA_PARTITION_MOUNTED_FIELD_NUMBER: _ClassVar[int]
+    ROTATION_FIELD_NUMBER: _ClassVar[int]
+    INVALID_LOADS_FIELD_NUMBER: _ClassVar[int]
+    SUCCESSFUL_LOADS_FIELD_NUMBER: _ClassVar[int]
+    LOGICAL_NAME_FIELD_NUMBER: _ClassVar[int]
+    PARENT_FIELD_NUMBER: _ClassVar[int]
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    SAVE_STATS_FIELD_NUMBER: _ClassVar[int]
+    FOLDED_FIELD_NUMBER: _ClassVar[int]
+    LAUNCH_PARAMETERS_FIELD_NUMBER: _ClassVar[int]
+    EMULATOR_BUILD_ID_FIELD_NUMBER: _ClassVar[int]
+    SYSTEM_IMAGE_BUILD_ID_FIELD_NUMBER: _ClassVar[int]
+    version: int
+    creation_time: int
+    images: _containers.RepeatedCompositeFieldContainer[Image]
+    host: Host
+    config: Config
+    failed_to_load_reason_code: int
+    guest_data_partition_mounted: bool
+    rotation: int
+    invalid_loads: int
+    successful_loads: int
+    logical_name: str
+    parent: str
+    description: str
+    save_stats: _containers.RepeatedCompositeFieldContainer[SaveStats]
+    folded: bool
+    launch_parameters: _containers.RepeatedScalarFieldContainer[str]
+    emulator_build_id: str
+    system_image_build_id: str
+    def __init__(self, version: _Optional[int] = ..., creation_time: _Optional[int] = ..., images: _Optional[_Iterable[_Union[Image, _Mapping]]] = ..., host: _Optional[_Union[Host, _Mapping]] = ..., config: _Optional[_Union[Config, _Mapping]] = ..., failed_to_load_reason_code: _Optional[int] = ..., guest_data_partition_mounted: bool = ..., rotation: _Optional[int] = ..., invalid_loads: _Optional[int] = ..., successful_loads: _Optional[int] = ..., logical_name: _Optional[str] = ..., parent: _Optional[str] = ..., description: _Optional[str] = ..., save_stats: _Optional[_Iterable[_Union[SaveStats, _Mapping]]] = ..., folded: bool = ..., launch_parameters: _Optional[_Iterable[str]] = ..., emulator_build_id: _Optional[str] = ..., system_image_build_id: _Optional[str] = ...) -> None: ...

android_env/proto/snapshot_pb2_grpc.py

@@ -1,4 +1,24 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
 
+
+GRPC_GENERATED_VERSION = '1.71.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in android_env/proto/snapshot_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

android_env/proto/snapshot_service.proto

@@ -1,4 +1,4 @@
-// Copyright 2023 DeepMind Technologies Limited.
+// Copyright 2024 DeepMind Technologies Limited.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.

android_env/proto/snapshot_service_pb2.py

@@ -1,11 +1,22 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
 # source: android_env/proto/snapshot_service.proto
+# Protobuf Python Version: 5.29.0
 """Generated protocol buffer code."""
-from google.protobuf.internal import builder as _builder
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'android_env/proto/snapshot_service.proto'
+)
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -16,28 +27,28 @@ from android_env.proto import snapshot_pb2 as android__env_dot_proto_dot_snapsho
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(android_env/proto/snapshot_service.proto\x12\x19\x61ndroid.emulation.control\x1a android_env/proto/snapshot.proto\"\xd3\x01\n\x0fSnapshotPackage\x12\x13\n\x0bsnapshot_id\x18\x01 \x01(\t\x12\x0f\n\x07payload\x18\x02 \x01(\x0c\x12\x0f\n\x07success\x18\x03 \x01(\x08\x12\x0b\n\x03\x65rr\x18\x04 \x01(\x0c\x12\x41\n\x06\x66ormat\x18\x05 \x01(\x0e\x32\x31.android.emulation.control.SnapshotPackage.Format\x12\x0c\n\x04path\x18\x06 \x01(\t\"+\n\x06\x46ormat\x12\t\n\x05TARGZ\x10\x00\x12\x07\n\x03TAR\x10\x01\x12\r\n\tDIRECTORY\x10\x02\"\x87\x01\n\x0eSnapshotFilter\x12J\n\x0cstatusFilter\x18\x01 \x01(\x0e\x32\x34.android.emulation.control.SnapshotFilter.LoadStatus\")\n\nLoadStatus\x12\x12\n\x0e\x43ompatibleOnly\x10\x00\x12\x07\n\x03\x41ll\x10\x01\"\xe5\x01\n\x0fSnapshotDetails\x12\x13\n\x0bsnapshot_id\x18\x01 \x01(\t\x12,\n\x07\x64\x65tails\x18\x02 \x01(\x0b\x32\x1b.emulator_snapshot.Snapshot\x12\x45\n\x06status\x18\x03 \x01(\x0e\x32\x35.android.emulation.control.SnapshotDetails.LoadStatus\x12\x0c\n\x04size\x18\x04 \x01(\x04\":\n\nLoadStatus\x12\x0e\n\nCompatible\x10\x00\x12\x10\n\x0cIncompatible\x10\x01\x12\n\n\x06Loaded\x10\x02\"M\n\x0cSnapshotList\x12=\n\tsnapshots\x18\x01 \x03(\x0b\x32*.android.emulation.control.SnapshotDetails\"\x80\x01\n\x0cIceboxTarget\x12\x0b\n\x03pid\x18\x01 \x01(\x03\x12\x14\n\x0cpackage_name\x18\x02 \x01(\t\x12\x13\n\x0bsnapshot_id\x18\x03 \x01(\t\x12\x0e\n\x06\x66\x61iled\x18\x04 \x01(\x08\x12\x0b\n\x03\x65rr\x18\x05 \x01(\t\x12\x1b\n\x13max_snapshot_number\x18\x06 \x01(\x05\x32\xf4\x05\n\x0fSnapshotService\x12\x65\n\rListSnapshots\x12).android.emulation.control.SnapshotFilter\x1a\'.android.emulation.control.SnapshotList\"\x00\x12j\n\x0cPullSnapshot\x12*.android.emulation.control.SnapshotPackage\x1a*.android.emulation.control.SnapshotPackage\"\x00\x30\x01\x12j\n\x0cPushSnapshot\x12*.android.emulation.control.SnapshotPackage\x1a*.android.emulation.control.SnapshotPackage\"\x00(\x01\x12h\n\x0cLoadSnapshot\x12*.android.emulation.control.SnapshotPackage\x1a*.android.emulation.control.SnapshotPackage\"\x00\x12h\n\x0cSaveSnapshot\x12*.android.emulation.control.SnapshotPackage\x1a*.android.emulation.control.SnapshotPackage\"\x00\x12j\n\x0e\x44\x65leteSnapshot\x12*.android.emulation.control.SnapshotPackage\x1a*.android.emulation.control.SnapshotPackage\"\x00\x12\x62\n\x0cTrackProcess\x12\'.android.emulation.control.IceboxTarget\x1a\'.android.emulation.control.IceboxTarget\"\x00\x42&\n\x1c\x63om.android.emulator.controlP\x01\xa2\x02\x03\x41\x45\x43\x62\x06proto3')
 
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'android_env.proto.snapshot_service_pb2', globals())
-if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\034com.android.emulator.controlP\001\242\002\003AEC'
-  _SNAPSHOTPACKAGE._serialized_start=106
-  _SNAPSHOTPACKAGE._serialized_end=317
-  _SNAPSHOTPACKAGE_FORMAT._serialized_start=274
-  _SNAPSHOTPACKAGE_FORMAT._serialized_end=317
-  _SNAPSHOTFILTER._serialized_start=320
-  _SNAPSHOTFILTER._serialized_end=455
-  _SNAPSHOTFILTER_LOADSTATUS._serialized_start=414
-  _SNAPSHOTFILTER_LOADSTATUS._serialized_end=455
-  _SNAPSHOTDETAILS._serialized_start=458
-  _SNAPSHOTDETAILS._serialized_end=687
-  _SNAPSHOTDETAILS_LOADSTATUS._serialized_start=629
-  _SNAPSHOTDETAILS_LOADSTATUS._serialized_end=687
-  _SNAPSHOTLIST._serialized_start=689
-  _SNAPSHOTLIST._serialized_end=766
-  _ICEBOXTARGET._serialized_start=769
-  _ICEBOXTARGET._serialized_end=897
-  _SNAPSHOTSERVICE._serialized_start=900
-  _SNAPSHOTSERVICE._serialized_end=1656
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'android_env.proto.snapshot_service_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\034com.android.emulator.controlP\001\242\002\003AEC'
+  _globals['_SNAPSHOTPACKAGE']._serialized_start=106
+  _globals['_SNAPSHOTPACKAGE']._serialized_end=317
+  _globals['_SNAPSHOTPACKAGE_FORMAT']._serialized_start=274
+  _globals['_SNAPSHOTPACKAGE_FORMAT']._serialized_end=317
+  _globals['_SNAPSHOTFILTER']._serialized_start=320
+  _globals['_SNAPSHOTFILTER']._serialized_end=455
+  _globals['_SNAPSHOTFILTER_LOADSTATUS']._serialized_start=414
+  _globals['_SNAPSHOTFILTER_LOADSTATUS']._serialized_end=455
+  _globals['_SNAPSHOTDETAILS']._serialized_start=458
+  _globals['_SNAPSHOTDETAILS']._serialized_end=687
+  _globals['_SNAPSHOTDETAILS_LOADSTATUS']._serialized_start=629
+  _globals['_SNAPSHOTDETAILS_LOADSTATUS']._serialized_end=687
+  _globals['_SNAPSHOTLIST']._serialized_start=689
+  _globals['_SNAPSHOTLIST']._serialized_end=766
+  _globals['_ICEBOXTARGET']._serialized_start=769
+  _globals['_ICEBOXTARGET']._serialized_end=897
+  _globals['_SNAPSHOTSERVICE']._serialized_start=900
+  _globals['_SNAPSHOTSERVICE']._serialized_end=1656
 # @@protoc_insertion_point(module_scope)

android_env/proto/snapshot_service_pb2.pyi

@@ -0,0 +1,86 @@
+from android_env.proto import snapshot_pb2 as _snapshot_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class SnapshotPackage(_message.Message):
+    __slots__ = ("snapshot_id", "payload", "success", "err", "format", "path")
+    class Format(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        TARGZ: _ClassVar[SnapshotPackage.Format]
+        TAR: _ClassVar[SnapshotPackage.Format]
+        DIRECTORY: _ClassVar[SnapshotPackage.Format]
+    TARGZ: SnapshotPackage.Format
+    TAR: SnapshotPackage.Format
+    DIRECTORY: SnapshotPackage.Format
+    SNAPSHOT_ID_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    SUCCESS_FIELD_NUMBER: _ClassVar[int]
+    ERR_FIELD_NUMBER: _ClassVar[int]
+    FORMAT_FIELD_NUMBER: _ClassVar[int]
+    PATH_FIELD_NUMBER: _ClassVar[int]
+    snapshot_id: str
+    payload: bytes
+    success: bool
+    err: bytes
+    format: SnapshotPackage.Format
+    path: str
+    def __init__(self, snapshot_id: _Optional[str] = ..., payload: _Optional[bytes] = ..., success: bool = ..., err: _Optional[bytes] = ..., format: _Optional[_Union[SnapshotPackage.Format, str]] = ..., path: _Optional[str] = ...) -> None: ...
+
+class SnapshotFilter(_message.Message):
+    __slots__ = ("statusFilter",)
+    class LoadStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        CompatibleOnly: _ClassVar[SnapshotFilter.LoadStatus]
+        All: _ClassVar[SnapshotFilter.LoadStatus]
+    CompatibleOnly: SnapshotFilter.LoadStatus
+    All: SnapshotFilter.LoadStatus
+    STATUSFILTER_FIELD_NUMBER: _ClassVar[int]
+    statusFilter: SnapshotFilter.LoadStatus
+    def __init__(self, statusFilter: _Optional[_Union[SnapshotFilter.LoadStatus, str]] = ...) -> None: ...
+
+class SnapshotDetails(_message.Message):
+    __slots__ = ("snapshot_id", "details", "status", "size")
+    class LoadStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        Compatible: _ClassVar[SnapshotDetails.LoadStatus]
+        Incompatible: _ClassVar[SnapshotDetails.LoadStatus]
+        Loaded: _ClassVar[SnapshotDetails.LoadStatus]
+    Compatible: SnapshotDetails.LoadStatus
+    Incompatible: SnapshotDetails.LoadStatus
+    Loaded: SnapshotDetails.LoadStatus
+    SNAPSHOT_ID_FIELD_NUMBER: _ClassVar[int]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    SIZE_FIELD_NUMBER: _ClassVar[int]
+    snapshot_id: str
+    details: _snapshot_pb2.Snapshot
+    status: SnapshotDetails.LoadStatus
+    size: int
+    def __init__(self, snapshot_id: _Optional[str] = ..., details: _Optional[_Union[_snapshot_pb2.Snapshot, _Mapping]] = ..., status: _Optional[_Union[SnapshotDetails.LoadStatus, str]] = ..., size: _Optional[int] = ...) -> None: ...
+
+class SnapshotList(_message.Message):
+    __slots__ = ("snapshots",)
+    SNAPSHOTS_FIELD_NUMBER: _ClassVar[int]
+    snapshots: _containers.RepeatedCompositeFieldContainer[SnapshotDetails]
+    def __init__(self, snapshots: _Optional[_Iterable[_Union[SnapshotDetails, _Mapping]]] = ...) -> None: ...
+
+class IceboxTarget(_message.Message):
+    __slots__ = ("pid", "package_name", "snapshot_id", "failed", "err", "max_snapshot_number")
+    PID_FIELD_NUMBER: _ClassVar[int]
+    PACKAGE_NAME_FIELD_NUMBER: _ClassVar[int]
+    SNAPSHOT_ID_FIELD_NUMBER: _ClassVar[int]
+    FAILED_FIELD_NUMBER: _ClassVar[int]
+    ERR_FIELD_NUMBER: _ClassVar[int]
+    MAX_SNAPSHOT_NUMBER_FIELD_NUMBER: _ClassVar[int]
+    pid: int
+    package_name: str
+    snapshot_id: str
+    failed: bool
+    err: str
+    max_snapshot_number: int
+    def __init__(self, pid: _Optional[int] = ..., package_name: _Optional[str] = ..., snapshot_id: _Optional[str] = ..., failed: bool = ..., err: _Optional[str] = ..., max_snapshot_number: _Optional[int] = ...) -> None: ...

android_env/proto/snapshot_service_pb2_grpc.py

@@ -1,9 +1,29 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
 
 from android_env.proto import snapshot_service_pb2 as android__env_dot_proto_dot_snapshot__service__pb2
 
+GRPC_GENERATED_VERSION = '1.71.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in android_env/proto/snapshot_service_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
 
 class SnapshotServiceStub(object):
     """The SnapshotService enables you to list, insert, store, and retrieve
@@ -39,37 +59,37 @@ class SnapshotServiceStub(object):
                 '/android.emulation.control.SnapshotService/ListSnapshots',
                 request_serializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotFilter.SerializeToString,
                 response_deserializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotList.FromString,
-                )
+                _registered_method=True)
         self.PullSnapshot = channel.unary_stream(
                 '/android.emulation.control.SnapshotService/PullSnapshot',
                 request_serializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
                 response_deserializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-                )
+                _registered_method=True)
         self.PushSnapshot = channel.stream_unary(
                 '/android.emulation.control.SnapshotService/PushSnapshot',
                 request_serializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
                 response_deserializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-                )
+                _registered_method=True)
         self.LoadSnapshot = channel.unary_unary(
                 '/android.emulation.control.SnapshotService/LoadSnapshot',
                 request_serializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
                 response_deserializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-                )
+                _registered_method=True)
         self.SaveSnapshot = channel.unary_unary(
                 '/android.emulation.control.SnapshotService/SaveSnapshot',
                 request_serializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
                 response_deserializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-                )
+                _registered_method=True)
         self.DeleteSnapshot = channel.unary_unary(
                 '/android.emulation.control.SnapshotService/DeleteSnapshot',
                 request_serializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
                 response_deserializer=android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-                )
+                _registered_method=True)
         self.TrackProcess = channel.unary_unary(
                 '/android.emulation.control.SnapshotService/TrackProcess',
                 request_serializer=android__env_dot_proto_dot_snapshot__service__pb2.IceboxTarget.SerializeToString,
                 response_deserializer=android__env_dot_proto_dot_snapshot__service__pb2.IceboxTarget.FromString,
-                )
+                _registered_method=True)
 
 
 class SnapshotServiceServicer(object):
@@ -249,6 +269,7 @@ def add_SnapshotServiceServicer_to_server(servicer, server):
     generic_handler = grpc.method_handlers_generic_handler(
             'android.emulation.control.SnapshotService', rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('android.emulation.control.SnapshotService', rpc_method_handlers)
 
 
  # This class is part of an EXPERIMENTAL API.
@@ -287,11 +308,21 @@ class SnapshotService(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/android.emulation.control.SnapshotService/ListSnapshots',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/android.emulation.control.SnapshotService/ListSnapshots',
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotFilter.SerializeToString,
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotList.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def PullSnapshot(request,
@@ -304,11 +335,21 @@ class SnapshotService(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_stream(request, target, '/android.emulation.control.SnapshotService/PullSnapshot',
+        return grpc.experimental.unary_stream(
+            request,
+            target,
+            '/android.emulation.control.SnapshotService/PullSnapshot',
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def PushSnapshot(request_iterator,
@@ -321,11 +362,21 @@ class SnapshotService(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.stream_unary(request_iterator, target, '/android.emulation.control.SnapshotService/PushSnapshot',
+        return grpc.experimental.stream_unary(
+            request_iterator,
+            target,
+            '/android.emulation.control.SnapshotService/PushSnapshot',
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def LoadSnapshot(request,
@@ -338,11 +389,21 @@ class SnapshotService(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/android.emulation.control.SnapshotService/LoadSnapshot',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/android.emulation.control.SnapshotService/LoadSnapshot',
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def SaveSnapshot(request,
@@ -355,11 +416,21 @@ class SnapshotService(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/android.emulation.control.SnapshotService/SaveSnapshot',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/android.emulation.control.SnapshotService/SaveSnapshot',
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def DeleteSnapshot(request,
@@ -372,11 +443,21 @@ class SnapshotService(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/android.emulation.control.SnapshotService/DeleteSnapshot',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/android.emulation.control.SnapshotService/DeleteSnapshot',
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.SerializeToString,
             android__env_dot_proto_dot_snapshot__service__pb2.SnapshotPackage.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def TrackProcess(request,
@@ -389,8 +470,18 @@ class SnapshotService(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/android.emulation.control.SnapshotService/TrackProcess',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/android.emulation.control.SnapshotService/TrackProcess',
             android__env_dot_proto_dot_snapshot__service__pb2.IceboxTarget.SerializeToString,
             android__env_dot_proto_dot_snapshot__service__pb2.IceboxTarget.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

android_env/proto/state.proto

@@ -1,4 +1,4 @@
-// Copyright 2023 DeepMind Technologies Limited.
+// Copyright 2024 DeepMind Technologies Limited.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.