add workdir.service

bin/workdir.py

@@ -0,0 +1,194 @@
+#!/usr/bin/env python -u
+
+from __future__ import with_statement
+
+import os
+import sys
+import errno
+
+from datetime import datetime, timedelta
+import time
+
+from fuse import FUSE, FuseOSError, Operations
+
+
+class Passthrough(Operations):
+    def __init__(self, root):
+        self.root = root
+        self.timeoffset = 2
+
+    # Helpers
+    # =======
+
+    def _full_path(self, partial):
+        if partial.startswith("/"):
+            partial = partial[1:]
+        today = datetime.now() - timedelta(hours=self.timeoffset)
+        path = os.path.join(check_dir(os.path.join(self.root,
+                                                   "workdir",
+                                                   today.strftime("%Y-%m-%d"))),
+                            partial)
+ #       path = os.path.join(check_dir(os.path.join(self.root, "workdir",
+ #                                                  str(time.strftime("%Y-%m-%d-%M", time.gmtime())))), 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(os.listdir(full_path))
+        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(target, self._full_path(name))
+
+    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(name), self._full_path(target))
+
+    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 cleanup_dirs(root):
+    for d in os.listdir(root):
+        if not os.listdir(os.path.join(root, d)):
+            print("Directory is empty -> remove it",
+                  os.listdir(os.path.join(root, d)))
+
+
+def check_dir(path, cleanup=False):
+    checkdir = os.path.isdir(path)
+    if not checkdir:
+        try:
+            os.makedirs(path, exist_ok=True)
+            print("Created directory %s", (path), flush=True)
+            if cleanup:
+                cleanup_dirs(path)
+        except:
+            print("[-] Makedir error")
+    return path
+
+def main(root, mountpoint):
+    #FUSE(Passthrough(root), mountpoint, nothreads=True, foreground=True)
+    check_dir(mountpoint)
+    FUSE(Passthrough(root), mountpoint, nothreads=True, foreground=True)
+
+if __name__ == '__main__':
+    #main(sys.argv[2], sys.argv[1])
+    root = os.environ['HOME']+'/archive'
+    mountpoint = os.environ['HOME']+'/WorkNew'
+    print(len(sys.argv))
+    print(sys.argv)
+    if len(sys.argv) == 1:
+        print("use default locations")
+    elif len(sys.argv) == 2:
+        mountpoint = sys.argv[1]
+    elif len(sys.argv) == 3:
+        mountpoint = sys.argv[2]
+        root = sys.argv[1]
+    else:
+        print("not correct count of arguments")
+        raise
+    print(root, mountpoint)
+    main(root, mountpoint)

functions.sh

@@ -29,6 +29,7 @@ create_symlinks() {
     #ln -sf "${MYSHELLCONFIG_BASE}/bashrc_add" ~/bashrc_add
     ln -sf "${MYSHELLCONFIG_BASE}/tmux" ~/.tmux
     ln -sf "${MYSHELLCONFIG_BASE}/tmux/tmux.conf" ~/.tmux.conf
+    ln -sf "${MYSHELLCONFIG_BASE}/systemd/workdir.service" ~/.config/systemd/user/default.target.wants/workdir.service
 
     # Configure to use githooks in .githooks, not in standardlocation .git/hooks
     $SGIT config core.hooksPath .githooks
@@ -36,6 +37,8 @@ create_symlinks() {
     # don't know, why i do it here. TODO: Check it
     find .git/hooks -type l -exec rm {} \; && find .githooks -type f -exec ln -sf ../../{} .git/hooks/ \;
 
+    systemctl --user daemon-reload
+    systemctl --user restart workdir.service
     cd ~-
 
 }

systemd/workdir.service

@@ -0,0 +1,5 @@
+[Unit]
+Description=Daily working directory ~/Work
+
+[Service]
+ExecStart=/usr/bin/python %h/.local/myshellconfig/bin/workdir.py