49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
import sys
|
|
import hashlib
|
|
from mozpack.packager.unpack import UnpackFinder
|
|
from collections import OrderedDict
|
|
|
|
'''
|
|
Find files duplicated in a given packaged directory, independently of its
|
|
package format.
|
|
'''
|
|
|
|
|
|
def find_dupes(source):
|
|
md5s = OrderedDict()
|
|
for p, f in UnpackFinder(source):
|
|
content = f.open().read()
|
|
m = hashlib.md5(content).digest()
|
|
if not m in md5s:
|
|
md5s[m] = (len(content), [])
|
|
md5s[m][1].append(p)
|
|
total = 0
|
|
num_dupes = 0
|
|
for m, (size, paths) in md5s.iteritems():
|
|
if len(paths) > 1:
|
|
print 'Duplicates %d bytes%s:' % (size,
|
|
' (%d times)' % (len(paths) - 1) if len(paths) > 2 else '')
|
|
print ''.join(' %s\n' % p for p in paths)
|
|
total += (len(paths) - 1) * size
|
|
num_dupes += 1
|
|
if num_dupes:
|
|
print "WARNING: Found %d duplicated files taking %d bytes" % \
|
|
(num_dupes, total) + " (uncompressed)"
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
import os
|
|
print >>sys.stderr, "Usage: %s directory" % \
|
|
os.path.basename(sys.argv[0])
|
|
sys.exit(1)
|
|
|
|
find_dupes(sys.argv[1])
|
|
|
|
if __name__ == "__main__":
|
|
main()
|