FileManagerResponse.py

#!/usr/bin/python3
import os
import re
import datetime
from PIL import Image

class FileManagerResponse(object):
    # absolute path to base folder
    root = os.path.join(os.path.dirname(os.path.abspath(__file__)),'files')
    def __init__(self,path):
        ''' Init '''
        self.path          = path # absolute path to file or folder
        self.relative_path = re.sub('^'+self.root, '', self.path) # path from file manager base folder
        self.type          = 'folder' if os.path.isdir(path) else 'file'
        self.statinfo      = os.stat(self.path)
        self.attributes    = None
        self.data          = None
        self.response      = None
        self.content       = None
#===============================================================================
    def set_data(self):
        ''' Build data dict for response '''
        data               = {}
        # set id (path from file manager base folder)
        data['id']         = self.relative_path
        if self.type == 'folder':
            # trailing slash is mandatory on folders
            data['id']     = self.relative_path.rstrip('/')+'/'
        data['type']       = self.type
        self.set_attributes()
        data['attributes'] = self.attributes
        self.data = data
#===============================================================================
    def set_attributes(self):
        ''' Build attributes dict for response '''
        attributes                  = {}
        attributes['name']          = self.relative_path.strip('/').split('/').pop()
        attributes['readable']      = 1 if os.access(self.path, os.R_OK) else 0
        attributes['writable']      = 1 if os.access(self.path, os.W_OK) else 0
        if self.type == 'file':
            attributes['extension'] = os.path.splitext(self.path)[1].lstrip('.')
            height                  = 0
            width                   = 0
            if attributes['extension'] in ['gif','jpg','jpeg','png']:
                im = Image.open(self.path)
                height,width = im.size
            attributes['height']    = height
            attributes['width']     = width
            attributes['size']      = os.path.getsize(self.path)
        else:
            if not os.access(self.path, os.X_OK):
                attributes['readable'] = 0
                attributes['writable'] = 0
        attributes['path']          = self.path
        attributes['created']     = int(self.statinfo.st_ctime)
        attributes['modified']     = int(self.statinfo.st_mtime)
        attributes['timestamp']     = int(self.statinfo.st_mtime)
        if self.content:
            attributes['content']   = self.content
        self.attributes = attributes
#===============================================================================
    def set_response(self):
        response         = {}
        self.set_data()
        response['data'] = self.data
        self.response    = response
#===============================================================================
    def set_content(self,content):
        ''' Set the content in the case where we must pass it to the client '''
        self.content = content