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 |