| OLD | NEW |
| (Empty) |
| 1 # Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import hashlib | |
| 6 import os | |
| 7 | |
| 8 | |
| 9 def CallAndRecordIfStale( | |
| 10 function, record_path=None, input_paths=None, input_strings=None, | |
| 11 force=False): | |
| 12 """Calls function if the md5sum of the input paths/strings has changed. | |
| 13 | |
| 14 The md5sum of the inputs is compared with the one stored in record_path. If | |
| 15 this has changed (or the record doesn't exist), function will be called and | |
| 16 the new md5sum will be recorded. | |
| 17 | |
| 18 If force is True, the function will be called regardless of whether the | |
| 19 md5sum is out of date. | |
| 20 """ | |
| 21 if not input_paths: | |
| 22 input_paths = [] | |
| 23 if not input_strings: | |
| 24 input_strings = [] | |
| 25 md5_checker = _Md5Checker( | |
| 26 record_path=record_path, | |
| 27 input_paths=input_paths, | |
| 28 input_strings=input_strings) | |
| 29 if force or md5_checker.IsStale(): | |
| 30 function() | |
| 31 md5_checker.Write() | |
| 32 | |
| 33 | |
| 34 def _UpdateMd5ForFile(md5, path, block_size=2**16): | |
| 35 with open(path, 'rb') as infile: | |
| 36 while True: | |
| 37 data = infile.read(block_size) | |
| 38 if not data: | |
| 39 break | |
| 40 md5.update(data) | |
| 41 | |
| 42 | |
| 43 def _UpdateMd5ForDirectory(md5, dir_path): | |
| 44 for root, _, files in os.walk(dir_path): | |
| 45 for f in files: | |
| 46 _UpdateMd5ForFile(md5, os.path.join(root, f)) | |
| 47 | |
| 48 | |
| 49 def _UpdateMd5ForPath(md5, path): | |
| 50 if os.path.isdir(path): | |
| 51 _UpdateMd5ForDirectory(md5, path) | |
| 52 else: | |
| 53 _UpdateMd5ForFile(md5, path) | |
| 54 | |
| 55 | |
| 56 class _Md5Checker(object): | |
| 57 def __init__(self, record_path=None, input_paths=None, input_strings=None): | |
| 58 if not input_paths: | |
| 59 input_paths = [] | |
| 60 if not input_strings: | |
| 61 input_strings = [] | |
| 62 | |
| 63 assert record_path.endswith('.stamp'), ( | |
| 64 'record paths must end in \'.stamp\' so that they are easy to find ' | |
| 65 'and delete') | |
| 66 | |
| 67 self.record_path = record_path | |
| 68 | |
| 69 md5 = hashlib.md5() | |
| 70 for i in sorted(input_paths): | |
| 71 _UpdateMd5ForPath(md5, i) | |
| 72 for s in input_strings: | |
| 73 md5.update(s) | |
| 74 self.new_digest = md5.hexdigest() | |
| 75 | |
| 76 self.old_digest = '' | |
| 77 if os.path.exists(self.record_path): | |
| 78 with open(self.record_path, 'r') as old_record: | |
| 79 self.old_digest = old_record.read() | |
| 80 | |
| 81 def IsStale(self): | |
| 82 return self.old_digest != self.new_digest | |
| 83 | |
| 84 def Write(self): | |
| 85 with open(self.record_path, 'w') as new_record: | |
| 86 new_record.write(self.new_digest) | |
| OLD | NEW |