1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
# 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
|
