bashclub/check-zfs-replication
1
0
Fork 0
mirror of https://github.com/bashclub/check-zfs-replication.git synced 2024-11-07 17:31:58 +01:00
Code Issues Projects Releases Wiki Activity

Updated to version 3.05

Browse Source
This commit is contained in:
Thorsten Spille 2021-06-23 13:25:58 +02:00
parent 277f81bb1f
commit 904abf38ed
1 changed files with 43 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
checkzfs.py
View File

@ -16,14 +16,14 @@
## GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT ## GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
## LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. ## LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
VERSION = 3.01 VERSION = 3.05
### for check_mk usage link or copy binary to check_mk_agent/local/checkzfs ### for check_mk usage link or copy binary to check_mk_agent/local/checkzfs
### create /etc/check_mk/checkzfs ## the config file name matches the filename in check_mk_agent/local/ ### create /etc/check_mk/checkzfs ## the config file name matches the filename in check_mk_agent/local/
### to create a diffent set, link script to check_mk_agent/local/checkzfs2 and create /etc/check_mk/checkzfs2 ### to create a diffent set, link script to check_mk_agent/local/checkzfs2 and create /etc/check_mk/checkzfs2
### ###
### source: host1 # [optional] comma seperated hostnames to check for source ### source: host1 # [optional] comma seperated hostnames to check for source
### remote: host1 # [optional] " " " ### remote: host1 # [optional] " " " remote
### prefix: host1 # [optional] Prefix for check_mk Servicename - default REPLICA ### prefix: host1 # [optional] Prefix for check_mk Servicename - default REPLICA
### filter: rpool/data|replica # [optional] regex filter to match source ### filter: rpool/data|replica # [optional] regex filter to match source
### replicafilter: remote # [optional] regex filter to match for replica snapshots ### replicafilter: remote # [optional] regex filter to match for replica snapshots
@ -105,7 +105,6 @@ class zfs_dataset(object):
self.lastsnapshot = "" self.lastsnapshot = ""
def add_snapshot(self,**kwargs): def add_snapshot(self,**kwargs):
#print ("Add snapshot"+repr(kwargs))
_obj = zfs_snapshot(self,**kwargs) ## neuen snapshot mit parametern erstellen _obj = zfs_snapshot(self,**kwargs) ## neuen snapshot mit parametern erstellen
self.snapshots[_obj.guid] = _obj ## zu lokalen snapshots diesem DS hinzu self.snapshots[_obj.guid] = _obj ## zu lokalen snapshots diesem DS hinzu
return _obj ## snapshot objeckt zurück return _obj ## snapshot objeckt zurück
@ -122,14 +121,12 @@ class zfs_dataset(object):
def sorted_snapshots(self): def sorted_snapshots(self):
return sorted(self.snapshots.values(), key=lambda x: x.age) ## snapshots nach alter sortiert return sorted(self.snapshots.values(), key=lambda x: x.age) ## snapshots nach alter sortiert
@property @property
def dataset_name(self): ## namen mit host prefixen def dataset_name(self): ## namen mit host prefixen
if self.remote: if self.remote:
return f"{self.remote}#{self.dataset}" return f"{self.remote}#{self.dataset}"
return self.dataset return self.dataset
@property @property
def latest_snapshot(self): ## letzten snapshot def latest_snapshot(self): ## letzten snapshot
if self.snapshots: if self.snapshots:
@ -139,17 +136,22 @@ class zfs_dataset(object):
def get_info(self,source,threshold=None): def get_info(self,source,threshold=None):
_latest = self._get_latest_snapshot(source if source != self else None) ## wenn das source dataset nicht man selber ist _latest = self._get_latest_snapshot(source if source != self else None) ## wenn das source dataset nicht man selber ist
_status = None _status = None
_has_zfs_autosnapshot = any(map(lambda x: str(x.snapshot).startswith("zfs-auto-snap_"),self.snapshots.values()))
_message = "" _message = ""
if source == self: if source == self:
if not self.replica: if not self.replica:
_status = 1 ## warn _status = 1 ## warn
_message = _("kein Replikat gefunden") _message = _("kein Replikat gefunden")
if self.autosnapshot == 2 and _has_zfs_autosnapshot:
_status = 1 ## warn
_message = _("com.sun:auto-snapshot ist auf der Quelle auf true und wird evtl. mit repliziert")
else: else:
if _has_zfs_autosnapshot: ## nur auf systemen mit zfs-aut-snapshot
if self.autosnapshot == 1: if self.autosnapshot == 1:
#_status = 1 ## warn _status = 1 ## warn
_message = _("com.sun:auto-snapshot ist nicht gesetzt") _message = _("com.sun:auto-snapshot ist nicht false")
elif self.autosnapshot == 2: elif self.autosnapshot == 2:
#_status = 2 ## crit _status = 2 ## crit
_message = _("com.sun:auto-snapshot ist auf Replikationspartner aktiviert") _message = _("com.sun:auto-snapshot ist auf Replikationspartner aktiviert")
if _latest: if _latest:
@ -203,6 +205,12 @@ class no_regex_class(object):
def search(*args): def search(*args):
return True return True
class negative_regex_class(object):
def __init__(self,compiled_regex):
self.regex = compiled_regex
def search(self,text):
return not self.regex.search(text)
class zfscheck(object): class zfscheck(object):
ZFSLIST_REGEX = re.compile("^(?P<dataset>.*?)(?:|@(?P<snapshot>.*?))\t(?P<type>\w*)\t(?P<creation>\d+)\t(?P<guid>\d+)\t(?P<used>\d+|-)\t(?P<available>\d+|-)\t(?P<written>\d+|-)\t(?P<autosnapshot>[-\w]+)\t(?P<checkzfs>[-\w]+)$",re.M) ZFSLIST_REGEX = re.compile("^(?P<dataset>.*?)(?:|@(?P<snapshot>.*?))\t(?P<type>\w*)\t(?P<creation>\d+)\t(?P<guid>\d+)\t(?P<used>\d+|-)\t(?P<available>\d+|-)\t(?P<written>\d+|-)\t(?P<autosnapshot>[-\w]+)\t(?P<checkzfs>[-\w]+)$",re.M)
ZFS_DATASETS = {} ZFS_DATASETS = {}
@ -248,7 +256,7 @@ class zfscheck(object):
} }
COLUMN_MAPPER = {} COLUMN_MAPPER = {}
def __init__(self,remote,source,legacyhosts,output,prefix='REPLICA',**kwargs): def __init__(self,remote,source,legacyhosts,output,mail=None,prefix='REPLICA',**kwargs):
_start_time = time.time() _start_time = time.time()
self.remote_hosts = remote.split(",") if remote else [""] if source else [] ## wenn nicht und source woanders ... "" (also lokal) als remote self.remote_hosts = remote.split(",") if remote else [""] if source else [] ## wenn nicht und source woanders ... "" (also lokal) als remote
self.source_hosts = source.split(",") if source else [""] ## wenn nix dann "" als local self.source_hosts = source.split(",") if source else [""] ## wenn nix dann "" als local
@ -256,9 +264,10 @@ class zfscheck(object):
self.filter = None self.filter = None
self.prefix = prefix.strip().replace(" ","_") ## service name bei checkmk leerzeichen durch _ ersetzen self.prefix = prefix.strip().replace(" ","_") ## service name bei checkmk leerzeichen durch _ ersetzen
self.rawdata = False self.rawdata = False
self.mail_address = mail
self._overall_status = [] self._overall_status = []
self.sortreverse = False self.sortreverse = False
self.output = output self.output = output if mail == None else "mail"
self._check_kwargs(kwargs) self._check_kwargs(kwargs)
self.get_data() self.get_data()
if self.output != "snaplist": if self.output != "snaplist":
@ -325,6 +334,9 @@ class zfscheck(object):
if _k in ("filter","snapshotfilter","replicafilter"): if _k in ("filter","snapshotfilter","replicafilter"):
if _v: if _v:
if _v.startswith("!"):
_v = negative_regex_class(re.compile(_v[1:]))
else:
_v = re.compile(_v) _v = re.compile(_v)
else: else:
_v = no_regex_class() ### dummy klasse .search immer True - spart abfrage ob filter vorhanden _v = no_regex_class() ### dummy klasse .search immer True - spart abfrage ob filter vorhanden
@ -580,26 +592,30 @@ class zfscheck(object):
_line_print = True _line_print = True
return "\n".join(_out) return "\n".join(_out)
def html_output(self,data): def html_output(self,data,columns=None):
if not data: if not data:
return "" return ""
_header = data[0].keys() if not self.columns else self.columns _header = data[0].keys() if not self.columns else self.columns
_header_names = [self.COLUMN_NAMES.get(i,i) for i in _header] _header_names = [self.COLUMN_NAMES.get(i,i) for i in _header]
_converter = dict((i,self.COLUMN_MAPPER.get(i,(lambda x: str(x)))) for i in _header) _converter = dict((i,self.COLUMN_MAPPER.get(i,(lambda x: str(x)))) for i in _header)
_hostname = socket.getfqdn()
_out = [] _out = []
_out.append("<html><head>") _out.append("<html><head>")
_out.append("<style type='text/css'>.warn { background-color: yellow } .crit { background-color: red}</style>") _out.append("<meta name='color-scheme' content='only'>")
_out.append("<style type='text/css'>td {font-weight: bold;} .warn { background-color: yellow; color: black; } .crit { background-color: red; color: black;}</style>")
_out.append("<title>ZFS</title></head>") _out.append("<title>ZFS</title></head>")
_out.append("<body><table border=1>") _out.append(f"<body><h2>{_hostname}</h2>")
_out.append("<table border=1>")
_out.append("<tr><th>{0}</th></tr>".format("</th><th>".join(_header_names))) _out.append("<tr><th>{0}</th></tr>".format("</th><th>".join(_header_names)))
for _item in self._datasort(data): for _item in self._datasort(data):
_out.append("<tr class='{1}'><td>{0}</td></tr>".format("</td><td>".join([_converter.get(_col)(_item.get(_col,"")) for _col in _header]),_item.get("status","ok"))) _out.append("<tr class='{1}'><td>{0}</td></tr>".format("</td><td>".join([_converter.get(_col)(_item.get(_col,"")) for _col in _header]),_converter["status"](_item.get("status","0"))))
_out.append("</table></body></html>") _out.append("</table></body></html>")
return "".join(_out) return "".join(_out)
def mail_output(self,data): def mail_output(self,data):
_hostname = socket.getfqdn() _hostname = socket.getfqdn()
_email = self.mail_address
if not _email:
_users = open("/etc/pve/user.cfg","rt").read() _users = open("/etc/pve/user.cfg","rt").read()
_email = "root@{0}".format(_hostname) _email = "root@{0}".format(_hostname)
_emailmatch = re.search("^user:root@pam:.*?:(?P<mail>[\w.]+@[\w.]+):.*?$",_users,re.M) _emailmatch = re.search("^user:root@pam:.*?:(?P<mail>[\w.]+@[\w.]+):.*?$",_users,re.M)
@ -608,7 +624,7 @@ class zfscheck(object):
#raise Exception("No PVE User Email found") #raise Exception("No PVE User Email found")
_msg = EmailMessage() _msg = EmailMessage()
_msg.set_content(self.table_output(data,color=False)) _msg.set_content(self.table_output(data,color=False))
#_msg.add_alternative(self.html_output(data),subtype="html") ## FIXME wollte irgendwie nicht als multipart .. _msg.add_alternative(self.html_output(data),subtype="html") ## FIXME wollte irgendwie nicht als multipart ..
#_attach = MIMEApplication(self.csv_output(data),Name="zfs-check_{0}.csv".format(_hostname)) #_attach = MIMEApplication(self.csv_output(data),Name="zfs-check_{0}.csv".format(_hostname))
#_attach["Content-Disposition"] = "attachement; filename=zfs-check_{0}.csv".format(_hostname) #_attach["Content-Disposition"] = "attachement; filename=zfs-check_{0}.csv".format(_hostname)
#_msg.attach(_attach) #_msg.attach(_attach)
@ -654,6 +670,8 @@ if __name__ == "__main__":
help=_("Zeige nur folgende Spalten ({0})".format(",".join(zfscheck.VALIDCOLUMNS)))) help=_("Zeige nur folgende Spalten ({0})".format(",".join(zfscheck.VALIDCOLUMNS))))
_parser.add_argument("--sort",type=str,choices=zfscheck.VALIDCOLUMNS, _parser.add_argument("--sort",type=str,choices=zfscheck.VALIDCOLUMNS,
help=_("Sortiere nach Spalte")) help=_("Sortiere nach Spalte"))
_parser.add_argument("--mail",type=str,
help=_("Email für den Versand"))
_parser.add_argument("--threshold",type=str, _parser.add_argument("--threshold",type=str,
help=_("Grenzwerte für Alter von Snapshots warn,crit")) help=_("Grenzwerte für Alter von Snapshots warn,crit"))
_parser.add_argument("--rawdata",action="store_true", _parser.add_argument("--rawdata",action="store_true",
@ -693,7 +711,6 @@ if __name__ == "__main__":
except: except:
pass pass
args.output = "checkmk" args.output = "checkmk"
#sys.stderr.write(repr(args.__dict__))
try: try:
ZFSCHECK_OBJ = zfscheck(**args.__dict__) ZFSCHECK_OBJ = zfscheck(**args.__dict__)
pass ## for debugger pass ## for debugger