Merge branch 'feature/travis-container' of https://github.com/AlexanderS/bcfg2

1 files changed, 14 insertions, 6 deletions

diff --git a/src/lib/Bcfg2/manage.py b/src/lib/Bcfg2/manage.py
index b156deb0f..9675a3db1 100755
--- a/src/lib/Bcfg2/manage.py
+++ b/src/lib/Bcfg2/manage.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python
-import os
+""" Wrapper for the django manage.py with the Bcfg2 Opitons parsing. """
+
 import sys
 import django
 import Bcfg2.Options
@@ -10,14 +11,21 @@ try:
 except ImportError:
     pass
 
-parser = Bcfg2.Options.get_parser()
-parser.add_options([Bcfg2.Options.PositionalArgument('django_command', nargs='*')])
-parser.parse()
 
-if __name__ == "__main__":
+def main():
+    parser = Bcfg2.Options.get_parser()
+    parser.add_options([
+        Bcfg2.Options.PositionalArgument('django_command', nargs='*')])
+    parser.parse()
+
     if django.VERSION[0] == 1 and django.VERSION[1] >= 6:
         from django.core.management import execute_from_command_line
-        execute_from_command_line(sys.argv[:1] + Bcfg2.Options.setup.django_command)
+        execute_from_command_line(
+            sys.argv[:1] + Bcfg2.Options.setup.django_command)
     else:
         from django.core.management import execute_manager
         execute_manager(Bcfg2.DBSettings.settings)
+
+
+if __name__ == "__main__":
+    main()