mirror/trigger.py

#!/usr/bin/python
import os, re

# Given a file full of rsync output, scan through it for things that look like
# legitimate releases. Return the simplified triggers for each of those files.

matches = (
	# The major tarballs
	r'linux-(2\.6\.\d+)\.tar\.bz2',
	# Stable releases
	r'patch-(2\.6\.\d+\.\d+)\.bz2',
	# -rc releases
	r'patch-(2\.6\.\d+-rc\d+)\.bz2',
	# -git releases
	r'patch-(2\.6\.\d+(-rc\d+)?-git\d+).bz2',
	# -mm tree
	r'(2\.6\.\d+(-rc\d+)?-mm\d+)\.bz2',
	  )

compiled_matches = [re.compile(r) for r in matches]


class Trigger(object):
	def __init__(self):
		self.__actions = []

	def __re_scan(self, pattern, line):
		"""
		First check to see whether the pattern matches.
			(eg. Does it match "linux-2.6.\d.tar.bz2" ?)
			Then we strip out the actual trigger itself from that,
			and return it.
			(eg. return "2.6.\d")
			Note that the pattern uses match,
			so you need the whole filename
		"""
		match = pattern.match(line)
		if match:
			return match.group(1)
		else:
			return None


	def scan(self, input_file):
		triggers = []
		for line in open(input_file, 'r').readlines():
			for pattern in compiled_matches:
				filename = os.path.basename(line)
				t = self.__re_scan(pattern, filename)
				if t:
					triggers.append(t)

		# Call each of the actions and pass in the kernel list
		for action in self.__actions:
			action(triggers)


	def add_action(self, func):
		self.__actions.append(func)