def process(self):
# Ensure that home directories are taken into account
path = os.path.expanduser(self.path)
# Check if the filepath exists
if os.path.exists(path) or os.path.islink(path):
exists = os.path.exists(path)
# Determine the type of file found
if os.path.islink(path):
file_type = 'symlink'
source = os.readlink(path)
elif os.path.isdir(path):
file_type = 'directory'
source = None
elif self.aliases:
# Determine if the file is an alias
alias_url = NSURL.fileURLWithPath_(path)
bookmark_data, _error = NSURL.bookmarkDataWithContentsOfURL_error_(
alias_url, None)
if bookmark_data:
file_type = 'alias'
source_url, _is_stale, _error = (
# pylint: disable=line-too-long
NSURL.
URLByResolvingBookmarkData_options_relativeToURL_bookmarkDataIsStale_error_( # noqa: E501
bookmark_data, NSURLBookmarkResolutionWithoutUI,
None, None, None)
# pylint: enable=line-too-long
)
source = source_url.path() if source_url else None
else:
file_type = 'file'
source = None
else:
file_type = 'file'
source = None
# Determine if the path is a mountpoint
mount = os.path.ismount(path)
# Determine what flags are set on the file
stat = os.stat(path, follow_symlinks=False)
flags_bin = stat.st_flags
flags = [
flag for flag, flag_bin in FLAGS.items()
if flags_bin & flag_bin
]
else:
exists = False
file_type = None
source = None
mount = None
flags = None
return self.ok(exists=exists,
file_type=file_type,
source=source,
mount=mount,
flags=flags)
def process(self):
# Ensure that home directories are taken into account
path = os.path.expanduser(self.path)
source = os.path.expanduser(self.source) if self.source else None
if self.state == 'file':
if source:
# The source provided does not exist or is not a file
if not os.path.isfile(source):
raise ActionError(
'the source provided could not be found or is not a file'
)
# If the destination provided is a path, then we place the file in it
if os.path.isdir(path):
path = os.path.join(path, os.path.basename(source))
# An existing file at the destination path was found so we compare them
# and avoid making changes if they're identical
exists = os.path.isfile(path)
if exists and self.md5(source) == self.md5(path):
changed = self.set_file_attributes(path)
return self.changed(path=path) if changed else self.ok(
path=path)
# Copy the source to the destination
try:
shutil.copyfile(source, path)
except OSError:
raise ActionError(
'unable to copy source file to path requested')
self.set_file_attributes(path)
return self.changed(path=path)
else:
# An existing file at the destination path was found
if os.path.isfile(path):
changed = self.set_file_attributes(path)
return self.changed(path=path) if changed else self.ok(
path=path)
# Create an empty file at the destination path
try:
with open(path, 'w'):
pass
except IsADirectoryError:
raise ActionError('the destination path is a directory')
except OSError:
raise ActionError(
'unable to create an empty file at the path requested')
self.set_file_attributes(path)
return self.changed(path=path)
elif self.state == 'directory':
# An existing directory was found
if os.path.isdir(path):
changed = self.set_file_attributes(path)
return self.changed(path=path) if changed else self.ok(
path=path)
# Clean any existing item in the path requested
self.remove(path)
# Create the directory requested
try:
os.mkdir(path)
except OSError:
raise ActionError(
'the requested directory could not be created')
self.set_file_attributes(path)
return self.changed(path=path)
elif self.state == 'alias':
# If the destination provided is a path, then we place the file in it
if os.path.isdir(path):
path = os.path.join(path, os.path.basename(source))
# When creating an alias, the source must be an absolute path and exist
source = os.path.abspath(source)
if not os.path.exists(source):
raise ActionError('the source file provided does not exist')
# An existing alias at the destination path was found so we compare them
# and avoid making changes if they're identical
exists = os.path.isfile(path)
path_url = NSURL.fileURLWithPath_(path)
if exists:
bookmark_data, _error = NSURL.bookmarkDataWithContentsOfURL_error_(
path_url, None)
if bookmark_data:
source_url, _is_stale, _error = (
# pylint: disable=line-too-long
NSURL.
URLByResolvingBookmarkData_options_relativeToURL_bookmarkDataIsStale_error_( # noqa: E501
bookmark_data, NSURLBookmarkResolutionWithoutUI,
None, None, None)
# pylint: enable=line-too-long
)
if source_url and source_url.path() == source:
changed = self.set_file_attributes(path)
return self.changed(path=path) if changed else self.ok(
path=path)
# Delete any existing file or symlink at the path
self.remove(path)
# Create an NSURL object for the source (absolute paths must be used for aliases)
source_url = NSURL.fileURLWithPath_(source)
# Build the bookmark for the alias
bookmark_data, _error = (
# pylint: disable=line-too-long
source_url.
bookmarkDataWithOptions_includingResourceValuesForKeys_relativeToURL_error_( # noqa: E501
NSURLBookmarkCreationSuitableForBookmarkFile, None, None,
None)
# pylint: enable=line-too-long
)
# Write the alias using the bookmark data
success, _error = NSURL.writeBookmarkData_toURL_options_error_(
bookmark_data, path_url,
NSURLBookmarkCreationSuitableForBookmarkFile, None)
if not success:
raise ActionError(
'unable to create an alias at the path requested')
self.set_file_attributes(path)
return self.changed(path=path)
elif self.state == 'symlink':
# If the destination provided is a path, then we place the file in it
if os.path.isdir(path) and not os.path.islink(path):
path = os.path.join(path, os.path.basename(source))
# An existing symlink at the destination path was found so we compare them
# and avoid making changes if they're identical
exists = os.path.islink(path)
if exists and os.readlink(path) == source:
changed = self.set_file_attributes(path)
return self.changed(path=path) if changed else self.ok(
path=path)
# Delete any existing file or symlink at the path
self.remove(path)
# Create the symlink requested
try:
os.symlink(source, path)
except OSError:
raise ActionError('the requested symlink could not be created')
self.set_file_attributes(path)
return self.changed(path=path)
else: # 'absent'
removed = self.remove(path)
return self.changed(path=path) if removed else self.ok(path=path)
