django源码(2.0.2)粗解之命令行执行

前言

 

django的命令行在整个的django web开发中都会经常用到，而且是必须得用到。所以，能够了解下django的命令行实现其实是非常有帮助的。

如果大家比较关心django命令的详细说明和使用，可以查看这里。

 

命令行执行入口

 

django通过django-admin.py和manage.py来执行命令，以下是这两个文件的源码：

from django.core import management

if __name__ == "__main__":
    management.execute_from_command_line()

它们都调用了management模块下的execute_from_command_line()方法。

这个方法是在django/core/management/__init__.py中定义：

def execute_from_command_line(argv=None):
    """Run a ManagementUtility."""
    utility = ManagementUtility(argv)
    utility.execute()

实现非常简单：生成一个ManagementUtility对象，并让这个对象执行相应的命令行命令。所以主要的工作都是在ManagementUtility这个类中实现的。

 

ManagementUtility类

 

python是一门面向的对象的语言，django作为python的一个著名web框架，它所使用当然也是面向对象的思想。所以我们在分析源码的时候应该尽量用面向对象的思想去思考。

ManagementUtility具有3个属性，我们可以从它的__init__函数中看到。

 

    def __init__(self, argv=None):
        self.argv = argv or sys.argv[:] # 从传入的参数获得，如果没有传入参数就从sys.argv中去取
        self.prog_name = os.path.basename(self.argv[0])
        if self.prog_name == '__main__.py':
            self.prog_name = 'python -m django'
        self.settings_exception = None

self.argv：命令行信息，包括命令和参数

self.prog_name：程序名

self.settings_excepiton：settings的异常信息，发现settings的设置有异常，会将异常信息存在这个变量里面

 

ManagementUtility主要的方法是execute()，它完成了command执行的所有过程。

1. 我们知道，django的命令行是具有一定的格式的，都是 command subcommand [arguments]，arguments有时是可选的。所以execute方法第一步就是获得subcommand，以便确定后续执行什么任务。

 

        try:
            subcommand = self.argv[1]
        except IndexError:
            subcommand = 'help'  # Display help if no arguments were given.    

这里提一下，为什么不先获取command呢？其实command是系统用来找程序入口的。

2. 用命令解析器CommandParser解析命令行。CommandParser继承了argparse模块的ArgumentParser类，但它只是对ArgumentParser的异常处理进行了加强。

 

class CommandParser(ArgumentParser):
    """
    Customized ArgumentParser class to improve some error messages and prevent
    SystemExit in several occasions, as SystemExit is unacceptable when a
    command is called programmatically.
    """
    def __init__(self, cmd, **kwargs):
        self.cmd = cmd
        super().__init__(**kwargs)

    def parse_args(self, args=None, namespace=None):
        # Catch missing argument for a better error message
        if (hasattr(self.cmd, 'missing_args_message') and
                not (args or any(not arg.startswith('-') for arg in args))):
            self.error(self.cmd.missing_args_message)
        return super().parse_args(args, namespace)

    def error(self, message):
        if self.cmd._called_from_command_line:
            super().error(message)
        else:
            raise CommandError("Error: %s" % message)

 

argparse官方文档 | argparse用法总结

3. 解析器解析出了subcommand的arguments，然后fetch_command根据subcommand导入相应的command包并生成相应的command对象，然后调用command对象的print_help方法或者run_from_argv方法去执行相应的命令。

        if subcommand == 'help':
            if '--commands' in args:    # only print the commands only
                sys.stdout.write(self.main_help_text(commands_only=True) + '\n')
            elif len(options.args) < 1: # print out the usages
                sys.stdout.write(self.main_help_text() + '\n')
            else: 
                self.fetch_command(options.args[0]).print_help(self.prog_name, options.args[0])
        # Special-cases: We want 'django-admin --version' and
        # 'django-admin --help' to work, for backwards compatibility.
        elif subcommand == 'version' or self.argv[1:] == ['--version']:
            sys.stdout.write(django.get_version() + '\n')
        elif self.argv[1:] in (['--help'], ['-h']):
            sys.stdout.write(self.main_help_text() + '\n')
        else:
            self.fetch_command(subcommand).run_from_argv(self.argv)        

最后看一眼fetch_command的代码：

    def fetch_command(self, subcommand):
        """
        Try to fetch the given subcommand, printing a message with the
        appropriate command called from the command line (usually
        "django-admin" or "manage.py") if it can't be found.
        """
        # Get commands outside of try block to prevent swallowing exceptions
        commands = get_commands()
        try:
            app_name = commands[subcommand]
        except KeyError:
            if os.environ.get('DJANGO_SETTINGS_MODULE'):
                # If `subcommand` is missing due to misconfigured settings, the
                # following line will retrigger an ImproperlyConfigured exception
                # (get_commands() swallows the original one) so the user is
                # informed about it.
                settings.INSTALLED_APPS
            else:
                sys.stderr.write("No Django settings specified.\n")
            sys.stderr.write(
                "Unknown command: %r\nType '%s help' for usage.\n"
                % (subcommand, self.prog_name)
            )
            sys.exit(1)
        if isinstance(app_name, BaseCommand):
            # If the command is already loaded, use it directly.
            klass = app_name
        else:
            klass = load_command_class(app_name, subcommand)
        return klass

这里主要用了load_command_class去导入相应的subcommand模块，并生成了一个command类对象。

def load_command_class(app_name, name):
    """
    Given a command name and an application name, return the Command
    class instance. Allow all errors raised by the import process
    (ImportError, AttributeError) to propagate.
    """
    module = import_module('%s.management.commands.%s' % (app_name, name))
    #print("import %s %s" %(app_name, name))
    return module.Command()

我们可以去django/core/management/command/目录下随便找一个command模块看一眼，比如说check.py，每个模块都有一个command类并继承自BaseCommand。上面提到的print_help方法或者run_from_argv方法都是在BaseCommand类中实现。

class Command(BaseCommand):
    help = "Checks the entire Django project for potential problems."

    requires_system_checks = False
    
    #...

 

 

-------------------------------------------------

以上是我的一点粗浅的理解，如果有觉得不对的地方，请多多指教，非常感谢！

2018-03-21 17:57:17