| OLD | NEW |
| (Empty) |
| 1 # Copyright (c) 2014 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 collections | |
| 6 import os | |
| 7 | |
| 8 | |
| 9 class PathSet(collections.MutableSet): | |
| 10 """A set of paths. | |
| 11 | |
| 12 All mutation methods can take both directories or individual files, but the | |
| 13 iterator yields the individual files. All paths are automatically normalized. | |
| 14 """ | |
| 15 def __init__(self, iterable=None): | |
| 16 self._paths = set() | |
| 17 if iterable: | |
| 18 self |= iterable | |
| 19 | |
| 20 def __contains__(self, path): | |
| 21 return os.path.realpath(path) in self._paths | |
| 22 | |
| 23 def __iter__(self): | |
| 24 return iter(self._paths) | |
| 25 | |
| 26 def __len__(self): | |
| 27 return len(self._paths) | |
| 28 | |
| 29 def add(self, path): | |
| 30 path = os.path.realpath(path) | |
| 31 if os.path.isfile(path): | |
| 32 self._paths.add(path) | |
| 33 for root, _, files in os.walk(path): | |
| 34 for basename in files: | |
| 35 file_path = os.path.join(root, basename) | |
| 36 if os.path.isfile(file_path): | |
| 37 self._paths.add(file_path) | |
| 38 | |
| 39 def discard(self, path): | |
| 40 path = os.path.realpath(path) | |
| 41 self._paths.discard(path) | |
| 42 for root, _, files in os.walk(path): | |
| 43 for basename in files: | |
| 44 self._paths.discard(os.path.join(root, basename)) | |
| OLD | NEW |