diff --git a/boggler/__main__.py b/boggler/__main__.py index 01dc7e9..bbc82fe 100755 --- a/boggler/__main__.py +++ b/boggler/__main__.py @@ -1,9 +1,9 @@ """Boggler Demo""" import sys from pathlib import Path -from boggler_utils import BoggleBoard, build_full_boggle_tree, read_boggle_file +from .boggler_utils import BoggleBoard, build_full_boggle_tree, read_boggle_file -if __name__ == '__main__': +def main(): if len(sys.argv) < 3 or len(sys.argv) > 4: print('Usage: python3 boggler.py [MAX_WORD_LENGTH]') sys.exit(1) @@ -32,3 +32,6 @@ if __name__ == '__main__': print("The [MAX_WORD_LENGTH] argument must be an integer.") print("Please try again.") sys.exit(1) + +if __name__ == '__main__': + main() diff --git a/pyproject.toml b/pyproject.toml index 6561b50..db31516 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,6 +28,9 @@ readme = "README.md" [tool.poetry.dependencies] python = "^3.11" +[tool.poetry.scripts] +boggler = "boggler.__main__:main" + [build-system] requires = ["poetry-core"]