986012
#!/usr/bin/python3
986012
import os
986012
import re
986012
import sys
986012
if len (sys.argv) < 3:
986012
    print ("Specify hpcups.drv and hpijs.drv pathnames")
986012
    sys.exit (1)
986012
986012
hpcups_drv = sys.argv[1]
986012
hpijs_drv = sys.argv[2]
986012
986012
# Match e.g.      Model "ModelName"
986012
# and catch 'ModelName' in group 0
986012
mn_re = re.compile ('^\s*ModelName\s+"(.*)"\s*$')
986012
986012
# Match e.g.      Attribute "1284DeviceID" "" "blah"
986012
# and catch everything before 'blah' in group 0, 'blah' in group 1,
986012
# trailing characters in group 2
986012
devid_re = re.compile ('^(\s*Attribute\s+"1284DeviceID"\s+""\s+")(.*)("\s*)$')
986012
986012
# Match e.g.   }
986012
end_re = re.compile ('^\s*}')
986012
986012
devid_by_mn = dict()
986012
986012
hpcups_lines = open (hpcups_drv, "r").readlines ()
986012
current_mn = None
986012
for line in hpcups_lines:
986012
    if current_mn == None:
986012
        match = mn_re.match (line)
986012
        if match == None:
986012
            continue
986012
986012
        current_mn = match.groups ()[0]
986012
    else:
986012
        match = devid_re.match (line)
986012
        if match:
986012
            devid_by_mn[current_mn] = match.groups ()[1]
986012
            continue
986012
986012
    if end_re.match (line):
986012
        current_mn = None
986012
986012
print("%d IEEE 1284 Device IDs loaded from %s" % (len (devid_by_mn),
986012
                                                os.path.basename (hpcups_drv)),
986012
      file=sys.stderr)
986012
986012
replaced = 0
986012
hpijs_lines = open (hpijs_drv, "r").readlines ()
986012
current_mn = None
986012
for line in hpijs_lines:
986012
    if current_mn == None:
986012
        match = mn_re.match (line)
986012
        if match:
986012
            current_mn = match.groups ()[0]
986012
            if current_mn.endswith (" hpijs"):
986012
                current_mn = current_mn[:-6]
986012
    else:
986012
        match = devid_re.match (line)
986012
        if match:
986012
            devid = devid_by_mn.get (current_mn)
986012
            if devid:
986012
                line = (match.groups ()[0] + devid + match.groups ()[2])
986012
                replaced += 1
986012
            else:
986012
                print ("Not matched: %s" % current_mn, file=sys.stderr)
986012
986012
    if end_re.match (line):
986012
        current_mn = None
986012
986012
    print (line.rstrip ("\n"))
986012
986012
print("%d IEEE 1284 Device IDs loaded in %s" % (replaced,
986012
                                                os.path.basename (hpijs_drv)),
986012
      file=sys.stderr)