Пример #1
0
    def setup(self, repo, hook):
        """Setup hook infrastructure.
           In addition to common things it also updates .hg/hgrc."""

        HookAPI.setup(self, repo, hook)
        rc_path = os.path.join(repo, ".hg", "hgrc")

        parser = ConfigParser()

        if os.path.exists(rc_path):
            parser.read([rc_path])

        section = "hooks"
        if not parser.has_section(section):
            parser.add_section(section)

        hook_path = self.hook_path(repo, hook)
        if parser.has_option(section, hook):
            if parser.get(section, hook) != hook_path:
                raise HookError("%s: %s hook is already set" % (rc_path, hook))
        else:
            parser.set(section, hook, hook_path)

        with open(rc_path, "w") as fhandle:
            parser.write(fhandle)
Пример #2
0
    def teardown(self, repo, hook):
        """Remove hooks infrastructure."""

        rc_path = os.path.join(repo, ".hg", "hgrc")

        if not os.path.exists(rc_path):
            return

        if HookAPI.teardown(self, repo, hook):
            parser = ConfigParser()
            parser.read([rc_path])
            section = "hooks"
            if parser.has_option(section, hook) and parser.get(section, hook) == self.hook_path(repo, hook):
                parser.remove_option(section, hook)
                return True