hplip/copy-deviceids.py

74 lines
2.0 KiB
Python
Executable File

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