import troggle.settings as settings import fileAbstraction from django.shortcuts import render_to_response from django.http import HttpResponse, Http404 import os import re # inline fileabstraction into here if it's not going to be useful anywhere else # keep things simple and ignore exceptions everywhere for now def getMimeType(extension): try: return {"txt": "text/plain", "html": "text/html", }[extension] except: print "unknown file type" return "text/plain" def listdir(request, path): #try: return HttpResponse(fileAbstraction.listdir(path), mimetype = "text/plain") #except: # raise Http404 def upload(request, path): pass def download(request, path): #try: return HttpResponse(fileAbstraction.readFile(path), mimetype=getMimeType(path.split(".")[-1])) #except: # raise Http404 # # julian's quick hack for something that works # could signal directories by ending with /, and forward cases where it's missing # extmimetypes = {".txt": "text/plain", ".html": "text/html", ".png": "image/png", ".jpg": "image/jpeg", } def jgtfile(request, f): fp = os.path.join(settings.SURVEYS, f) # could also surf through SURVEX_DATA # directory listing if os.path.isdir(fp): listdirfiles = [ ] listdirdirs = [ ] for lf in sorted(os.listdir(fp)): hpath = os.path.join(f, lf) # not absolute path if lf[0] == "." or lf[-1] == "~": continue hpath = hpath.replace("\\", "/") # for windows users href = hpath.replace("#", "%23") # '#' in file name annoyance flf = os.path.join(fp, lf) if os.path.isdir(flf): nfiles = len([sf for sf in os.listdir(flf) if sf[0] != "."]) listdirdirs.append((href, hpath + "/", nfiles)) else: listdirfiles.append((href, hpath, os.path.getsize(flf))) upperdirs = [ ] lf = f while lf: hpath = lf.replace("\\", "/") # for windows users if hpath[-1] != "/": hpath += "/" href = hpath.replace("#", "%23") lf = os.path.split(lf)[0] upperdirs.append((href, hpath)) upperdirs.append(("", "/")) return render_to_response('listdir.html', {'file':f, 'listdirfiles':listdirfiles, 'listdirdirs':listdirdirs, 'upperdirs':upperdirs, 'settings': settings}) # flat output of file when loaded if os.path.isfile(fp): ext = os.path.splitext(fp)[1].lower() mimetype = extmimetypes.get(ext, "text/plain") fin = open(fp) ftext = fin.read() fin.close() return HttpResponse(ftext, mimetype=mimetype) return HttpResponse("unknown file::%s::" % f, mimetype = "text/plain") def SaveImageInDir(name, imgdir, fdata): print ("hihihihi", fdata, settings.SURVEYS) print os.path.join(settings.SURVEYS, imgdir) if not os.path.isdir(os.path.join(settings.SURVEYS, imgdir)): print "*** Must have directory '%s' in '%s'" % (imgdir, settings.SURVEYS) while True: fname = os.path.join(settings.SURVEYS, imgdir, name) if not os.path.exists(fname): break mname = re.match("(.*?)(?:-(\d+))?\.(png|jpg|jpeg)$(?i)", name) if mname: name = "%s-%d.%s" % (mname.group(1), int(mname.group(2) or "0") + 1, mname.group(3)) print "saving file", fname fout = open(fname, "wb") fout.write(fdata.read()) fout.close() res = os.path.join(imgdir, name) return res.replace("\\", "/") def jgtuploadfile(request): filesuploaded = [ ] project, user, tunnelversion = request.POST["project"], request.POST["user"], request.POST["tunnelversion"] print (project, user, tunnelversion) for uploadedfile in request.FILES.values(): if uploadedfile.field_name in ["tileimage", "backgroundimage"] and \ uploadedfile.content_type in ["image/png", "image/jpeg"]: fname = user + "_" + re.sub("[\\\\/]", "-", uploadedfile.name) # very escaped \ print fname fileuploaded = SaveImageInDir(fname, uploadedfile.field_name, uploadedfile) filesuploaded.append(settings.URL_ROOT + "/jgtfile/" + fileuploaded) #print "FF", request.FILES #print ("FFF", request.FILES.values()) message = "" print "gothere" return render_to_response('fileupload.html', {'message':message, 'filesuploaded':filesuploaded, 'settings': settings})