File: mpi2udev.py

package info (click to toggle)
media-player-info 22-2
  • links: PTS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 1,412 kB
  • ctags: 38
  • sloc: sh: 494; python: 138; makefile: 25
file content (129 lines) | stat: -rwxr-xr-x 3,930 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#!/usr/bin/env python3
# Generate udev rules from music player identification (.mpi) files
#
# (C) 2009 Canonical Ltd.
# Author: Martin Pitt <martin.pitt@ubuntu.com>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.

import sys, configparser, os.path, os

# translation of mpi Device keys to udev attributes
mpi2udev = {
    'vendorid': 'ATTRS{idVendor}=="%s"',
    'productid': 'ATTRS{idProduct}=="%s"',
    'usbvendor': 'ATTRS{vendor}=="%s"',
    'usbmodel': 'ATTRS{model}=="%s"',
    'usbproduct': 'ATTRS{product}=="%s"',
    'usbmanufacturer': 'ATTRS{manufacturer}=="%s"',
}

def parse_mpi(mpi, hwdb):
    '''Print udev rule for given ConfigParser object.'''

    cp = configparser.RawConfigParser()
    assert cp.read(mpi, encoding='UTF-8')

    rule = ''

    for name in ['usbvendor', 'usbproduct', 'usbmodel', 'usbmanufacturer']:
        try:
            value = cp.get('Device', name)
            rule += mpi2udev[name] % value + ', '
        except configparser.NoOptionError:
            continue


    # if using hwdb and no info other than idVendor+idProduct was found, we don't need to write
    # an udev rule
    if hwdb and not rule:
        return

    try:
        usbids = {}
        for usbid in cp.get('Device', 'devicematch').split(';'):
            if len(usbid.split(':')) != 3:
                continue
            (subsystem, vid, pid) = usbid.split(':')
            if subsystem != "usb":
                continue
            if vid in usbids:
                usbids[vid].append(pid)
            else:
                usbids[vid] = [ pid ]

        for vid, pids in usbids.items():
            rule += 'ATTRS{idVendor}=="%s", ATTRS{idProduct}=="%s"'% (vid, '|'.join(pids)) + ', '
    except configparser.NoOptionError:
        pass

    # if no information was found, don't write a rule at all
    if not rule:
        return

    try:
        m = cp.get('Device', 'product')
        rule = ('# %s\n' % m) + rule
    except configparser.NoOptionError:
        pass

    rule += 'ENV{ID_MEDIA_PLAYER}="%s"' % os.path.splitext(os.path.basename(mpi))[0]

    # do we have an icon?
    try:
        icon = cp.get('Device', 'icon')
        # breaks media player detection : https://bugs.launchpad.net/ubuntu/+source/gvfs/+bug/657609
        # rule += ', ENV{UDISKS_PRESENTATION_ICON_NAME}="%s"' % icon
    except configparser.NoOptionError:
        pass

    # empty line between blocks
    rule += '\n\n'

    # write bytes so that we are independent of the locale
    os.write(sys.stdout.fileno(), rule.encode('UTF-8'))

#
# main
#

# udev rules header
os.write(sys.stdout.fileno(), b'''ACTION!="add|change", GOTO="media_player_end"
# catch MTP devices
ENV{DEVTYPE}=="usb_device", GOTO="media_player_start"

# catch UMS devices
SUBSYSTEM!="block", GOTO="media_player_end"
SUBSYSTEMS=="usb", GOTO="media_player_start"
GOTO="media_player_end"

LABEL="media_player_start"

''')

# the first argument should be "hwdb" or "udev"
hwdb = False
if sys.argv[1] == 'hwdb':
    hwdb = True
elif sys.argv[1] != 'udev':
    sys.stderr.write('Unknown mode %s. Must be "hwdb" or "udev".\n' % sys.argv[1])
    sys.exit(1)

# parse MPI files
for f in sys.argv[2:]:
    parse_mpi(f, hwdb)

# udev rules footer
os.write(sys.stdout.fileno(), b'\nLABEL="media_player_end"\n')