Ticket #2868: runscript.diff

django/core/management.py

@@ -1189 +1189 @@
         code.interact()
 run_shell.args = '[--plain]'
 
+def runscript(script):
+    "Runs a python script in the current environment."
+    import os.path
+    script = os.path.abspath(script)
+    new_globals = globals().copy()
+    new_globals.update({"__name__": "__main__",
+                        "__file__": script})
+    try:
+        execfile(script, new_globals, {})
+    except IOError, e:
+        sys.stderr.write(style.ERROR("%s\n" % e))
+        sys.exit(1)
+runscript.args = '[path/to/python/script]'
+
 def dbshell():
     "Runs the command-line client for the current DATABASE_ENGINE."
     from django.db import runshell
@@ -1244 +1258 @@
     'runfcgi': runfcgi,
     'runserver': runserver,
     'shell': run_shell,
+    'runscript': runscript,
     'sql': get_sql_create,
     'sqlall': get_sql_all,
     'sqlclear': get_sql_delete,
@@ -1368 +1383 @@
         except IndexError:
             parser.print_usage_and_exit()
         action_mapping[action](name, os.getcwd())
+    elif action == 'runscript':
+        if len(args) != 2:
+            sys.stderr.write(style.ERROR("Error: You must specify a script to run.\n"))
+            sys.exit(1)
+        script = args[1]
+        action_mapping[action](script)
     elif action == 'runserver':
         if len(args) < 2:
             addr = ''