troggle-unchained/flatpages/views.py
Wookey f60c4cba23 Add odt and ods mime types to our list.
Maybe this should just be read in from the real list...
2011-07-12 00:57:48 +01:00

128 lines
5.0 KiB
Python

import troggle.settings as settings
from troggle.helper import login_required_if_public
from utils import render_with_context
from django.http import HttpResponse, HttpResponseRedirect, Http404
from django.core.urlresolvers import reverse
from django.template import Context, loader
import django.forms as forms
from tinymce.widgets import TinyMCE
from troggle.flatpages.models import Redirect, EntranceRedirect
from troggle.core.models import Cave
import troggle.core.views_caves
import os
import re
def flatpage(request, path):
print "gggggg", path
try:
r = Redirect.objects.get(originalURL = path)
return HttpResponseRedirect(r.newURL) # Redirect after POST
except Redirect.DoesNotExist:
pass
try:
r = Cave.objects.get(url = path)
return troggle.core.views_caves.caveSlug(request, r.slug)
except Cave.DoesNotExist:
pass
try:
r = EntranceRedirect.objects.get(originalURL = path)
return troggle.core.views_caves.enranceSlug(request, r.entrance.slug)
except EntranceRedirect.DoesNotExist:
pass
if path.startswith("noinfo") and settings.PUBLIC_SITE and not request.user.is_authenticated():
return HttpResponseRedirect(reverse("auth_login") + '?next=%s' % request.path)
if path.endswith("/") or path == "":
try:
o = open(os.path.normpath(settings.EXPOWEB + path + "index.html"), "rb")
path = path + "index.html"
except IOError:
try:
o = open(os.path.normpath(settings.EXPOWEB + path + "index.htm"), "rb")
path = path + "index.html"
except IOError:
raise Http404
else:
try:
o = open(os.path.normpath(settings.EXPOWEB + path), "rb")
except IOError:
raise Http404
if path.endswith(".htm") or path.endswith(".html"):
html = o.read()
m = re.search(r"<head>(.*)</head>.*<body>(.*)</body>", html, re.DOTALL)
mwithid = re.search(r'<head>(.*)</head>.*<body id="([^"]*)">(.*)</body>', html, re.DOTALL)
if m:
head, body = m.groups()
bodyid = None
elif mwithid:
head, bodyid, body = mwithid.groups()
else:
return HttpResponse(html + "Page could not be split into header and body")
if re.search(r"iso-8859-1", html):
body = unicode(body, "iso-8859-1")
return render_with_context(request, 'flatpage.html', {'editable': True, 'path': path, 'head': head, 'body': body, "bodyid": bodyid})
else:
return HttpResponse(o.read(), mimetype=getmimetype(path))
def getmimetype(path):
if path.endswith(".png"): return "image/png"
if path.endswith(".tif"): return "image/tif"
if path.endswith(".gif"): return "image/gif"
if path.endswith(".jpeg"): return "image/jpeg"
if path.endswith(".jpg"): return "image/jpeg"
if path.endswith("svg"): return "image/svg+xml"
if path.endswith(".pdf"): return "application/pdf"
if path.endswith(".ps"): return "application/postscript"
if path.endswith(".svx"): return "application/x-survex-svx"
if path.endswith(".3d"): return "application/x-survex-3d"
if path.endswith(".pos"): return "application/x-survex-pos"
if path.endswith(".err"): return "application/x-survex-err"
if path.endswith(".odt"): return "application/vnd.oasis.opendocument.text"
if path.endswith(".ods"): return "application/vnd.oasis.opendocument.spreadsheet"
return ""
@login_required_if_public
def editflatpage(request, path):
try:
r = Cave.objects.get(url = path)
return troggle.core.views_caves.editCave(request, r.cave.slug)
except Cave.DoesNotExist:
pass
try:
filepath = os.path.normpath(settings.EXPOWEB + path)
o = open(filepath, "r")
except IOError:
raise Http404
html = o.read()
m = re.search(r"<head>(.*)</head>.*<body>(.*)</body>", html, re.DOTALL)
if m:
head, body = m.groups()
if re.search(r"iso-8859-1", html):
body = unicode(body, "iso-8859-1")
else:
return HttpResponse("Page could not be split into header and body")
if request.method == 'POST': # If the form has been submitted...
flatpageForm = FlatPageForm(request.POST) # A form bound to the POST data
if flatpageForm.is_valid():# Form valid therefore write file
f = open(filepath, "w")
template = loader.get_template('dataformat/flatfile.html')
context = Context({'form': flatpageForm.cleaned_data, 'head': head})
f.write(template.render(context))
f.close()
return HttpResponseRedirect(reverse('flatpage', args=[path])) # Redirect after POST
else:
flatpageForm = FlatPageForm({"html": body})
return render_with_context(request, 'editflatpage.html', {'path': path, 'form': flatpageForm, })
class FlatPageForm(forms.Form):
html = forms.CharField(widget=TinyMCE(attrs={'cols': 80, 'rows': 30}))