def docs_action(request, mode, docid):
doc = docs.objects.get(pk=docid)
if mode == 'view':
form = DocsForm(user=request.user, instance=doc, view_only=True)
add_breadcrumbs(request, docid, '*', caption=doc.caption[:20])
files = docs_files.objects.filter(doc=doc, active_record=True)
paginator = Paginator(files, 10)
page = request.GET.get('page')
try:
files_lines = paginator.page(page)
except PageNotAnInteger:
# If page is not an integer, deliver first page.
files_lines = paginator.page(1)
except EmptyPage:
# If page is out of range (e.g. 9999), deliver last page of results.
files_lines = paginator.page(paginator.num_pages)
return render(request, 'docs/view.html', {'layout': 'horizontal', 'form': form, 'doc': doc, 'files': files_lines})
elif mode == 'edit':
if request.method == 'POST':
form = DocsForm(request.POST, user=request.user, instance=doc)
if form.is_valid():
doc = form.save()
return HttpResponseRedirect('/docs/view/%s/' % doc.pk)
form = DocsForm(user=request.user, instance=doc)
return render(request, 'docs/edit.html', {'layout': 'horizontal', 'form': form, 'doc': doc})
elif mode == 'delete':
doc.delete(user=request.user)
return HttpResponseRedirect('/search/?q=%s&models=yats.docs&models=web.test' % request.session.get('last_fulltextsearch', '*'))
elif mode == 'ticket':
ticket = get_ticket_model()
tic = ticket()
tic.caption = doc.caption
tic.description = doc.text
if hasattr(settings, 'KEEP_IT_SIMPLE_DEFAULT_CUSTOMER') and settings.KEEP_IT_SIMPLE_DEFAULT_CUSTOMER:
if settings.KEEP_IT_SIMPLE_DEFAULT_CUSTOMER == -1:
tic.customer = request.organisation
else:
tic.customer_id = settings.KEEP_IT_SIMPLE_DEFAULT_CUSTOME
if not tic.component_id and hasattr(settings, 'KEEP_IT_SIMPLE_DEFAULT_COMPONENT') and settings.KEEP_IT_SIMPLE_DEFAULT_COMPONENT:
tic.component_id = settings.KEEP_IT_SIMPLE_DEFAULT_COMPONENT
tic.save(user=request.user)
add_breadcrumbs(request, str(tic.pk), '#', caption=tic.caption[:20])
if hasattr(settings, 'KEEP_IT_SIMPLE') and settings.KEEP_IT_SIMPLE:
return HttpResponseRedirect('/tickets/simple/%s/' % tic.pk)
else:
return HttpResponseRedirect('/tickets/edit/%s/' % tic.pk)
elif mode == 'download':
fileid = request.GET.get('file', -1)
file_data = docs_files.objects.get(id=fileid, doc=doc)
src = '%s%s.dat' % (get_doc_files_folder(), fileid)
content_type = file_data.content_type
if request.GET.get('preview') == 'yes' and os.path.isfile('%s%s.preview' % (get_doc_files_folder(), fileid)):
src = '%s%s.preview' % (get_doc_files_folder(), fileid)
content_type = 'imgae/png'
if request.GET.get('resize', 'no') == 'yes' and ('image' in file_data.content_type or 'pdf' in file_data.content_type):
img = resize_image('%s' % (src), (200, 150), 75)
output = io.BytesIO()
img.save(output, 'PNG')
output.seek(0)
response = StreamingHttpResponse(output, content_type='image/png')
else:
response = StreamingHttpResponse(open('%s' % (src), "rb"), content_type=content_type)
if 'noDisposition' not in request.GET:
if request.GET.get('preview') == 'yes' and os.path.isfile('%s%s.preview' % (get_doc_files_folder(), fileid)):
response['Content-Disposition'] = 'attachment;filename=%s' % content_type
else:
response['Content-Disposition'] = 'attachment;filename=%s' % smart_str(file_data.name)
return response
elif mode == 'upload':
if request.method == 'POST':
form = UploadFileForm(request.POST, request.FILES)
if form.is_valid():
if docs_files.objects.filter(active_record=True, doc=doc, checksum=request.FILES['file'].hash).count() > 0:
messages.add_message(request, messages.ERROR, _('File already exists: %s') % request.FILES['file'].name)
if request.GET.get('Ajax') == '1':
return HttpResponse('OK')
return HttpResponseRedirect('/docs/view/%s/' % doc.id)
f = docs_files()
f.name = request.FILES['file'].name
f.size = request.FILES['file'].size
f.checksum = request.FILES['file'].hash
f.content_type = request.FILES['file'].content_type
f.doc_id = doc.id
f.public = True
f.save(user=request.user)
#add_history(request, tic, 5, request.FILES['file'].name)
dest = get_doc_files_folder()
if not os.path.exists(dest):
os.makedirs(dest)
with open('%s%s.dat' % (dest, f.id), 'wb+') as destination:
for chunk in request.FILES['file'].chunks():
destination.write(chunk)
if 'pdf' in f.content_type:
convertPDFtoImg('%s/%s.dat' % (dest, f.id), '%s/%s.preview' % (dest, f.id))
else:
if 'image' not in f.content_type and isPreviewable(f.content_type):
tmp = convertOfficeTpPDF('%s/%s.dat' % (dest, f.id))
convertPDFtoImg(tmp, '%s/%s.preview' % (dest, f.id))
if os.path.isfile(tmp):
os.unlink(tmp)
return HttpResponseRedirect('/docs/view/%s/' % doc.pk)
else:
msg = unicode(form.errors['file'])
msg = re.sub('<[^<]+?>', '', msg)
messages.add_message(request, messages.ERROR, msg)
if request.GET.get('Ajax') == '1':
return HttpResponse('OK')
return HttpResponseRedirect('/docs/view/%s/' % doc.pk)
else:
form = UploadFileForm()
return render(request, 'docs/file.html', {'docid': doc, 'layout': 'horizontal', 'form': form})
elif mode == 'delfile':
file = docs_files.objects.get(pk=request.GET['fileid'], doc=doc)
file.delete(user=request.user)
#add_history(request, tic, 8, file.name)
return HttpResponseRedirect('/docs/view/%s/#files' % doc.pk)
def handle(self, *args, **options):
if options['username']:
username = options['username']
else:
raise CommandError('missing username')
if options['server']:
server = options['server']
else:
raise CommandError('missing server')
if options['yatsuser']:
yatsuser = options['yatsuser']
else:
raise CommandError('missing yatsuser')
password = self._get_pass()
user = get_user_model().objects.get(username=yatsuser)
url = furl(server)
url.password = password
url.username = username
rpc_srv = xmlrpclib.ServerProxy(url.tostr(), allow_none=True, use_datetime=True)
documents = rpc_srv.wiki.getAllPages()
for doc in documents:
if doc.startswith('Trac'):
continue
if doc.startswith('Wiki'):
continue
page = rpc_srv.wiki.getPage(doc)
info = rpc_srv.wiki.getPageInfo(doc)
# print info
d = docs()
d.caption = doc
d.text = convert(page, '/docs/').strip()
d.u_date = info['lastModified']
d.save(user=user)
atts = rpc_srv.wiki.listAttachments(doc)
for att in atts:
print '=> %s' % att
path, filename = os.path.split(att)
file = rpc_srv.wiki.getAttachment(att)
mimetype = mimetypes.guess_type(filename)
f = docs_files()
f.name = filename
f.size = 0
f.content_type = mimetype[0] if mimetype[0] else ''
f.doc_id = d.id
f.public = True
f.save(user=user)
dest = get_doc_files_folder()
if not os.path.exists(dest):
os.makedirs(dest)
with open('%s%s.dat' % (dest, f.id), 'wb') as destination:
destination.write(file.data)
destination.close()
hash_md5 = hashlib.md5()
with open('%s%s.dat' % (dest, f.id), "rb") as fo:
for chunk in iter(lambda: fo.read(4096), b""):
hash_md5.update(chunk)
f.checksum = hash_md5.hexdigest()
f.save(user=user)
if 'pdf' in f.content_type:
convertPDFtoImg('%s/%s.dat' % (dest, f.id), '%s/%s.preview' % (dest, f.id))
else:
if 'image' not in f.content_type and isPreviewable(f.content_type):
tmp = convertOfficeTpPDF('%s/%s.dat' % (dest, f.id))
convertPDFtoImg(tmp, '%s/%s.preview' % (dest, f.id))
if os.path.isfile(tmp):
os.unlink(tmp)
self.stdout.write('%s => %s' % (doc, d.pk))
def docs_action(request, mode, docid):
doc = docs.objects.get(pk=docid)
if mode == 'view':
form = DocsForm(user=request.user, instance=doc, view_only=True)
add_breadcrumbs(request, docid, '*', caption=doc.caption[:20])
files = docs_files.objects.filter(doc=doc, active_record=True)
paginator = Paginator(files, 10)
page = request.GET.get('page')
try:
files_lines = paginator.page(page)
except PageNotAnInteger:
# If page is not an integer, deliver first page.
files_lines = paginator.page(1)
except EmptyPage:
# If page is out of range (e.g. 9999), deliver last page of results.
files_lines = paginator.page(paginator.num_pages)
return render(request, 'docs/view.html', {'layout': 'horizontal', 'form': form, 'doc': doc, 'files': files_lines})
elif mode == 'edit':
if request.method == 'POST':
form = DocsForm(request.POST, user=request.user, instance=doc)
if form.is_valid():
doc = form.save()
return HttpResponseRedirect('/docs/view/%s/' % doc.pk)
form = DocsForm(user=request.user, instance=doc)
return render(request, 'docs/edit.html', {'layout': 'horizontal', 'form': form, 'doc': doc})
elif mode == 'delete':
doc.delete(user=request.user)
return HttpResponseRedirect('/search/?q=%s&models=yats.docs&models=web.test' % request.session.get('last_fulltextsearch', '*'))
elif mode == 'ticket':
ticket = get_ticket_model()
tic = ticket()
tic.caption = doc.caption
tic.description = doc.text
if hasattr(settings, 'KEEP_IT_SIMPLE_DEFAULT_CUSTOMER') and settings.KEEP_IT_SIMPLE_DEFAULT_CUSTOMER:
if settings.KEEP_IT_SIMPLE_DEFAULT_CUSTOMER == -1:
tic.customer = request.organisation
else:
tic.customer_id = settings.KEEP_IT_SIMPLE_DEFAULT_CUSTOME
if not tic.component_id and hasattr(settings, 'KEEP_IT_SIMPLE_DEFAULT_COMPONENT') and settings.KEEP_IT_SIMPLE_DEFAULT_COMPONENT:
tic.component_id = settings.KEEP_IT_SIMPLE_DEFAULT_COMPONENT
tic.save(user=request.user)
# add ref to doc
com = tickets_comments()
com.comment = _('see §%s') % doc.pk
com.ticket_id = tic
com.action = 3
com.save(user=request.user)
add_breadcrumbs(request, str(tic.pk), '#', caption=tic.caption[:20])
if hasattr(settings, 'KEEP_IT_SIMPLE') and settings.KEEP_IT_SIMPLE:
return HttpResponseRedirect('/tickets/simple/%s/' % tic.pk)
else:
return HttpResponseRedirect('/tickets/edit/%s/' % tic.pk)
elif mode == 'download':
fileid = request.GET.get('file', -1)
file_data = docs_files.objects.get(id=fileid, doc=doc)
src = '%s%s.dat' % (get_doc_files_folder(), fileid)
content_type = file_data.content_type
if request.GET.get('preview') == 'yes' and os.path.isfile('%s%s.preview' % (get_doc_files_folder(), fileid)):
src = '%s%s.preview' % (get_doc_files_folder(), fileid)
content_type = 'imgae/png'
if request.GET.get('resize', 'no') == 'yes' and ('image' in file_data.content_type or 'pdf' in file_data.content_type):
img = resize_image('%s' % (src), (200, 150), 75)
output = io.BytesIO()
img.save(output, 'PNG')
output.seek(0)
response = StreamingHttpResponse(output, content_type='image/png')
else:
response = StreamingHttpResponse(open('%s' % (src), "rb"), content_type=content_type)
if 'noDisposition' not in request.GET:
if request.GET.get('preview') == 'yes' and os.path.isfile('%s%s.preview' % (get_doc_files_folder(), fileid)):
response['Content-Disposition'] = 'attachment;filename=%s' % content_type
else:
response['Content-Disposition'] = 'attachment;filename=%s' % smart_str(file_data.name)
return response
elif mode == 'upload':
if request.method == 'POST':
form = UploadFileForm(request.POST, request.FILES)
if form.is_valid():
if docs_files.objects.filter(active_record=True, doc=doc, checksum=request.FILES['file'].hash).count() > 0:
messages.add_message(request, messages.ERROR, _('File already exists: %s') % request.FILES['file'].name)
if request.GET.get('Ajax') == '1':
return HttpResponse('OK')
return HttpResponseRedirect('/docs/view/%s/' % doc.id)
f = docs_files()
f.name = request.FILES['file'].name
f.size = request.FILES['file'].size
f.checksum = request.FILES['file'].hash
f.content_type = request.FILES['file'].content_type
f.doc_id = doc.id
f.public = True
f.save(user=request.user)
#add_history(request, tic, 5, request.FILES['file'].name)
dest = get_doc_files_folder()
if not os.path.exists(dest):
os.makedirs(dest)
with open('%s%s.dat' % (dest, f.id), 'wb+') as destination:
for chunk in request.FILES['file'].chunks():
destination.write(chunk)
if 'pdf' in f.content_type:
convertPDFtoImg('%s/%s.dat' % (dest, f.id), '%s/%s.preview' % (dest, f.id))
else:
if 'image' not in f.content_type and isPreviewable(f.content_type):
tmp = convertOfficeTpPDF('%s/%s.dat' % (dest, f.id))
convertPDFtoImg(tmp, '%s/%s.preview' % (dest, f.id))
if os.path.isfile(tmp):
os.unlink(tmp)
return HttpResponseRedirect('/docs/view/%s/' % doc.pk)
else:
msg = unicode(form.errors['file'])
msg = re.sub('<[^<]+?>', '', msg)
messages.add_message(request, messages.ERROR, msg)
if request.GET.get('Ajax') == '1':
return HttpResponse('OK')
return HttpResponseRedirect('/docs/view/%s/' % doc.pk)
else:
form = UploadFileForm()
return render(request, 'docs/file.html', {'docid': doc, 'layout': 'horizontal', 'form': form})
elif mode == 'delfile':
file = docs_files.objects.get(pk=request.GET['fileid'], doc=doc)
file.delete(user=request.user)
#add_history(request, tic, 8, file.name)
return HttpResponseRedirect('/docs/view/%s/#files' % doc.pk)