blob: 85d5fb5e3c093b4fcab6104e1f08c235d07f217a [file] [log] [blame] [edit]
# main.py
#
# Copyright (C) 2011 Carlos Garcia Campos <carlosgc@gnome.org>
#
# 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
from __future__ import absolute_import, division, print_function
import sys
import argparse
import commands
import backends
import os
from Config import Config
from multiprocessing import cpu_count
class ListAction(argparse.Action):
def __call__(self, parser, namespace, values, option_string = None):
setattr(namespace, self.dest, values.split(','))
class HelpAction(argparse.Action):
def __call__(self, parser, namespace, values, option_string = None):
if option_string == '--help-command':
commands.run([values, '--help'])
sys.exit(0)
parser.print_help()
commands.print_help()
sys.exit(0)
def main(args):
n_cpus = cpu_count()
parser = argparse.ArgumentParser(
description = 'Poppler regression tests',
prog = 'poppler-regtest',
usage = '%(prog)s [options ...] command [command-options ...] tests',
add_help = False)
parser.add_argument('-h', '--help',
action = HelpAction, nargs = 0)
parser.add_argument('--help-command', metavar = 'COMMAND',
action = HelpAction,
help = 'Show help for a given command')
parser.add_argument('-v', '--verbose',
action = 'store_true', dest = 'verbose', default = False,
help = 'Run in verbose mode')
parser.add_argument('--utils-dir',
action = 'store', dest = 'utils_dir', default = os.path.abspath("../build/utils"),
help = 'Directory of poppler utils used for the tests')
parser.add_argument('-b', '--backends',
action = ListAction, dest = 'backends',
help = 'List of backends that will be used (separated by comma)')
parser.add_argument('--skip', metavar = 'FILE',
action = 'store', dest = 'skipped_file',
help = 'File containing tests to skip')
parser.add_argument('-p', '--passwords', metavar = 'FILE',
action = 'store', dest = 'passwords_file',
help = 'File containing the documents passwords')
parser.add_argument('-t', '--threads',
action = 'store', dest = 'threads', type = int, default = n_cpus,
help = 'Number of worker threads (Default: %d)' % n_cpus)
ns, args = parser.parse_known_args(args)
if not args:
parser.print_help()
sys.exit(0)
c = Config(vars(ns))
if c.threads <= 0:
c.threads = n_cpus - c.threads
try:
return commands.run(args)
except commands.UnknownCommandError:
sys.stderr.write("Unknown command: %s\n" % (args[0]))
commands.print_help()
return 1
except backends.UnknownBackendError as e:
sys.stderr.write(str(e) + "\n")
sys.stdout.write("Backends are: %s\n" % (", ".join([backend.get_name() for backend in backends.get_all_backends()])))
return 1
if __name__ == '__main__':
import sys
sys.exit(main(sys.argv[1:]))