| # -*- coding: utf-8 -*- | |
| # | |
| # pyhwp : hwp file format parser in python | |
| # Copyright (C) 2010-2023 mete0r <https://github.com/mete0r> | |
| # | |
| # This program is free software: you can redistribute it and/or modify | |
| # it under the terms of the GNU Affero General Public License as published by | |
| # the Free Software Foundation, either version 3 of the License, or | |
| # (at your option) any later version. | |
| # | |
| # This program is distributed in the hope that it will be useful, | |
| # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| # GNU Affero General Public License for more details. | |
| # | |
| # You should have received a copy of the GNU Affero General Public License | |
| # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| # | |
| from __future__ import absolute_import | |
| from __future__ import print_function | |
| from __future__ import unicode_literals | |
| import io | |
| import os.path | |
| class FileSystemStorage(object): | |
| ''' Directory-based stroage. ''' | |
| def __init__(self, path): | |
| self.path = path | |
| def __iter__(self): | |
| return iter(sorted(os.listdir(self.path))) | |
| def __getitem__(self, name): | |
| path = os.path.join(self.path, name) | |
| if os.path.isdir(path): | |
| return FileSystemStorage(path) | |
| elif os.path.exists(path): | |
| return FileSystemStream(path) | |
| else: | |
| raise KeyError(name) | |
| class FileSystemStream(object): | |
| ''' File-based stream. ''' | |
| def __init__(self, path): | |
| self.path = path | |
| def open(self): | |
| return io.open(self.path, 'rb') | |