forked from expo/troggle
109 lines
4.7 KiB
Python
109 lines
4.7 KiB
Python
import os, stat
|
|
import re
|
|
from pathlib import Path
|
|
from urllib.parse import urljoin, unquote as urlunquote
|
|
from urllib.request import urlopen
|
|
|
|
from django.conf import settings
|
|
from django.shortcuts import render
|
|
from django.http import HttpResponse, Http404
|
|
|
|
from troggle.core.models_survex import ScansFolder, SingleScan, SurvexBlock, TunnelFile
|
|
from troggle.core.views.expo import getmimetype
|
|
import parsers.surveys
|
|
|
|
'''Some of these views serve files as binary blobs, and simply set the mime type based on the file extension,
|
|
as does the urls.py dispatcher which sends them here. Here they should actually have the filetype checked
|
|
by looking inside the file before being served.
|
|
|
|
need to check if inavlid query string is invalid, or produces multiple replies
|
|
and render a user-friendly error page.
|
|
'''
|
|
|
|
def surveyscansfolder(request, path):
|
|
#print [ s.walletname for s in ScansFolder.objects.all() ]
|
|
scansfolder = ScansFolder.objects.get(walletname=urlunquote(path)) # need to check if inavlid query string and produce friendly error
|
|
return render(request, 'scansfolder.html', { 'scansfolder':scansfolder, 'settings': settings })
|
|
|
|
def surveyscansingle(request, path, file):
|
|
'''sends a single binary file to the user,
|
|
'''
|
|
scansfolder = ScansFolder.objects.get(walletname=urlunquote(path)) # need to check if inavlid query string and produce friendly error
|
|
singlescan = SingleScan.objects.get(scansfolder=scansfolder, name=file)
|
|
# print(" - surveyscansingle {}:{}:{}:".format(path, file, getmimetype(file)))
|
|
return HttpResponse(content=open(singlescan.ffile,"rb"), content_type=getmimetype(file)) # any type of image
|
|
|
|
|
|
def surveyscansfolders(request):
|
|
manyscansfolders = ScansFolder.objects.all()
|
|
return render(request, 'manyscansfolders.html', { 'manyscansfolders':manyscansfolders, 'settings': settings })
|
|
|
|
|
|
def tunneldata(request):
|
|
tunnelfiles = TunnelFile.objects.all()
|
|
return render(request, 'tunnelfiles.html', { 'tunnelfiles':tunnelfiles, 'settings': settings })
|
|
|
|
|
|
def dwgfilesingle(request, path):
|
|
'''sends a single binary file to the user, We should have a renderer that syntax-colours this Tunnel xml
|
|
but it might be a Therion file
|
|
'''
|
|
tunnelfile = TunnelFile.objects.get(tunnelpath=urlunquote(path)) # need to check if invalid query string and produce friendly error
|
|
tfile = Path(settings.TUNNEL_DATA, tunnelfile.tunnelpath)
|
|
try: # for display not download
|
|
return HttpResponse(content=open(tfile, errors='strict'), content_type="text/xhtml")
|
|
except UnicodeDecodeError:
|
|
try:
|
|
return HttpResponse(content=open(tfile,encoding='iso-8859-1'), content_type="text/xhtml")
|
|
except:
|
|
return HttpResponse(content=open(tfile,mode='rb'), content_type="text/xhtml")
|
|
else:
|
|
return HttpResponse(content=open(tfile, errors='ignore'), content_type="text/xhtml")
|
|
else:
|
|
return HttpResponse(content="Unable to understand the encoding for this file: not UTF-8 nor iso-8859-1, or some other read error happened.")
|
|
|
|
|
|
def tunnelfileupload(request, path):
|
|
tunnelfile = TunnelFile.objects.get(tunnelpath=urlunquote(path)) # need to check if inavlid query string and produce friendly error
|
|
tfile = Path(settings.TUNNEL_DATA, tunnelfile.tunnelpath)
|
|
|
|
project, user, password, tunnelversion = request.POST["tunnelproject"], request.POST["tunneluser"], request.POST["tunnelpassword"], request.POST["tunnelversion"]
|
|
print((project, user, tunnelversion))
|
|
|
|
|
|
if not (len(list(request.FILES.values())) == 1): # "only one file to upload"
|
|
return HttpResponse(content="Error: more than one file selected for upload", content_type="text/plain")
|
|
|
|
uploadedfile = list(request.FILES.values())[0]
|
|
|
|
if uploadedfile.field_name != "sketch":
|
|
return HttpResponse(content="Error: non-sketch file uploaded", content_type="text/plain")
|
|
if uploadedfile.content_type != "text/plain":
|
|
return HttpResponse(content="Error: non-plain content type", content_type="text/plain")
|
|
|
|
# could use this to add new files
|
|
if os.path.split(path)[1] != uploadedfile.name:
|
|
return HttpResponse(content="Error: name disagrees", content_type="text/plain")
|
|
|
|
orgsize = tunnelfile.filesize # = os.stat(tfile)[stat.ST_SIZE]
|
|
|
|
ttext = uploadedfile.read()
|
|
|
|
# could check that the user and projects agree here
|
|
|
|
fout = open(tfile, "w")
|
|
fout.write(ttext)
|
|
fout.close()
|
|
|
|
# redo its settings of
|
|
parsers.surveys.SetTunnelfileInfo(tunnelfile)
|
|
tunnelfile.save()
|
|
|
|
uploadedfile.close()
|
|
message = "File size %d overwritten with size %d" % (orgsize, tunnelfile.filesize)
|
|
return HttpResponse(content=message, content_type="text/plain")
|
|
|
|
|
|
|
|
|