Regularly cleanup files

[people/jschlag/pbs.git] / src / scripts / pakfire-build-service

#!/usr/bin/python

import sys

# Use tornado's logging options
import tornado.options
tornado.options.parse_command_line()

import pakfire.buildservice

class Cli(object):
	def __init__(self, *args, **kwargs):
		# Initialise backend
		self.backend = pakfire.buildservice.Backend(*args, **kwargs)

		self._commands = {
			# Run mirror check
			"check-mirrors"    : self.backend.mirrors.check,

			# Cleanup files
			"cleanup-files" : self.backend.cleanup_files,

			# Cleanup sessions
			"cleanup-sessions" : self.backend.sessions.cleanup,

			# Cleanup uploads
			"cleanup-uploads" : self.backend.uploads.cleanup,

			# Sends all queued messages
			"process-message-queue" : self.backend.messages.process_queue,

			# Send bug updates to Bugzilla
			"send-bug-updates" : self.backend.bugzilla.send_all,
		}

	def __call__(self, *args):
		if not len(args) >= 2:
			print >>sys.stderr, "Insufficient number of arguments"
			return 2

		args = list(args)
		basename = args.pop(0)
		command  = args.pop(0)

		# Get called command
		try:
			command = self._commands[command]
		except KeyError:
			print >>sys.stderr, "Command not found: %s" % command
			return 2

		# Execute command
		r = command(*args)

		# Exit with error code
		sys.exit(r or 0)

# main

cli = Cli()
cli(*sys.argv)