Mercurial > hg > config
view python/find_duplicate_files.py @ 718:cd9d65e6e2ab
monkeypatch example
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Thu, 30 Oct 2014 15:22:29 -0700 |
parents | ab831c7621e9 |
children | dbd2562cb03e |
line wrap: on
line source
#!/usr/bin/env python # -*- coding: utf-8 -*- """ find duplicate files in a directory """ # imports import argparse import os import subprocess import sys # module globals __all__ = ['main', 'Parser'] class Parser(argparse.ArgumentParser): """CLI option parser""" def __init__(self, **kwargs): kwargs.setdefault('description', __doc__) argparse.ArgumentParser.__init__(self, **kwargs) self.add_argument('directory') self.options = None def parse_args(self, *args, **kw): options = argparse.ArgumentParser.parse_args(self, *args, **kw) self.validate(options) self.options = options return options def validate(self, options): """validate options""" if not os.path.isdir(options.directory): self.error("Not a directory: {}".format(options.directory)) def main(args=sys.argv[1:]): """CLI""" # parse command line options parser = Parser() options = parser.parse_args(args) output = subprocess.check_output(['ls', '-l', options.directory]).strip() rows = [row.strip().split() for row in output.splitlines()[1:]] sizes = {} for row in rows: size = int(row[4]) filename = row[-1] sizes.setdefault(size, []).append(filename) duplicates = {} for size, filenames in sizes.items(): if len(filenames) < 2: continue duplicates[size] = filenames for size in sorted(duplicates.keys()): print ('{} : '.format(size)) print ('\n'.join(duplicates[size])) print ('\n') if __name__ == '__main__': main()