pyserve.py

#!/usr/bin/env python
import os
from flask import Flask, send_from_directory, request



class dirmanager:

  def __init__(self,programdir):
    self.programdir = programdir
    self.subjectParentDir = programdir
    self.subjects = []
    self.getSubjectList()


  def isSubject(self,prospect):
    prospect = str(prospect)
    if not os.path.isdir(prospect):
      return False
    subdirlist = os.listdir(prospect)
    foundmeshes = 0
    for mfile in subdirlist:
      if 'resliced_mesh' in mfile:
        foundmeshes += 1
    if foundmeshes == 14:
      return True
    return False

  def setSubjectParentDir(self,path):
    path = str(path)
    if not os.path.isdir(path):
      return False
    dirlist = os.listdir(path) 
    for prospect in dirlist:
      if self.isSubject(path +'/'+ prospect):
        self.subjectParentDir = path
        return True
    return False

  def getSubjectList(self):
    del self.subjects[:]
    slist = '['
    for subject in os.listdir(str(self.subjectParentDir)):
      if self.isSubject(self.subjectParentDir +'/'+ subject):
        slist += '"' + subject + '",'
        self.subjects.append(subject)
    slist = slist[:-1] + ']'
    return slist

  def getCurrentSubject(self,index):
    if (len(self.subjects)) < 1:
      self.getSubjectList()
    return self.subjectParentDir +'/'+ self.subjects[int(index)] +'/'


dirmanager = dirmanager(os.getcwd())




app = Flask(__name__)  

@app.route('/')
def home():
  return send_from_directory(dirmanager.programdir,'index.html')

@app.route('/mesh/<subject>/<roi>')
def mesh(subject,roi):
  if 'resliced_mesh' in roi:
    #print(roi,subject,dirmanager.getCurrentSubject(subject))
    return send_from_directory(dirmanager.getCurrentSubject(subject), roi)
  return 'permission denied'

@app.route('/subjects')
def subjects():
  return dirmanager.getSubjectList()

@app.route('/scripts/<path>')
def scripts(path):
  if path in ['main.js','math.js']:
    return send_from_directory(dirmanager.programdir,path)
  return 'permission denied'

@app.route('/change',methods=['POST','GET'])
def change():
  newpath = str(request.args.get('newdir'))
  if dirmanager.setSubjectParentDir(newpath):
    response = 'Directory changed to ' + newpath
  else:
    response = newpath + ' is not a directory with enigma data.'
  print(response)
  return response

if __name__ == '__main__':
    app.run(debug=False)