FUSE Filesystem that picks only JPEG

onlyimageslink.py

#!/usr/bin/env python
#
# Emanuele Ruffaldi 2017
#
# Original Passthrough without links: https://github.com/skorokithakis/python-fuse-sample
# Alternative to that Passthrough is Loopback in the distribution https://github.com/terencehonles/fusepy/blob/master/examples/loopback.py
#
# TODO: Refresh
from __future__ import with_statement

import os
import sys
import errno
import stat
from fuse import FUSE, FuseOSError, Operations


class PassthroughLink(Operations):
    def __init__(self, root):
        self.root = root

    # Helpers
    # =======

    def _full_path(self, partial):
        if partial.startswith("/"):
            partial = partial[1:]
        path = os.path.join(self.root, partial)
        return path

    # Filesystem methods
    # ==================

    def access(self, path, mode):
        full_path = self._full_path(path)
        if not os.access(full_path, mode):
            raise FuseOSError(errno.EACCES)

    def chmod(self, path, mode):
        return -1

    def chown(self, path, uid, gid):
        return -1

    def getattr(self, path, fh=None):
        full_path = self._full_path(path)
        st = os.lstat(full_path)
        x = dict((key, getattr(st, key)) for key in ('st_atime', 'st_ctime',
                     'st_gid', 'st_mode', 'st_mtime', 'st_nlink', 'st_size', 'st_uid'))
        if not stat.S_ISDIR(x["st_mode"]):
            x["st_mode"] |= stat.S_IFLNK
        return x

    def readdir(self, path, fh):
        full_path = self._full_path(path)

        dirents = ['.', '..']
        if os.path.isdir(full_path):
            dirents.extend([x for x in os.listdir(full_path) if x[0] == "." or x.endswith(".jpg") or x.endswith(".JPG") or os.path.isdir(os.path.join(full_path,x))])
        for r in dirents:
            yield r

    def readlink(self, path):
        # we do not support double symbolic links
        return self._full_path(path)

    def mknod(self, path, mode, dev):
        return -1

    def rmdir(self, path):
        return -1

    def mkdir(self, path, mode):
        return -1

    def statfs(self, path):
        full_path = self._full_path(path)
        stv = os.statvfs(full_path)
        return dict((key, getattr(stv, key)) for key in ('f_bavail', 'f_bfree',
            'f_blocks', 'f_bsize', 'f_favail', 'f_ffree', 'f_files', 'f_flag',
            'f_frsize', 'f_namemax'))

    def unlink(self, path):
        return -1

    def symlink(self, name, target):
        return -1

    def rename(self, old, new):
        return -1

    def link(self, target, name):
        return -1

    def utimens(self, path, times=None):
        return os.utime(self._full_path(path), times)

    # File methods
    # ============

    def open(self, path, flags):
        full_path = self._full_path(path)
        return os.open(full_path, flags)

    def create(self, path, mode, fi=None):
        full_path = self._full_path(path)
        return os.open(full_path, os.O_WRONLY | os.O_CREAT, mode)

    def read(self, path, length, offset, fh):
        os.lseek(fh, offset, os.SEEK_SET)
        return os.read(fh, length)

    def write(self, path, buf, offset, fh):
        os.lseek(fh, offset, os.SEEK_SET)
        return os.write(fh, buf)

    def truncate(self, path, length, fh=None):
        full_path = self._full_path(path)
        with open(full_path, 'r+') as f:
            f.truncate(length)

    def flush(self, path, fh):
        return os.fsync(fh)

    def release(self, path, fh):
        return os.close(fh)

    def fsync(self, path, fdatasync, fh):
        return self.flush(path, fh)



