split surveys->scans + drawings

1 files changed, 98 insertions, 0 deletions

diff --git a/core/views/drawings.py b/core/views/drawings.py
new file mode 100644
index 0000000..7657d3d
--- /dev/null
+++ b/core/views/drawings.py
@@ -0,0 +1,98 @@
+import os, stat
+import re
+from pathlib import Path
+from urllib.parse import urljoin, unquote as urlunquote
+
+from django.conf import settings
+from django.shortcuts import render
+from django.http import HttpResponse
+
+from troggle.core.models.survex import DrawingFile
+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 invalid query string is invalid, or produces multiple replies
+and render a user-friendly error page.
+'''
+
+def dwgdata(request):
+    '''Report on all the drawing files in the system. These were loaded by parsing the entire directory tree
+    '''
+    dwgfiles = DrawingFile.objects.all()
+    return render(request, 'dwgfiles.html', { 'dwgfiles':dwgfiles, '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
+    '''
+    try:
+        dwgfile = DrawingFile.objects.get(dwgpath=urlunquote(path)) 
+    except:
+        message = f'Drawing file error or not found \'{path}\' .'
+        return render(request, 'errors/generic.html', {'message': message})   
+        
+    tfile = Path(settings.DRAWINGS_DATA, dwgfile.dwgpath)
+    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 dwgfileupload(request, path):
+    '''uploads a drawing file, but where is the Form? This just processes POST info. Apparently unfinished?
+    '''
+    try:
+        dwgfile = DrawingFile.objects.get(dwgpath=urlunquote(path)) # need to check if inavlid query string and produce friendly error
+    except:
+        message = f'Drawing file error or not found \'{path}\' .'
+        return render(request, 'errors/generic.html', {'message': message})   
+    tfile = Path(settings.DRAWINGS_DATA, dwgfile.dwgpath)
+    
+    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 = dwgfile.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(dwgfile) # commented out
+    dwgfile.save()
+    
+    uploadedfile.close()
+    message = "File size %d overwritten with size %d" % (orgsize, dwgfile.filesize)
+    return HttpResponse(content=message, content_type="text/plain")
+