Fix DictFS to use auto-conversion convention

[invirt/packages/python-routefs.git] / routefs / __init__.py

diff --git a/routefs/__init__.py b/routefs/__init__.py
index e0ff03f..5137256 100644 (file)
--- a/routefs/__init__.py
+++ b/routefs/__init__.py
@@ -86,6 +86,10 @@ class RouteFS(fuse.Fuse):
             return
         controller = match.pop('controller')
         result = getattr(self, controller)(**match)
+        if type(result) is str:
+            result = File(result)
+        if type(result) is list:
+            result = Directory(result)
         return result
     
     def readdir(self, path, offset):
@@ -113,14 +117,14 @@ class RouteFS(fuse.Fuse):
         
         st = RouteStat()
         if type(obj) is Directory:
-            st.st_mode = stat.S_IFDIR | 0755
+            st.st_mode = stat.S_IFDIR | obj.mode
             st.st_nlink = 2
         elif type(obj) is Symlink:
-            st.st_mode = stat.S_IFLNK | 0777
+            st.st_mode = stat.S_IFLNK | obj.mode
             st.st_nlink = 1
             st.st_size = len(obj)
         else:
-            st.st_mode = stat.S_IFREG | 0444
+            st.st_mode = stat.S_IFREG | obj.mode
             st.st_nlink = 1
             st.st_size = len(obj)
         
@@ -149,18 +153,38 @@ class RouteFS(fuse.Fuse):
         else:
             return obj
 
-class Directory(list):
+class TreeEntry(object):
+    default_mode = 0444
+    
+    def __new__(cls, contents, mode=None):
+        return super(TreeEntry, cls).__new__(cls, contents)
+    
+    def __init__(self, contents, mode=None):
+        if mode is None:
+            self.mode = self.default_mode
+        else:
+            self.mode = mode
+        
+        super(TreeEntry, self).__init__(contents)
+
+class Directory(TreeEntry, list):
     """
     A dummy class representing a filesystem entry that should be a
     directory
     """
-    pass
+    default_mode = 0555
 
-class Symlink(str):
+class Symlink(TreeEntry, str):
     """
     A dummy class representing something that should be a symlink
     """
-    pass
+    default_mode = 0777
+
+class File(TreeEntry, str):
+    """
+    A dummy class representing something that should be a file
+    """
+    default_mode = 0444
 
 def main(cls):
     """
@@ -174,4 +198,4 @@ def main(cls):
 
 from dictfs import DictFS
 
-__all__ = ['RouteFS', 'DictFS', 'Symlink', 'Directory', 'main']
+__all__ = ['RouteFS', 'DictFS', 'Symlink', 'Directory', 'File', 'main']