Django snippets: Management command decorator

from django.core.management.base import BaseCommand

def command(help=None, args=''):
    def _decorate(func):
        _args = args
        _help = help
        class Command(BaseCommand):
            help = _help if _help is not None else func.__doc__
            args = _args
            def handle(self, *args, **kwargs):
                return func(*args, **kwargs)
        func.func_globals['Command'] = Command
        return func
    return _decorate

More like this

Quick-and-dirty decorator for static navigation context by pablo_PXL 2 years, 3 months ago
django_template decorator by fredd4 5 years, 3 months ago
Cache Any Function by jeffwheeler 6 years, 2 months ago
Python-like string interpolation in Javascript by phxx 2 years, 11 months ago
update primary key (and cascade to child tables) by guettli 1 year, 2 months ago

A quick-and-dirty, and extremely simple, decorator to turn a simple function into a management command.

This still requires you to have the management directory structure, but allows you to name your primary method whatever you want, and encapsulates the basic functionality of an argument-accepting management commmand.

The function's docstring will be used for the command's help text if the help arg is not passed to the decorator.

Simple usage:

from myapp.utils import command

@command()
def my_command():
    print "Hello, world"

I'm not too familiar with the intricacies of decorators and management commands, so this could probably (most likely) be improved upon, but it's a start.

Update: I've taken this a bit farther and put my work up on bitbucket: https://bitbucket.org/eternicode/django-management-decorators/src

Author:: eternicode
Posted:: August 31, 2010
Language:: Python
Django Version:: 1.1
Tags:: decorator management command
Score:: 2 (after 2 ratings)

Tools

Comments

gmandx (on August 31, 2010):

Well, some how (and I think it can be done) the decorator can introspect the function's positional arguments and keyword arguments, and then use those for the command's arguments and options, respectively. Then the only parameter needed for the decorator usage would be the help text.

eternicode (on August 31, 2010):

gmandx, interesting idea. I've done some light reading around, and it looks like the functools module has some stuff that could do this.

Django snippets

Management command decorator

More like this

Tools

Comments

Snippets

About