1 #!/usr/bin/python -B 2 3 # Copyright 2017 The Android Open Source Project 4 # 5 # Licensed under the Apache License, Version 2.0 (the "License"); 6 # you may not use this file except in compliance with the License. 7 # You may obtain a copy of the License at 8 # 9 # http://www.apache.org/licenses/LICENSE-2.0 10 # 11 # Unless required by applicable law or agreed to in writing, software 12 # distributed under the License is distributed on an "AS IS" BASIS, 13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 # See the License for the specific language governing permissions and 15 # limitations under the License. 16 17 """Utility methods to work with Zip archives.""" 18 19 import itertools 20 21 from operator import attrgetter 22 from zipfile import ZipFile 23 24 25 def ZipCompare(path_a, path_b): 26 """Compares the contents of two Zip archives, returns True if equal.""" 27 28 with ZipFile(path_a, 'r') as zip_a: 29 info_a = zip_a.infolist() 30 31 with ZipFile(path_b, 'r') as zip_b: 32 info_b = zip_b.infolist() 33 34 if len(info_a) != len(info_b): 35 return False 36 37 info_a.sort(key=attrgetter('filename')) 38 info_b.sort(key=attrgetter('filename')) 39 40 return all( 41 a.filename == b.filename and 42 a.file_size == b.file_size and 43 a.CRC == b.CRC 44 for a, b in itertools.izip(info_a, info_b)) 45