class Passthrough(Operations):
    def __init__(self, root):
        self.root = root

    # Helpers
    # =======

    def _full_path(self, partial):
        if partial.startswith("/"):
            partial = partial[1:]
        path = os.path.join(self.root, partial)
        return path

    # Filesystem methods
    # ==================

    def access(self, path, mode):
        full_path = self._full_path(path)
        if not os.access(full_path, mode):
            raise FuseOSError(errno.EACCES)

    def chmod(self, path, mode):
        full_path = self._full_path(path)
        return os.chmod(full_path, mode)

    def chown(self, path, uid, gid):
        full_path = self._full_path(path)
        return os.chown(full_path, uid, gid)

    def getattr(self, path, fh=None):
        full_path = self._full_path(path)
        st = os.lstat(full_path)
        return dict((key, getattr(st, key)) for key in ('st_atime', 'st_ctime',
                     'st_gid', 'st_mode', 'st_mtime', 'st_nlink', 'st_size', 'st_uid'))

    def readdir(self, path, fh):
        full_path = self._full_path(path)

        dirents = ['.', '..']
        if os.path.isdir(full_path):
            dirents.extend([x for x in os.listdir(full_path) if x[0] == "." or x.endswith(".jpg") or x.endswith(".JPG") or os.path.isdir(os.path.join(full_path,x))])
        for r in dirents:        
            yield r

    def readlink(self, path):
        pathname = os.readlink(self._full_path(path))
        if pathname.startswith("/"):
            # Path name is absolute, sanitize it.
            return os.path.relpath(pathname, self.root)
        else:
            return pathname

    def mknod(self, path, mode, dev):
        return os.mknod(self._full_path(path), mode, dev)

    def rmdir(self, path):
        full_path = self._full_path(path)
        return os.rmdir(full_path)

    def mkdir(self, path, mode):
        return os.mkdir(self._full_path(path), mode)

    def statfs(self, path):
        full_path = self._full_path(path)
        stv = os.statvfs(full_path)
        return dict((key, getattr(stv, key)) for key in ('f_bavail', 'f_bfree',
            'f_blocks', 'f_bsize', 'f_favail', 'f_ffree', 'f_files', 'f_flag',
            'f_frsize', 'f_namemax'))

    def unlink(self, path):
        return os.unlink(self._full_path(path))

    def symlink(self, name, target):
        return os.symlink(name, self._full_path(target))

    def rename(self, old, new):
        return os.rename(self._full_path(old), self._full_path(new))

    def link(self, target, name):
        return os.link(self._full_path(target), self._full_path(name))

    def utimens(self, path, times=None):
        return os.utime(self._full_path(path), times)

    # File methods
    # ============

    def open(self, path, flags):
        full_path = self._full_path(path)
        return os.open(full_path, flags)

    def create(self, path, mode, fi=None):
        full_path = self._full_path(path)
        return os.open(full_path, os.O_WRONLY | os.O_CREAT, mode)

    def read(self, path, length, offset, fh):
        os.lseek(fh, offset, os.SEEK_SET)
        return os.read(fh, length)

    def write(self, path, buf, offset, fh):
        os.lseek(fh, offset, os.SEEK_SET)
        return os.write(fh, buf)

    def truncate(self, path, length, fh=None):
        full_path = self._full_path(path)
        with open(full_path, 'r+') as f:
            f.truncate(length)

    def flush(self, path, fh):
        return os.fsync(fh)

    def release(self, path, fh):
        return os.close(fh)

    def fsync(self, path, fdatasync, fh):
        return self.flush(path, fh)

def mainlink(mountpoint, root,nothreads=True):
    FUSE(PassthroughLink(root), mountpoint, nothreads=nothreads, foreground=True,**{'allow_other': True})

def main(mountpoint, root,nothreads=True):
    FUSE(Passthrough(root), mountpoint, nothreads=nothreads, foreground=True,**{'allow_other': True})

if __name__ == '__main__':
    import argparse

    parser = argparse.ArgumentParser(description='Filtered accessor')
    parser.add_argument('-m','--point',help="mount point",required=True)
    parser.add_argument('-b','--bind',help="originating",required=True)
    parser.add_argument('-l','--link',help="link mode (default)",action="store_true")
    parser.add_argument('-c','--content',help="content mode",action="store_true")
    parser.add_argument('--multithread',default=True,type=bool)

    args = parser.parse_args()
    if args.content:
        main(args.point,args.bind,not args.multithread)
    else:
        mainlink(args.point,args.bind,not args.multithread)