taskdo

#!/usr/bin/env python3
# coding: utf-8

# Copyright 2015 by Leipzig University Library, http://ub.uni-leipzig.de
#                by The Finc Authors, http://finc.info
#                by Martin Czygan, <martin.czygan@uni-leipzig.de>
#
# This file is part of some open source application.
#
# Some open source application 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 3 of the License, or (at your option) any later version.
#
# Some open source application 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 Foobar.  If not, see <http://www.gnu.org/licenses/>.
#
# @license GPL-3.0+ <http://spdx.org/licenses/GPL-3.0+>
#

"""
The main command line program to interact with the tasks.
"""

from __future__ import print_function

import importlib
import os
import sys

import luigi
from luigi.parameter import MissingParameterException
from luigi.task_register import TaskClassNotFoundException

from siskin.utils import get_task_import_cache

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print('usage: taskdo TASKNAME [--param PARAM, ...]', file=sys.stderr)
        sys.exit(1)

    taskname = sys.argv[1]

    task_import_cache, path = get_task_import_cache()
    if taskname in task_import_cache:
        importlib.import_module(task_import_cache[taskname])
    else:
        try:
            os.remove(path)
        except OSError:
            pass

        from siskin.sources import *
        from siskin.workflows import *

    try:
        luigi.run()
    except MissingParameterException as exc:
        print('missing parameter: %s' % exc, file=sys.stderr)
        sys.exit(1)
    except TaskClassNotFoundException as exc:
        print(exc, file=sys.stderr)
        sys.exit(1)
    except Exception as exc:
        # TODO(martin): do we need to set the exit code?
        print(exc, file=sys.stderr)
        sys.exit(1)