#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
"""FileSystems interface class for accessing the correct filesystem"""
# pytype: skip-file
import logging
import re
from typing import BinaryIO # pylint: disable=unused-import
from apache_beam.io.filesystem import BeamIOError
from apache_beam.io.filesystem import CompressionTypes
from apache_beam.io.filesystem import FileSystem
from apache_beam.metrics.metric import Lineage
from apache_beam.options.value_provider import RuntimeValueProvider
_LOGGER = logging.getLogger(__name__)
# All filesystem implements should be added here as
# best effort imports. We don't want to force loading
# a module if the user doesn't supply the correct
# packages that these filesystems rely on.
#
# pylint: disable=wrong-import-position, unused-import
try:
from apache_beam.io.hadoopfilesystem import HadoopFileSystem
except ModuleNotFoundError:
# optional file system packages are not installed.
pass
except ImportError as e:
_LOGGER.warning(
'Failed to import HadoopFileSystem; '
'loading of this filesystem will be skipped. '
'Error details: %s',
e)
try:
from apache_beam.io.localfilesystem import LocalFileSystem
except ModuleNotFoundError:
pass
except ImportError as e:
# optional file system packages are installed but failed to load.
_LOGGER.warning(
'Failed to import LocalFileSystem; '
'loading of this filesystem will be skipped. '
'Error details: %s',
e)
try:
from apache_beam.io.gcp.gcsfilesystem import GCSFileSystem
except ModuleNotFoundError:
pass
except ImportError as e:
_LOGGER.warning(
'Failed to import GCSFileSystem; '
'loading of this filesystem will be skipped. '
'Error details: %s',
e)
try:
from apache_beam.io.aws.s3filesystem import S3FileSystem
except ModuleNotFoundError:
pass
except ImportError as e:
_LOGGER.warning(
'Failed to import S3FileSystem; '
'loading of this filesystem will be skipped. '
'Error details: %s',
e)
try:
from apache_beam.io.azure.blobstoragefilesystem import BlobStorageFileSystem
except ModuleNotFoundError:
pass
except ImportError as e:
_LOGGER.warning(
'Failed to import BlobStorageFileSystem; '
'loading of this filesystem will be skipped. '
'Error details: %s',
e)
# pylint: enable=wrong-import-position, unused-import
__all__ = ['FileSystems']
[docs]
class FileSystems(object):
"""A class that defines the functions that can be performed on a filesystem.
All methods are static and access the underlying registered filesystems.
"""
URI_SCHEMA_PATTERN = re.compile('(?P<scheme>[a-zA-Z][-a-zA-Z0-9+.]*)://.*')
_pipeline_options = None
[docs]
@classmethod
def set_options(cls, pipeline_options):
"""Set filesystem options.
Args:
pipeline_options: Instance of ``PipelineOptions``.
"""
cls._pipeline_options = pipeline_options
[docs]
@staticmethod
def get_scheme(path):
match_result = FileSystems.URI_SCHEMA_PATTERN.match(path.strip())
if match_result is None:
return None
return match_result.groupdict()['scheme']
[docs]
@staticmethod
def get_filesystem(path):
# type: (str) -> FileSystems
"""Get the correct filesystem for the specified path
"""
try:
path_scheme = FileSystems.get_scheme(path)
systems = [
fs for fs in FileSystem.get_all_subclasses()
if fs.scheme() == path_scheme
]
if len(systems) == 0:
raise ValueError(
'Unable to get filesystem from specified path, please use the '
'correct path or ensure the required dependency is installed, '
'e.g., pip install apache-beam[gcp]. Path specified: %s' % path)
elif len(systems) == 1:
# Pipeline options could come either from the Pipeline itself (using
# direct runner), or via RuntimeValueProvider (other runners).
options = (
FileSystems._pipeline_options or
RuntimeValueProvider.runtime_options)
return systems[0](pipeline_options=options)
else:
raise ValueError('Found more than one filesystem for path %s' % path)
except ValueError:
raise
except Exception as e:
raise BeamIOError('Unable to get the Filesystem', {path: e})
[docs]
@staticmethod
def join(basepath, *paths):
# type: (str, *str) -> str
"""Join two or more pathname components for the filesystem
Args:
basepath: string path of the first component of the path
paths: path components to be added
Returns: full path after combining all the passed components
"""
filesystem = FileSystems.get_filesystem(basepath)
return filesystem.join(basepath, *paths)
[docs]
@staticmethod
def split(path):
"""Splits the given path into two parts.
Splits the path into a pair (head, tail) such that tail contains the last
component of the path and head contains everything up to that.
For file-systems other than the local file-system, head should include the
prefix.
Args:
path: path as a string
Returns:
a pair of path components as strings.
"""
filesystem = FileSystems.get_filesystem(path)
return filesystem.split(path)
[docs]
@staticmethod
def mkdirs(path):
"""Recursively create directories for the provided path.
Args:
path: string path of the directory structure that should be created
Raises:
IOError: if leaf directory already exists.
"""
filesystem = FileSystems.get_filesystem(path)
return filesystem.mkdirs(path)
[docs]
@staticmethod
def match(patterns, limits=None):
"""Find all matching paths to the patterns provided.
Pattern matching is done using each filesystem's ``match`` method (e.g.
:meth:`.filesystem.FileSystem.match`).
.. note::
- Depending on the :class:`.FileSystem` implementation, file listings
(the ``.FileSystem._list`` method) may not be recursive.
- If the file listing is not recursive, a pattern like
``scheme://path/*/foo`` will not be able to mach any files.
See Also:
:meth:`.filesystem.FileSystem.match`
Pattern syntax:
The pattern syntax is based on the fnmatch_ syntax, with the following
differences:
- ``*`` Is equivalent to ``[^/\\]*`` rather than ``.*``.
- ``**`` Is equivalent to ``.*``.
.. _`fnmatch`: https://docs.python.org/2/library/fnmatch.html
Args:
patterns: list of string for the file path pattern to match against
limits: list of maximum number of responses that need to be fetched
Returns: list of ``MatchResult`` objects.
Raises:
``BeamIOError``: if any of the pattern match operations fail
"""
if len(patterns) == 0:
return []
filesystem = FileSystems.get_filesystem(patterns[0])
return filesystem.match(patterns, limits)
[docs]
@staticmethod
def create(
path,
mime_type='application/octet-stream',
compression_type=CompressionTypes.AUTO):
# type: (...) -> BinaryIO
"""Returns a write channel for the given file path.
Args:
path: string path of the file object to be written to the system
mime_type: MIME type to specify the type of content in the file object
compression_type: Type of compression to be used for this object. See
``CompressionTypes`` for possible values.
Returns: file handle with a ``close`` function for the user to use.
"""
filesystem = FileSystems.get_filesystem(path)
return filesystem.create(path, mime_type, compression_type)
[docs]
@staticmethod
def open(
path,
mime_type='application/octet-stream',
compression_type=CompressionTypes.AUTO):
# type: (...) -> BinaryIO
"""Returns a read channel for the given file path.
Args:
path: string path of the file object to be written to the system
mime_type: MIME type to specify the type of content in the file object
compression_type: Type of compression to be used for this object. See
``CompressionTypes`` for possible values.
Returns: file handle with a ``close`` function for the user to use.
"""
filesystem = FileSystems.get_filesystem(path)
return filesystem.open(path, mime_type, compression_type)
[docs]
@staticmethod
def copy(source_file_names, destination_file_names):
"""Recursively copy the file list from the source to the destination
Args:
source_file_names: list of source file objects that needs to be copied
destination_file_names: list of destination of the new object
Raises:
``BeamIOError``: if any of the copy operations fail
"""
if len(source_file_names) == 0:
return
filesystem = FileSystems.get_filesystem(source_file_names[0])
return filesystem.copy(source_file_names, destination_file_names)
[docs]
@staticmethod
def rename(source_file_names, destination_file_names):
"""Rename the files at the source list to the destination list.
Source and destination lists should be of the same size.
Args:
source_file_names: List of file paths that need to be moved
destination_file_names: List of destination_file_names for the files
Raises:
``BeamIOError``: if any of the rename operations fail
"""
if len(source_file_names) == 0:
return
filesystem = FileSystems.get_filesystem(source_file_names[0])
return filesystem.rename(source_file_names, destination_file_names)
[docs]
@staticmethod
def exists(path):
"""Check if the provided path exists on the FileSystem.
Args:
path: string path that needs to be checked.
Returns: boolean flag indicating if path exists
"""
filesystem = FileSystems.get_filesystem(path)
return filesystem.exists(path)
[docs]
@staticmethod
def last_updated(path):
"""Get UNIX Epoch time in seconds on the FileSystem.
Args:
path: string path of file.
Returns: float UNIX Epoch time
Raises:
``BeamIOError``: if path doesn't exist.
"""
filesystem = FileSystems.get_filesystem(path)
return filesystem.last_updated(path)
[docs]
@staticmethod
def checksum(path):
"""Fetch checksum metadata of a file on the
:class:`~apache_beam.io.filesystem.FileSystem`.
This operation returns checksum metadata as stored in the underlying
FileSystem. It should not read any file data. Checksum type and format are
FileSystem dependent and are not compatible between FileSystems.
Args:
path: string path of a file.
Returns: string containing checksum
Raises:
``BeamIOError``: if path isn't a file or doesn't exist.
"""
filesystem = FileSystems.get_filesystem(path)
return filesystem.checksum(path)
[docs]
@staticmethod
def delete(paths):
"""Deletes files or directories at the provided paths.
Directories will be deleted recursively.
Args:
paths: list of paths that give the file objects to be deleted
Raises:
``BeamIOError``: if any of the delete operations fail
"""
if isinstance(paths, str):
raise BeamIOError(
'Delete passed string argument instead of list: %s' % paths)
if len(paths) == 0:
return
filesystem = FileSystems.get_filesystem(paths[0])
return filesystem.delete(paths)
[docs]
@staticmethod
def get_chunk_size(path):
"""Get the correct chunk size for the FileSystem.
Args:
path: string path that needs to be checked.
Returns: integer size for parallelization in the FS operations.
"""
filesystem = FileSystems.get_filesystem(path)
return filesystem.CHUNK_SIZE
[docs]
@staticmethod
def report_source_lineage(path, level=None):
"""
Report source :class:`~apache_beam.metrics.metric.Lineage`.
Args:
path: string path to be reported.
level: the level of file path. default to
:class:`~apache_beam.io.filesystem.FileSystem.Lineage`.FILE.
"""
filesystem = FileSystems.get_filesystem(path)
filesystem.report_lineage(path, Lineage.sources(), level=level)
[docs]
@staticmethod
def report_sink_lineage(path, level=None):
"""
Report sink :class:`~apache_beam.metrics.metric.Lineage`.
Args:
path: string path to be reported.
level: the level of file path. default to
:class:`~apache_beam.io.filesystem.FileSystem.Lineage`.FILE.
"""
filesystem = FileSystems.get_filesystem(path)
filesystem.report_lineage(path, Lineage.sinks(), level=level)