45 lines
1.9 KiB
Python
45 lines
1.9 KiB
Python
|
|
||
|
|
import ops.cmd
|
||
|
|
import dsz
|
||
|
|
import sys
|
||
|
|
from ops.parseargs import ArgumentParser
|
||
|
|
|
||
|
|
def monitor_drives(interval):
|
||
|
|
known_drives = []
|
||
|
|
drives = ops.cmd.getDszCommand('drives')
|
||
|
|
while True:
|
||
|
|
results = drives.execute()
|
||
|
|
current_drives = []
|
||
|
|
for drive in results.driveitem:
|
||
|
|
current_drives.append(drive.drive)
|
||
|
|
if (drive.type == 'Removable'):
|
||
|
|
if (drive.drive not in known_drives):
|
||
|
|
known_drives.append(drive.drive)
|
||
|
|
ops.alert('Removable media decteced on drive {0} on host {1}'.format(drive.drive, ops.project.getTarget().hostname), dsz.GOOD, stamp=None)
|
||
|
|
for drive in known_drives:
|
||
|
|
if (drive not in current_drives):
|
||
|
|
ops.alert('Removable Media removed from drive {0} on host {1}'.format(drive, ops.project.getTarget().hostname), stamp=None)
|
||
|
|
known_drives = current_drives
|
||
|
|
dsz.Sleep(interval)
|
||
|
|
|
||
|
|
def get_parser():
|
||
|
|
parser = ArgumentParser(version='1.0.0', description='usb_monitor.py should be run in the background. It runs the Drives command at a given interval (default = 300) and monitors if removable media is plugged in or removed. It will run until killed.')
|
||
|
|
group_types = parser.add_argument_group('usb_monitor.py Arguments')
|
||
|
|
group_types.add_argument('--interval', dest='interval', type=int, action='store', nargs=1, help='The interval in SECONDS at which to run the Drives command. Default = 300')
|
||
|
|
return parser
|
||
|
|
|
||
|
|
def parse_usbargs(args):
|
||
|
|
parser = get_parser()
|
||
|
|
options = parser.parse_args(args)
|
||
|
|
return (options, parser.print_help)
|
||
|
|
|
||
|
|
def main(arguments):
|
||
|
|
interval = 0
|
||
|
|
if (len(arguments) == 0):
|
||
|
|
interval = 300000
|
||
|
|
else:
|
||
|
|
(options, help) = parse_usbargs(arguments)
|
||
|
|
interval = (options.interval[0] * 1000)
|
||
|
|
monitor_drives(interval)
|
||
|
|
if (__name__ == '__main__'):
|
||
|
|
main(sys.argv[1:])
|