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 |