#!/usr/bin/python

def is_system_disk(device, hostvars):
    """
    device: /dev/sda1 or /dev/sda or /dev/vda1 or /dev/vda1 
    hostvars: built-in vars for ansible
    """
    if device not in hostvars['ansible_device_links']['ids']:
        for disk_info in hostvars['ansible_mounts']:
            if device == disk_info['device'].split("/")[-1].strip("123456789"):
                return True
            else:
                return False
    else:
        return True


class FilterModule(object):
    def filters(self):
        return {
            'is_system_disk': is_system_disk
        }