forum_modules/exporter/views.py

   1 import os, tarfile, ConfigParser, datetime
   2
   3 from StringIO import StringIO
   4 from django.http import HttpResponse, HttpResponseRedirect, Http404
   5 from django.utils.translation import ugettext as _
   6 from django.utils import simplejson
   7 from django.core.cache import cache
   8 from django.core.urlresolvers import reverse
   9 from forum.views.admin import admin_tools_page, admin_page
  10 from forum.models import User
  11 from forms import ExporterForm
  12 from threading import Thread
  13 import settings as selsettings
  14 from forum import settings
  15
  16 from exporter import export, CACHE_KEY, EXPORT_STEPS, LAST_BACKUP, DATE_AND_AUTHOR_INF_SECTION, DATETIME_FORMAT
  17 from importer import start_import
  18
  19
  20 @admin_tools_page(_('exporter'), _('XML data export'))
  21 def exporter(request):
  22     state = cache.get(CACHE_KEY)
  23
  24     if state and state['running']:
  25         return HttpResponseRedirect(reverse('exporter_running', kwargs=dict(mode='exporter')))
  26
  27     if request.method == 'POST':
  28         form = ExporterForm(request.POST)
  29
  30         if form.is_valid():
  31             thread = Thread(target=export, args=[form.cleaned_data, request.user])
  32             thread.setDaemon(True)
  33             thread.start()
  34
  35             return HttpResponseRedirect(reverse('exporter_running', kwargs=dict(mode='exporter')))
  36     else:
  37         form = ExporterForm()
  38
  39     available = []
  40
  41     folder = unicode(selsettings.EXPORTER_BACKUP_STORAGE)
  42
  43     for f in os.listdir(folder):
  44         if (not os.path.isdir(os.path.join(folder, f))) and f.endswith('.backup.inf'):
  45             try:
  46                 with open(os.path.join(folder, f), 'r') as inffile:
  47                     inf = ConfigParser.SafeConfigParser()
  48                     inf.readfp(inffile)
  49
  50                     if inf.get(DATE_AND_AUTHOR_INF_SECTION, 'site') == settings.APP_URL and os.path.exists(
  51                                     os.path.join(folder, inf.get(DATE_AND_AUTHOR_INF_SECTION, 'file-name'))):
  52                         available.append({
  53                             'author': User.objects.get(id=inf.get(DATE_AND_AUTHOR_INF_SECTION, 'author')),
  54                             'date': datetime.datetime.strptime(inf.get(DATE_AND_AUTHOR_INF_SECTION, 'finished'), DATETIME_FORMAT)
  55                         })
  56             except Exception, e:
  57                 pass
  58
  59     return ('modules/exporter/exporter.html', {
  60         'form': form,
  61         'available': available,
  62     })
  63
  64 @admin_page
  65 def running(request, mode):
  66     state = cache.get(CACHE_KEY)
  67     if state is None:
  68         return HttpResponseRedirect(reverse('admin_tools', args=[_('exporter')]))
  69
  70     return ('modules/exporter/running.html', {
  71         'mode': mode,
  72         'steps': EXPORT_STEPS
  73     })
  74
  75 def state(request):
  76     return HttpResponse(simplejson.dumps(cache.get(CACHE_KEY)), mimetype="application/json")
  77
  78 @admin_page
  79 def download(request):
  80     fname = LAST_BACKUP
  81
  82     if not os.path.exists(fname):
  83         raise Http404
  84
  85     response = HttpResponse(open(fname, 'rb').read(), content_type='application/x-gzip')
  86     response['Content-Length'] = os.path.getsize(fname)
  87     response['Content-Disposition'] = 'attachment; filename=backup.tar.gz'
  88     return response
  89
  90
  91 @admin_page
  92 def importer(request):
  93     thread = Thread(target=start_import, args=['/Users/admin/dev/pyenv/osqa/maintain/forum_modules/exporter/backups/localhost-201010121118.tar.gz', request.user])
  94     thread.setDaemon(True)
  95     thread.start()
  96
  97     return HttpResponseRedirect(reverse('exporter_running', kwargs=dict(mode='importer')))
  98
  99     #return ('modules/exporter/importer.html', {
 100
 101     #})
 102
 103