apps/api/management/commands/doc_update.py

   1 #!/usr/bin/env python
   2 # -*- conding: utf-8 -*-
   3 __author__="lreqc"
   4 __date__ ="$2009-09-08 14:31:26$"
   5
   6 from django.core.management.base import BaseCommand
   7 from django.utils import simplejson as json
   8 from django.test.client import Client
   9 from django.core.urlresolvers import reverse
  10
  11 from optparse import make_option
  12
  13 class Command(BaseCommand):
  14
  15     option_list = BaseCommand.option_list + (
  16         make_option('-u', '--user', action='store', dest='username'),
  17         make_option('-p', '--password', action='store', dest='password'),
  18     )
  19
  20     def handle(self, *args, **options):
  21         client = Client()
  22         if not options['username'] or not options['password']:
  23             raise CommandError("You must provide login data")
  24
  25         client.login(username=options['username'], \
  26             password=options['password'])
  27
  28         print options['username'], options['password']
  29
  30         filename = args[0]
  31         docid = args[1]
  32
  33         url = reverse("document_view", args=[docid])
  34         print "Quering %s" % url
  35         resp = client.get(url)
  36
  37         result = json.loads(resp.content)
  38         print result
  39
  40         print "Current revision for '%s': %s" % (docid, result['user_revision'])
  41         url = reverse("doctext_view", args=[ docid, result['user_revision'] ])
  42         print "Sending PUT to %s" % url
  43
  44         resp = client.put(url, {
  45             'contents': open(filename).read(),
  46             'message': options['message'] or ''
  47
  48         })
  49
  50         print resp.status_code, resp.content