diff --git a/colab/management/__init__.py b/colab/management/__init__.py new file mode 100644 index 0000000..10b913c --- /dev/null +++ b/colab/management/__init__.py @@ -0,0 +1,14 @@ + +import os +from django.core.management import ManagementUtility + + +def execute_from_command_line(argv=None): + """ + A simple method that runs a ManagementUtility. + """ + os.environ.setdefault("DJANGO_SETTINGS_MODULE", "colab.settings") + os.environ.setdefault("COLAB_SETTINGS", "/etc/colab/settings.yaml") + + utility = ManagementUtility(argv) + utility.execute() diff --git a/scripts/colab-admin b/scripts/colab-admin deleted file mode 100755 index 9d84e43..0000000 --- a/scripts/colab-admin +++ /dev/null @@ -1,11 +0,0 @@ -#!/usr/bin/env python -import os -import sys - -if __name__ == "__main__": - os.environ.setdefault("DJANGO_SETTINGS_MODULE", "colab.settings") - os.environ.setdefault("COLAB_SETTINGS", "/etc/colab/settings.yaml") - - from django.core.management import execute_from_command_line - - execute_from_command_line(sys.argv) diff --git a/setup.py b/setup.py index 70a0db5..1a0cfb7 100644 --- a/setup.py +++ b/setup.py @@ -19,6 +19,9 @@ setup( license='LICENSE.txt', packages=find_packages(exclude=EXCLUDE_FROM_PACKAGES), include_package_data=True, + entry_points={'console_scripts': [ + 'colab-admin = colab.management:execute_from_command_line', + ]}, zip_safe=False, long_description=open('README.rst').read(), install_requires=reqs, -- libgit2 0.21.2