Exemplo n.º 1
0
class PostReceiveDispatcher(object):
    def __init__(self, config, bot):
        self.config = config
        self.bot = bot
        self.shortener = UrlShortener()

    def _dispatch_commit(self, repository, branch, commit):
        d = self.shortener.make_short_url(commit['url'])

        def onUrlShortened(short_url):
            self.bot.send_message(repository.channel,
                                  repository.format % {
                'repository': repository.name,
                'branch': branch,

                'commit_id': commit['id'][:7],
                'url': short_url,
                'author': _get_commit_author(commit),
                'summary': commit['message'].splitlines()[0]
            })
        d.addCallback(onUrlShortened)

    def _dispatch_bundle(self, info, repository, branch, commits):
        authors = set()
        for commit in commits:
            authors.add(_get_commit_author(commit))
        before = info['before']
        after = info['after']
        commit_range = before[:7] + '..' + after[:7]
        url = "http://github.com/%s/compare/%s...%s" % (repository.name,
                                                        before,
                                                        after)

        d = self.shortener.make_short_url(url)
        def onUrlShortened(short_url):
            self.bot.send_message(repository.channel,
                                  repository.bundled_format % {
                'repository': repository.name,
                'branch': branch,
                'authors': ', '.join(authors),
                'commit_count': len(commits),
                'commit_range': commit_range,
                'url': short_url,
            })
        d.addCallback(onUrlShortened)

    def dispatch(self, payload):
        parsed = json.loads(payload)
        repository_name = (parsed['repository']['owner']['name'] + '/' +
                           parsed['repository']['name'])
        repository = self.config.repositories_by_name[repository_name]
        branch = parsed['ref'].split('/')[-1]
        commits = parsed['commits']

        if not repository.branches or branch in repository.branches:
            if len(commits) <= repository.max_commit_count:
                for commit in commits:
                    self._dispatch_commit(repository, branch, commit)
            else:
                self._dispatch_bundle(parsed, repository, branch, commits)
Exemplo n.º 2
0
 def __init__(self, config, bot):
     self.config = config
     self.bot = bot
     self.shortener = UrlShortener()