从命令行通过shell_plus运行python脚本

时间:2011-07-14 23:45:36

标签: python django

我有一个python脚本,我运行以填充我的数据库。我通常在shell_plus中运行脚本,因为需要依赖项。有没有办法将脚本加载到shell_plus并从我的linux命令行运行所有内容而不实际打开shell_plus接口?

2 个答案:

答案 0 :(得分:2)

答案 1 :(得分:2)

你打赌!

我甚至不推荐使用shell_plus。我倾向于将我的实用程序脚本存储在我的app utility文件夹中。然后我只需要从cron作业中调用它们,或者根据需要手动调用它们。这是框架脚本我基于此。 (有点简化)

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import sys
import logging
import time
import time
import optparse

# DO NOT IMPORT DJANGO MODELS HERE - THIS NEED TO HAPPEN BELOW!!
# This needs to be able to be run when django isn't in the picture (cron) so we need
# to be able to add in the django paths when needed.

def getArgs():
    """
        Simply get the options for running update people.
    """

    p = optparse.OptionParser()

    help = "The Python path to a settings module, e.g. "
    help += "\"myproject.settings.main\". If this isn't provided, the "
    help += "DJANGO_SETTINGS_MODULE environment variable will be used."
    p.add_option("-s", "--settings", action = "store", type = "string",
                 dest = "settings", help = help)

    help = "A directory to add to the Python path, e.g."
    help += " \"/home/djangoprojects/myproject\"."
    p.add_option("-y", "--pythonpath", action = "store", type = "string",
                 dest = "pythonpath", help = help)
    p.add_option("-v", "--verbose", action = "count", dest = "verbose",
                 help = "Turn on verbose debugging")
    p.set_defaults(settings = os.environ.get("DJANGO_SETTINGS_MODULE",
                                              "settings"),
                   pythonpath = "", verbose = 0,
                   )

    return p.parse_args()


def update(opt, loglevel=None):
    """
        This is the main script used for updating people
    """

    start = time.time()

    # This ensures that our sys.path is ready.
    for path in opt.pythonpath.split(":"):
        if os.path.abspath(path) not in sys.path and os.path.isdir(path):
            sys.path.append(os.path.abspath(path))
    os.environ['DJANGO_SETTINGS_MODULE'] = opt.settings

    from django.conf import settings    
    try:
        if settings.SITE_ROOT not in sys.path: pass
    except ImportError:
        return("Your setting file cannot be imported - not in sys.path??")

    # IMPORT YOUR CODE MODELS HERE
    from apps.core.utility.ExampleExtractor import ExampleExtractor

    # YOUR DJANGO STUFF GOES HERE..
    example = ExampleExtractor(loglevel=loglevel, singleton=not(opt.multiple))
    raw = example.get_raw()
    results = example.update_django(raw)

    log.info("Time to update %s entries : %s" % (len(results), time.time() - start))
    return results


if __name__ == '__main__':

    logging.basicConfig(format = "%(asctime)s %(levelname)-8s %(module)s \
      %(funcName)s %(message)s", datefmt = "%H:%M:%S", stream = sys.stderr)
    log = logging.getLogger("")
    log.setLevel(logging.DEBUG)

    opts, args = getArgs()
    sys.exit(update(opts))

HTH!