diff options
Diffstat (limited to 'catapult/common/py_vulcanize/py_vulcanize/fake_fs.py')
-rw-r--r-- | catapult/common/py_vulcanize/py_vulcanize/fake_fs.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/catapult/common/py_vulcanize/py_vulcanize/fake_fs.py b/catapult/common/py_vulcanize/py_vulcanize/fake_fs.py index dfcb5e60..40b01bb5 100644 --- a/catapult/common/py_vulcanize/py_vulcanize/fake_fs.py +++ b/catapult/common/py_vulcanize/py_vulcanize/fake_fs.py @@ -2,14 +2,19 @@ # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. +from __future__ import absolute_import +from __future__ import division +from __future__ import print_function + import codecs +import collections import os import sys -import collections -import StringIO + +import six -class WithableStringIO(StringIO.StringIO): +class WithableStringIO(six.StringIO): def __enter__(self, *args): return self @@ -23,7 +28,7 @@ class FakeFS(object): def __init__(self, initial_filenames_and_contents=None): self._file_contents = {} if initial_filenames_and_contents: - for k, v in initial_filenames_and_contents.iteritems(): + for k, v in six.iteritems(initial_filenames_and_contents): self._file_contents[k] = v self._bound = False @@ -106,7 +111,7 @@ class FakeFS(object): def _FakeWalk(self, top): assert os.path.isabs(top) - all_filenames = self._file_contents.keys() + all_filenames = list(self._file_contents.keys()) pending_prefixes = collections.deque() pending_prefixes.append(top) visited_prefixes = set() |