102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible 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 General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from __future__ import annotations
|
|
|
|
import fcntl
|
|
import os
|
|
|
|
|
|
def get_file_content(path, default=None, strip=True):
|
|
"""
|
|
Return the contents of a given file path
|
|
|
|
:args path: path to file to return contents from
|
|
:args default: value to return if we could not read file
|
|
:args strip: controls if we strip whitespace from the result or not
|
|
|
|
:returns: String with file contents (optionally stripped) or 'default' value
|
|
"""
|
|
data = default
|
|
if os.path.exists(path) and os.access(path, os.R_OK):
|
|
datafile = None
|
|
try:
|
|
datafile = open(path)
|
|
try:
|
|
# try to not enter kernel 'block' mode, which prevents timeouts
|
|
fd = datafile.fileno()
|
|
flag = fcntl.fcntl(fd, fcntl.F_GETFL)
|
|
fcntl.fcntl(fd, fcntl.F_SETFL, flag | os.O_NONBLOCK)
|
|
except Exception:
|
|
pass # not required to operate, but would have been nice!
|
|
|
|
# actually read the data
|
|
data = datafile.read()
|
|
|
|
if strip:
|
|
data = data.strip()
|
|
|
|
if len(data) == 0:
|
|
data = default
|
|
|
|
except Exception:
|
|
# ignore errors as some jails/containers might have readable permissions but not allow reads
|
|
pass
|
|
finally:
|
|
if datafile is not None:
|
|
datafile.close()
|
|
|
|
return data
|
|
|
|
|
|
def get_file_lines(path, strip=True, line_sep=None):
|
|
"""get list of lines from file"""
|
|
data = get_file_content(path, strip=strip)
|
|
if data:
|
|
if line_sep is None:
|
|
ret = data.splitlines()
|
|
else:
|
|
if len(line_sep) == 1:
|
|
ret = data.rstrip(line_sep).split(line_sep)
|
|
else:
|
|
ret = data.split(line_sep)
|
|
else:
|
|
ret = []
|
|
return ret
|
|
|
|
|
|
def get_mount_size(mountpoint):
|
|
mount_size = {}
|
|
|
|
try:
|
|
statvfs_result = os.statvfs(mountpoint)
|
|
mount_size['size_total'] = statvfs_result.f_frsize * statvfs_result.f_blocks
|
|
mount_size['size_available'] = statvfs_result.f_frsize * (statvfs_result.f_bavail)
|
|
|
|
# Block total/available/used
|
|
mount_size['block_size'] = statvfs_result.f_bsize
|
|
mount_size['block_total'] = statvfs_result.f_blocks
|
|
mount_size['block_available'] = statvfs_result.f_bavail
|
|
mount_size['block_used'] = mount_size['block_total'] - mount_size['block_available']
|
|
|
|
# Inode total/available/used
|
|
mount_size['inode_total'] = statvfs_result.f_files
|
|
mount_size['inode_available'] = statvfs_result.f_favail
|
|
mount_size['inode_used'] = mount_size['inode_total'] - mount_size['inode_available']
|
|
except OSError:
|
|
pass
|
|
|
|
return mount_size
|