[7989] | 1 | #!/usr/bin/python |
---|
| 2 | # |
---|
| 3 | # inf2mondb.py: convert MicroSoft .inf files for monitors to MonitorDB |
---|
| 4 | # |
---|
| 5 | # originally by Matt Wilson <msw@redhat.com> |
---|
| 6 | # option parsing and database comparison by Fred New |
---|
| 7 | # ini parsing completely rewritten by Matt Domsch <Matt_Domsch@dell.com> 2006 |
---|
| 8 | # |
---|
| 9 | # Copyright 2002 Red Hat, Inc. |
---|
| 10 | # Copyright 2006 Dell, Inc. |
---|
| 11 | # |
---|
| 12 | # This software may be freely redistributed under the terms of the GNU |
---|
| 13 | # library public license. |
---|
| 14 | # |
---|
| 15 | # You should have received a copy of the GNU Library Public License |
---|
| 16 | # along with this program; if not, write to the Free Software |
---|
| 17 | # Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
---|
| 18 | """ |
---|
| 19 | """ |
---|
| 20 | |
---|
| 21 | |
---|
| 22 | import sys |
---|
| 23 | import string |
---|
| 24 | import re |
---|
| 25 | import ConfigParser |
---|
| 26 | |
---|
| 27 | # this is a class to deal with various file line endings and leading whitespace |
---|
| 28 | # converts all \r line endings to \n. |
---|
| 29 | # It also strips leading whitespace. |
---|
| 30 | # NOTE: be sure to always return _something_, even if it is just "\n", or we |
---|
| 31 | # break the file API. (nothing == eof) |
---|
| 32 | class myFile(object): |
---|
| 33 | def __init__(self, *args): |
---|
| 34 | self.fd = open(*args) |
---|
| 35 | |
---|
| 36 | def close(self): |
---|
| 37 | return self.fd.close() |
---|
| 38 | |
---|
| 39 | def readline(self, *args): |
---|
| 40 | line = self.fd.readline(*args) |
---|
| 41 | line = line.replace('\r', '\n') |
---|
| 42 | line = line.replace('\n\n', '\n') |
---|
| 43 | line = line.lstrip(" \t") |
---|
| 44 | return line |
---|
| 45 | |
---|
| 46 | |
---|
| 47 | # we will use this to override default option parsing in ConfigParser to handle |
---|
| 48 | # Microsoft-style "INI" files. (Which do not necessarily have " = value " after |
---|
| 49 | # the option name |
---|
| 50 | OPTCRE = re.compile( |
---|
| 51 | r'(?P<option>[^:=\s][^:=]*)' # very permissive! |
---|
| 52 | r'\s*(?P<vi>[:=]{0,1})\s*' # any number of space/tab, |
---|
| 53 | # optionally followed by |
---|
| 54 | # separator (either : or =) |
---|
| 55 | # optionally followed |
---|
| 56 | # by any # space/tab |
---|
| 57 | r'(?P<value>.*)$' # everything up to eol |
---|
| 58 | ) |
---|
| 59 | |
---|
| 60 | def usage(): |
---|
| 61 | print "No monitor.inf file specified." |
---|
| 62 | sys.exit(1) |
---|
| 63 | |
---|
| 64 | percentSplit = re.compile(r'%(?P<field>.*)%') |
---|
| 65 | def percent_to_string(ini, strings, name): |
---|
| 66 | mo = percentSplit.match(name) |
---|
| 67 | if (mo): |
---|
| 68 | field = mo.group('field') |
---|
| 69 | try: |
---|
| 70 | val = strings[field.lower()] |
---|
| 71 | except KeyError: |
---|
| 72 | return "" |
---|
| 73 | return val.strip() |
---|
| 74 | return "" |
---|
| 75 | |
---|
| 76 | |
---|
| 77 | def main(): |
---|
| 78 | defaultDB = "/usr/share/hwdata/MonitorsDB" |
---|
| 79 | |
---|
| 80 | from optparse import OptionParser |
---|
| 81 | parser = OptionParser(usage= sys.argv[0] + " [options] filename.inf") |
---|
| 82 | parser.add_option("-n", "--new", |
---|
| 83 | action="store_true", dest="newonly", default=False, |
---|
| 84 | help="Compare results with the monitors database and only " |
---|
| 85 | "show DB lines that don't already appear in the " |
---|
| 86 | "database.") |
---|
| 87 | parser.add_option("-v", "--verbose", |
---|
| 88 | action="store_true", dest="verbose", default=False, |
---|
| 89 | help="Used with --new to permit us to see which monitors " |
---|
| 90 | "already appear in the monitors database.") |
---|
| 91 | parser.add_option("-d", "--database", |
---|
| 92 | dest="database", |
---|
| 93 | default=defaultDB, |
---|
| 94 | help="Used with --new to specify a different " |
---|
| 95 | "monitors database. " |
---|
| 96 | "The default is " + defaultDB + ".", |
---|
| 97 | metavar="FILE") |
---|
| 98 | parser.add_option("-i", "--inflist", |
---|
| 99 | dest="inflist", |
---|
| 100 | default=None, |
---|
| 101 | help="Read list of inf files from inflist", |
---|
| 102 | metavar="FILE") |
---|
| 103 | |
---|
| 104 | (options, args) = parser.parse_args() |
---|
| 105 | |
---|
| 106 | ini = ConfigParser.ConfigParser() |
---|
| 107 | ini.optionxform = __builtins__.str |
---|
| 108 | ini.OPTCRE = OPTCRE |
---|
| 109 | if options.inflist is not None: |
---|
| 110 | f = open(options.inflist) |
---|
| 111 | lines = f.readlines() |
---|
| 112 | f.close |
---|
| 113 | for a in lines: |
---|
| 114 | f = myFile(a.strip()) |
---|
| 115 | ini.readfp(f) |
---|
| 116 | f.close |
---|
| 117 | |
---|
| 118 | for a in args: |
---|
| 119 | f = myFile(a) |
---|
| 120 | ini.readfp(f) |
---|
| 121 | f.close() |
---|
| 122 | |
---|
| 123 | # First, build a dictionary of monitors we already know about. |
---|
| 124 | # When we get ready to print the monitors we found in the .inf |
---|
| 125 | # file, we can decide whether it is new or not. |
---|
| 126 | |
---|
| 127 | # Dictionary of booleans showing whether certain monitor EDIDs are known. |
---|
| 128 | # knownids[knownID] = True |
---|
| 129 | # knownids[unknownID] isn't defined. |
---|
| 130 | knownids = {} |
---|
| 131 | |
---|
| 132 | if options.newonly: |
---|
| 133 | try: |
---|
| 134 | mdb = open(options.database, 'r') |
---|
| 135 | except IOError, (errno, str): |
---|
| 136 | print "Unable to open %s: %s" % (options.database, str) |
---|
| 137 | sys.exit(1) |
---|
| 138 | |
---|
| 139 | knowns = mdb.readlines() |
---|
| 140 | mdb.close() |
---|
| 141 | |
---|
| 142 | for known in knowns: |
---|
| 143 | if len(string.strip(known)) == 0 or known[0] == '#': |
---|
| 144 | continue |
---|
| 145 | knownids[string.lower(string.strip(string.split(known, ';')[2]))] = True |
---|
| 146 | |
---|
| 147 | |
---|
| 148 | # a dictionary of manufacturers we're looking at |
---|
| 149 | manufacturers = {} |
---|
| 150 | # a big fat dictionary of strings to use later on. |
---|
| 151 | strings = {} |
---|
| 152 | |
---|
| 153 | # This RE is for EISA info lines |
---|
| 154 | # %D5259A%=D5259A, Monitor\HWP0487 |
---|
| 155 | monitor1Re = re.compile(r'.*,.*Monitor\\(?P<id>[^\s]*)') |
---|
| 156 | # This one is for legacy entries |
---|
| 157 | # %3020% =PB3020, MonID_PB3020 |
---|
| 158 | monitor2Re = re.compile(r'.*,.*MonID_(?P<id>[^\s]*)') |
---|
| 159 | |
---|
| 160 | for section in ini.sections(): |
---|
| 161 | if section.lower() == "manufacturer": |
---|
| 162 | for mfr in ini.options(section): |
---|
| 163 | # generate the vendor.arch funny entries |
---|
| 164 | manufacturer_values = string.split(ini.get(section, mfr), ',') |
---|
| 165 | manufacturers[manufacturer_values[0]] = mfr |
---|
| 166 | while len(manufacturer_values) > 1: |
---|
| 167 | manufacturers["%s.%s" % (manufacturer_values[0], manufacturer_values[-1])] = mfr |
---|
| 168 | manufacturer_values = manufacturer_values[0:-1] |
---|
| 169 | |
---|
| 170 | elif section.lower() == "strings": |
---|
| 171 | for key in ini.options(section): |
---|
| 172 | strings[key.lower()] = string.strip(ini.get(section, key)).replace('"','') |
---|
| 173 | # exceptions |
---|
| 174 | strings["dell"] = "Dell" |
---|
| 175 | |
---|
| 176 | |
---|
| 177 | for mfr in manufacturers.keys(): |
---|
| 178 | if ini.has_section(mfr): |
---|
| 179 | monitor_vendor_name = manufacturers[mfr] |
---|
| 180 | for monitor_name in ini.options(mfr): |
---|
| 181 | v = ini.get(mfr, monitor_name) |
---|
| 182 | v = v.split(',') |
---|
| 183 | install_key = v[0].strip() |
---|
| 184 | |
---|
| 185 | line = ini.get(mfr, monitor_name) |
---|
| 186 | # Find monitor inf IDs and EISA ids |
---|
| 187 | |
---|
| 188 | edid = "0" |
---|
| 189 | mo = monitor1Re.match(line) |
---|
| 190 | if mo: |
---|
| 191 | edid = mo.group('id') |
---|
| 192 | else: |
---|
| 193 | mo = monitor2Re.match(line) |
---|
| 194 | if mo: |
---|
| 195 | edid = mo.group('id').strip() |
---|
| 196 | |
---|
| 197 | if knownids.has_key(edid.lower()): |
---|
| 198 | continue |
---|
| 199 | |
---|
| 200 | if ini.has_section(install_key): |
---|
| 201 | line = ini.get(install_key, "AddReg") |
---|
| 202 | if line: |
---|
| 203 | sline = line.split(',') |
---|
| 204 | registry = sline[0] |
---|
| 205 | try: |
---|
| 206 | resolution = sline[1] |
---|
| 207 | except IndexError: |
---|
| 208 | resolution = "" |
---|
| 209 | try: |
---|
| 210 | dpms = sline[2] |
---|
| 211 | except IndexError: |
---|
| 212 | dpms = "" |
---|
| 213 | |
---|
| 214 | if ini.has_section(registry): |
---|
| 215 | for line in ini.options(registry): |
---|
| 216 | if string.find(line, 'HKR,"MODES') >= 0: |
---|
| 217 | sline = line.split('"') |
---|
| 218 | try: |
---|
| 219 | syncline = sline[3] |
---|
| 220 | except IndexError: |
---|
| 221 | syncline = "," |
---|
| 222 | |
---|
| 223 | syncline = syncline.split(',') |
---|
| 224 | hsync = syncline[0].strip() |
---|
| 225 | vsync = syncline[1].strip() |
---|
| 226 | |
---|
| 227 | output = "%s; %s; %s; %s; %s" % (percent_to_string(ini, strings, monitor_vendor_name), |
---|
| 228 | percent_to_string(ini, strings, monitor_name), |
---|
| 229 | edid, hsync, vsync) |
---|
| 230 | if dpms.lower().strip() == "dpms": |
---|
| 231 | output = output + "; 1" |
---|
| 232 | |
---|
| 233 | if not knownids.has_key(edid.lower()): |
---|
| 234 | print output |
---|
| 235 | knownids[edid.lower()] = True |
---|
| 236 | |
---|
| 237 | |
---|
| 238 | |
---|
| 239 | if __name__ == "__main__": |
---|
| 240 | sys.exit(main()) |
---|