#!/usr/bin/python3

import pyudev
import psutil
import time
import os

context = pyudev.Context()
monitor = pyudev.Monitor.from_netlink(context)
monitor.filter_by("block")

autoMount = False


def printDeviceInfo(dev):
    print("")
    print("<BLOCK INFORMATION>")
    print("Device name: %s" % dev.get("DEVNAME"))
    print("Device type: %s" % dev.get("DEVTYPE"))
    print("Bus system: %s" % dev.get("ID_BUS"))
    print("Partition label: %s" % dev.get("ID_FS_LABEL"))
    print("FS: %s" % dev.get("ID_FS_SYSTEM_ID"))
    print("FS type: %s" % dev.get("ID_FS_TYPE"))
    print("Device usage: %s" % dev.get("ID_FS_USAGE"))
    print("Device model: %s" % dev.get("ID_MODEL"))
    print("Partition type: %s" % dev.get("ID_PART_TABLE_TYPE"))
    print("USB driver: %s" % dev.get("ID_USB_DRIVER"))
    print("Path id: %s" % dev.get("ID_PATH"))
    # print('Capacity: %s' % stdOut[0].strip())
    print("</BLOCK INFORMATION>")


# enumerate at device connection
for device in iter(monitor.poll, None):
    if "ID_FS_TYPE" in device:
        if device.action == "add":
            if device.device_node[5:7] == "sd" and device.get("DEVTYPE") == "partition":
                printDeviceInfo(device)
                print("New device {}".format(device.device_node))
                # loop until device is mounted
                if autoMount:
                    found = False
                    loop = 0
                    while (not found) and (loop < 10):
                        # need to sleep before devide is mounted
                        time.sleep(1)
                        for partition in psutil.disk_partitions():
                            if partition.device == device.device_node:
                                print("Mounted at {}".format(partition.mountpoint))
                                found = True
                        loop += 1
                else:
                    print("mount device to /media/box")
                    res = os.system("pmount " + device.device_node + " box")
                    print("Return type: ", res)

        if device.action == "remove":
            if device.device_node[5:7] == "sd" and device.get("DEVTYPE") == "partition":
                print("Device removed")
                if not autoMount:
                    print("unmount device /media/box")
                    res = os.system("pumount /media/box")
                    print("Return type: ", res)