Initial Commit

Author: tlusk

.gitignore

@@ -0,0 +1,2 @@
+.idea/
+*.iml

LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2016 Timothy Lusk
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

MANIFEST.in

@@ -0,0 +1 @@
+recursive-include src/raid_status_notifier *

example.cfg

@@ -0,0 +1,7 @@
+[settings]
+pushover_api_token=abcdefghijklmnopqrstuvwxyz
+pushover_user_key=abcdefghijklmnopqrstuvwxyz
+
+[btrfs_mount_points]
+mount1=/mnt/data
+mount2=/mnt/data2

requirements.txt

@@ -0,0 +1,5 @@
+python-pushover>=0.2
+
+--index-url https://pypi.python.org/simple/
+
+-e .

setup.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python3
+
+from setuptools import setup
+
+reqs = ['python-pushover>=0.2']
+
+setup(
+    name='raid-status-notifier',
+    version="1.0",
+    url='https://github.com/tlusk/raid-status-notifier',
+    author='Timothy Lusk',
+    author_email='[email protected]',
+    description='BTRFS/ZFS Raid Status Notifier',
+    license='MIT',
+    packages=['raid_status_notifier'],
+    package_dir={'': 'src'},
+    include_package_data=True,
+    entry_points={'console_scripts': [
+        'raid-status-notifier = raid_status_notifier.main:main'
+    ]},
+    install_requires=reqs,
+    zip_safe=False,
+    test_suite='unittest2.collector',
+    classifiers=[
+        'Environment :: Console',
+        'Intended Audience :: Developers',
+        'License :: OSI Approved :: MIT License',
+        'Operating System :: OS Independent',
+        'Programming Language :: Python',
+        'Topic :: Software Development :: Libraries :: Python Modules'
+    ],
+)

src/raid_status_notifier/__init__.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+
+import sys
+import re
+import os
+import pickle
+
+from argparse import ArgumentParser
+from subprocess import check_output
+from pushover import init, Client
+from configparser import ConfigParser
+
+
+class RaidStatusChecker(object):
+    def __init__(self, config):
+        init(config.get("settings", "pushover_api_token"))
+        self.client = Client(config.get("settings", "pushover_user_key"))
+        self.btrfs_mount_points = [path for key, path in config.items("btrfs_mount_points")]
+
+    def check_btrfs_stats(self):
+        for mount_point in self.btrfs_mount_points:
+            stats_filename = "data/btrfs-stats_%s.p" % mount_point[1:].replace("/", "-")
+
+            device_stats = {}
+            if os.path.exists(stats_filename):
+                device_stats = pickle.load(open(stats_filename, "rb"))
+
+            status = check_output(["sudo", "btrfs", "device", "stats", mount_point]).decode("utf-8").strip()
+            new_errors = False
+            regex = re.compile('\[/dev/(.*)\]\.(\S*)\s*(\d*)')
+
+            for line in status.split('\n'):
+                match = regex.match(line)
+                if match is not None:
+                    if match.group(1) not in device_stats:
+                        device_stats[match.group(1)] = {}
+                    previous_stats = device_stats[match.group(1)].get(match.group(2), 0)
+                    if int(match.group(3)) > previous_stats:
+                        device_stats[match.group(1)][match.group(2)] = int(match.group(3))
+                        new_errors = True
+
+            if not os.path.exists("data"):
+                os.mkdir("data")
+
+            pickle.dump(device_stats, open(stats_filename, "wb"))
+
+            if new_errors is not False:
+                self.client.send_message(status, title="BTRFS Errors: %s" % mount_point)
+
+    def check_btrfs_drives(self):
+        status = check_output(["sudo", "btrfs", "fi", "show", "-d"]).decode("utf-8").strip()
+
+        regex = re.compile('(missing|warning)')
+        if regex.match(status) is not None:
+            self.client.send_message(status, title="BTRFS Array Error")
+
+    def check_zfs_drives(self):
+        status = check_output(["sudo", "zpool", "status", "-x"])
+        if status != "all pools are healthy":
+            self.client.send_message(status, title="ZFS Array Error")
+
+    def run(self):
+        self.check_zfs_drives()
+        self.check_btrfs_stats()
+        self.check_btrfs_drives()
+
+
+def main(argv=None):
+    if argv is None:
+        argv = sys.argv[1:]
+
+    parser = ArgumentParser()
+    parser.add_argument("-c", "--config", action="store", help="Configuration File", metavar="CONFIGFILE")
+    parser.set_defaults(config="settings.cfg")
+
+    options = parser.parse_args(argv)
+
+    config = ConfigParser()
+    config.read(options.config)
+
+    RaidStatusChecker(config).run()
+
+if __name__ == "__main__":
+    sys.exit(main())