Add script to run background tasks

[ipfire.org.git] / src / scripts / ipfire.org.in

#!/usr/bin/python3

import sys
import tornado.gen
import tornado.ioloop

import ipfire

class TaskRunner(object):
	def __init__(self, *args, **kwargs):
		self.backend = ipfire.Backend(*args, **kwargs)

		# Create an IOLoop
		self.ioloop = tornado.ioloop.IOLoop.current()

	def run_task(self, name, *args, **kwargs):
		"""
			This method runs the task with the given name and
			arguments asynchronically and exits the program in
			case on a non-zero exit code
		"""
		@tornado.gen.coroutine
		def task():
			yield self.backend.run_task(name, *args, **kwargs)

		return self.ioloop.run_sync(task)


def main():
	z = TaskRunner("@configsdir@/@PACKAGE_NAME@.conf")

	if len(sys.argv) < 2:
		sys.stderr.write("Argument needed\n")
		sys.exit(1)

	basename = sys.argv[0]
	task = sys.argv[1]
	args = sys.argv[2:]

	# Run the task
	z.run_task(task, *args)

main()