Запись в модель из своей команды - Python
Формулировка задачи:
Доброго времени суток. Как из своей команды записывать данные в нужную модель.? Сейчас мой скрипт/команда записывает данные напрямую в БД.
Спасибо!
Решение задачи: «Запись в модель из своей команды»
textual
Листинг программы
from django.core.management.base import BaseCommand, CommandError
from polls.models import Poll
class Command(BaseCommand):
help = 'Closes the specified poll for voting'
def add_arguments(self, parser):
parser.add_argument('poll_id', nargs='+', type=int)
def handle(self, *args, **options):
for poll_id in options['poll_id']:
try:
poll = Poll.objects.get(pk=poll_id)
except Poll.DoesNotExist:
raise CommandError('Poll "%s" does not exist' % poll_id)
poll.opened = False
poll.save()
self.stdout.write('Successfully closed poll "%s"' % poll_id)