gutenprint/gutenprint-foomaticppdupdate

138 lines
3.7 KiB
Plaintext
Raw Normal View History

#!/usr/bin/python3
## gutenprint-foomaticppdupdate
## A utility for updating foomatic-generated PPDs so that they work with
## a newly-installed gutenprint package.
## Copyright (C) 2007, 2009, 2014 Red Hat, Inc.
## Author: Tim Waugh <twaugh@redhat.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., 675 Mass Ave, Cambridge, MA 02139, USA.
import sys
import glob
import os
import subprocess
import tempfile
import cups
if len (sys.argv) < 2 or sys.argv[1] == "--help":
print("Usage: gutenprint-foomaticppdupdate <version>")
sys.exit (1)
gutenprint_version = sys.argv[1]
dry_run = True
def generate_ppd (ppdfile, printer, driver):
p = subprocess.Popen (["foomatic-ppdfile", "-p", printer, "-d", driver],
stdin=subprocess.DEVNULL,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
(ppd, stderr) = p.communicate ()
fname = ppdfile + ".tmp"
try:
with open (fname, "w") as f:
f.write (ppd)
except IOError as e:
print (e)
raise
ppdobj = cups.PPD (fname)
os.remove (fname)
return ppdobj
def update_ppdfile (ppdfile):
try:
ppd = cups.PPD (ppdfile)
except RuntimeError:
# Invalid PPD in some way.
return
attr = ppd.findAttr ("FoomaticIDs")
if not attr:
return
IDs = attr.value.split (" ")
if len (IDs) != 2:
print ("Don't understand FoomaticIDs: %s" % IDs)
return
if not IDs[1].startswith ("gutenprint"):
return
attr = ppd.findAttr ("FoomaticRIPCommandLine")
if not attr:
return
cmdline = attr.value
STP_VERSION="STP_VERSION="
i = cmdline.find (STP_VERSION)
if i == -1:
return
i += len (STP_VERSION)
j = i + 1
end = len (cmdline)
while j < end:
ch = cmdline[j]
if ch != '.' and not ch.isdigit ():
break
j += 1
version = cmdline[i:j]
if gutenprint_version == version:
return
# Needs updating.
firstdot = gutenprint_version.find ('.')
seconddot = firstdot + 1 + gutenprint_version[1 + firstdot:].find ('.')
major = gutenprint_version[:seconddot]
driver = IDs[1]
dot = driver.find ('.')
driver = driver[:dot] + "." + major
try:
genppd = generate_ppd (ppdfile, IDs[0], driver)
except:
return
os.environ['FILE'] = ppdfile
os.system ('cp -af "$FILE" "$FILE".bak')
def update_options (options, newppd, origppd):
for newopt in options:
origopt = origppd.findOption (newopt.keyword)
if origopt:
newppd.markOption (newopt.keyword, origopt.defchoice)
genppd.markDefaults ()
for group in genppd.optionGroups:
update_options (group.options, genppd, ppd)
for subgroup in group.subgroups:
update_options (subgroup.options, genppd, ppd)
ps = genppd.findOption ("PageSize")
if ps:
update_options ([ps], genppd, ppd)
with open (ppdfile, "w") as f:
genppd.writeFd (f.fileno ())
print ("Updated PPD file %s" % ppdfile)
for ppdfile in glob.glob ("/etc/cups/ppd/*.ppd"):
update_ppdfile (ppdfile)