precommit.py

from iafisher_precommit import Precommit, Problem, checks


def main():
    precommit = Precommit()
    
    # Generic checks
    precommit.register(checks.NoStagedAndUnstagedChanges())
    precommit.register(checks.NoWhiteSpaceInFilePath())

    # Python checks
    precommit.register(checks.PythonFormat())
    precommit.register(checks.PythonStyle())
    
    precommit.register(MyCustomCheck())

    precommit.run()


class MyCustomCheck:
    def check(self, repo_info, path):
        if "beef" in path:
            return Problem("file path is not vegetarian")


if __name__ == "__main__":
    main()