1# Copyright (C) 2021 The Android Open Source Project
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7#      http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14
15import argparse
16
17from android.snapshot import snapshot_pb2
18
19def main():
20    parser = argparse.ArgumentParser()
21    parser.add_argument('type', type = str, help = 'Type (snapshot or update)')
22    parser.add_argument('file', type = str, help = 'Input file')
23    args = parser.parse_args()
24
25    with open(args.file, 'rb') as fp:
26        data = fp.read()
27
28    if args.type == 'snapshot':
29        msg = snapshot_pb2.SnapshotStatus()
30    elif args.type == 'update':
31        msg = snapshot_pb2.SnapshotUpdateStatus()
32    else:
33        raise Exception('Unknown proto type')
34
35    msg.ParseFromString(data)
36    print(msg)
37
38if __name__ == '__main__':
39    main()
40