OLD | NEW |
1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import os | 5 import os |
6 | 6 |
7 import file_system | 7 import file_system |
8 from future import Future | 8 from future import Future |
9 | 9 |
10 class LocalFileSystem(file_system.FileSystem): | 10 class LocalFileSystem(file_system.FileSystem): |
11 """FileSystem implementation which fetches resources from the local | 11 """FileSystem implementation which fetches resources from the local |
12 filesystem. | 12 filesystem. |
13 """ | 13 """ |
14 def __init__(self, base_path): | 14 def __init__(self, base_path): |
15 self._base_path = self._ConvertToFilepath(base_path) | 15 self._base_path = self._ConvertToFilepath(base_path) |
16 | 16 |
17 def _ConvertToFilepath(self, path): | 17 def _ConvertToFilepath(self, path): |
18 return path.replace('/', os.sep) | 18 return path.replace('/', os.sep) |
19 | 19 |
20 def _ReadFile(self, filename): | 20 def _ReadFile(self, filename, binary): |
21 with open(os.path.join(self._base_path, filename), 'r') as f: | 21 with open(os.path.join(self._base_path, filename), 'r') as f: |
22 return file_system._ProcessFileData(f.read(), filename) | 22 contents = f.read() |
| 23 if binary: |
| 24 return contents |
| 25 return file_system._ProcessFileData(contents, filename) |
23 | 26 |
24 def _ListDir(self, dir_name): | 27 def _ListDir(self, dir_name): |
25 all_files = [] | 28 all_files = [] |
26 full_path = os.path.join(self._base_path, dir_name) | 29 full_path = os.path.join(self._base_path, dir_name) |
27 for path in os.listdir(full_path): | 30 for path in os.listdir(full_path): |
28 if path.startswith('.'): | 31 if path.startswith('.'): |
29 continue | 32 continue |
30 if os.path.isdir(os.path.join(full_path, path)): | 33 if os.path.isdir(os.path.join(full_path, path)): |
31 all_files.append(path + '/') | 34 all_files.append(path + '/') |
32 else: | 35 else: |
33 all_files.append(path) | 36 all_files.append(path) |
34 return all_files | 37 return all_files |
35 | 38 |
36 def Read(self, paths): | 39 def Read(self, paths, binary=False): |
37 result = {} | 40 result = {} |
38 for path in paths: | 41 for path in paths: |
39 if path.endswith('/'): | 42 if path.endswith('/'): |
40 result[path] = self._ListDir(self._ConvertToFilepath(path)) | 43 result[path] = self._ListDir(self._ConvertToFilepath(path)) |
41 else: | 44 else: |
42 result[path] = self._ReadFile(self._ConvertToFilepath(path)) | 45 result[path] = self._ReadFile(self._ConvertToFilepath(path), binary) |
43 return Future(value=result) | 46 return Future(value=result) |
44 | 47 |
45 def Stat(self, path): | 48 def Stat(self, path): |
46 return self.StatInfo(os.stat(os.path.join(self._base_path, path)).st_mtime) | 49 return self.StatInfo(os.stat(os.path.join(self._base_path, path)).st_mtime) |
OLD | NEW |