]> git.openstreetmap.org Git - osqa.git/blob - forum_modules/exporter/views.py
e6b2e8b51b309372fcd163cf8a0c1581ba37cb5d
[osqa.git] / 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_tools_page(_('importer'), _('XML data restore'))
92 def importer(request):
93     thread = Thread(target=start_import, args=[
94             '/Users/admin/dev/pyenv/osqa/maintain/forum_modules/exporter/backups/localhost-201010121118.tar.gz',
95             {
96                 'sql2008': 'sql-server-2008',
97                 'sql2005': 'sql-server-2005',
98                 'sql2000': 'sql-server-2000',
99                 'design' : 'database-design',
100                 'fulltextsearch' : 'full-text',
101                 'access': 'microsoft-access',
102                 'subquery': 'sub-query',
103                 'count': 'aggregates',
104                 'join': 'joins',
105                 'metadata': 'meta-data',
106                 'stored-procedure': 'stored-procedures',
107                 'temp-tables': 'temporary-tables',
108                 'sqlce': 'sql-server-ce',
109                 'maintenance-plan': 'maintenance-plans',
110                 'meta': 'meta-askssc',
111                 'msaccess2000': 'microsoft-access',
112                 'agent': 'sql-agent',
113                 'udf': 'user-defined-function',
114                 'report': 'reporting',
115                 'ssas2005': 'ssas',
116                 'case': 'case-statement',
117                 'export': 'export-data',
118                 'recursive': 'recursion',
119                 'table-variables': 'table-variable',
120                 'sqldmo': 'dmo',
121                 'install': 'installation',
122                 'function': 'user-defined-function',
123                 'average': 'aggregates',
124                 'aggregate-function': 'aggregates',
125                 'email': 'database-email',
126                 'distinct': 'aggregates',
127                 'dynamic-query': 'dynamic',
128                 'learn': 'learning',
129                 'permission': 'permissions',
130                 'shrink': 'shrink-database',
131                 'normalise': 'normalization',
132                 'datatype-text': 'datatypes',
133                 'reporting-services': 'ssrs',
134                 'aggregate-sum': 'aggregates',
135                 'aggregate-max': 'aggregates',
136                 'bulk-import': 'bulk-insert',
137                 'attach-database': 'attach',
138                 'scripting': 'script',
139                 'analysis-services': 'ssas',
140                 'create-table': 'create',
141                 '2005': 'sql-server-2005'
142             },
143             request.user])
144     thread.setDaemon(True)
145     thread.start()
146
147     return HttpResponseRedirect(reverse('exporter_running', kwargs=dict(mode='importer')))
148
149     #return ('modules/exporter/importer.html', {
150
151     #})
152