workdirfs/workdirfs.py

236 lines
7.5 KiB
Python
Raw Normal View History

2020-02-11 15:14:26 +01:00
#!/usr/bin/env -S python3 -u
2016-04-11 00:40:01 +02:00
from __future__ import with_statement
import os
import sys
import errno
2020-02-11 15:14:26 +01:00
from datetime import datetime, timedelta
import time
2020-02-12 00:19:51 +01:00
import fileinput
2020-02-11 15:14:26 +01:00
try:
from fuse import FUSE, FuseOSError, Operations
except:
try:
from fusepy import FUSE, FuseOSError, Operations
except:
print("please install fusepy")
raise errno.ENOENT
2016-04-11 00:40:01 +02:00
class Passthrough(Operations):
def __init__(self, root):
self.root = root
2020-02-11 15:14:26 +01:00
self.timeoffset = 2 #hours
2020-02-12 00:19:51 +01:00
self.yearlydir = True
self.monthlydir = True
2016-04-11 00:40:01 +02:00
# Helpers
# =======
def _full_path(self, partial):
2020-02-12 00:19:51 +01:00
today = datetime.now() - timedelta(hours=self.timeoffset)
if self.yearlydir:
path = os.path.join(self.root,"workdir",today.strftime("%Y"))
if self.monthlydir:
path = os.path.join(path, today.strftime("%m"))
else:
path = os.path.join(self.root, "workdir")
2016-04-11 00:40:01 +02:00
if partial.startswith("/"):
partial = partial[1:]
2020-02-12 00:19:51 +01:00
path = os.path.join(check_dir(os.path.join(path, today.strftime("%Y-%m-%d"))), partial)
2020-02-11 15:14:26 +01:00
2016-04-11 00:40:01 +02:00
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)
2020-02-11 15:14:26 +01:00
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'))
2016-04-11 00:40:01 +02:00
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)
2020-02-11 15:14:26 +01:00
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'))
2016-04-11 00:40:01 +02:00
def unlink(self, path):
return os.unlink(self._full_path(path))
def symlink(self, name, target):
return os.symlink(target, self._full_path(name))
2016-04-11 00:40:01 +02:00
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))
2016-04-11 00:40:01 +02:00
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)
2020-02-11 15:14:26 +01:00
def cleanup_dirs(root):
2020-02-12 00:19:51 +01:00
today = datetime.now() - timedelta(hours=2)
today = today.strftime("%Y-%m-%d")
for root, dirs, files in os.walk(root, topdown=False):
for _dir in dirs:
print("cleanup",os.path.join(root, _dir))
if not _dir == today and not os.listdir(os.path.join(root, _dir)):
print("Directory is empty -> remove it",
os.path.join(root, _dir))
2020-02-11 15:14:26 +01:00
2020-02-12 00:19:51 +01:00
def check_dir(path):
2020-02-11 15:14:26 +01:00
checkdir = os.path.isdir(path)
if not checkdir:
try:
os.makedirs(path, exist_ok=True)
2020-02-12 00:19:51 +01:00
print("Created directory {}".format(path), flush=True)
2020-02-11 15:14:26 +01:00
except:
print("[-] Makedir error")
return path
def main(root, mountpoint):
#FUSE(Passthrough(root), mountpoint, nothreads=True, foreground=True)
2020-02-12 00:19:51 +01:00
check_dir(root)
2020-02-11 15:14:26 +01:00
check_dir(mountpoint)
2020-02-12 00:19:51 +01:00
cleanup_dirs(root)
# first search if configuration exists for xdg-userdirs
# to use it with alias gowork and goarchive
foundarchive=False
foundwork=False
with fileinput.input(os.environ['HOME']+'/.config/user-dirs.dirs',
2020-02-11 17:56:42 +01:00
inplace=True) as fh:
for line in fh:
2020-02-12 00:19:51 +01:00
if line.startswith('XDG_ARCHIVE_DIR'):
print("XDG_ARCHIVE_DIR=\""+root+'"')
foundarchive=True
elif line.startswith('XDG_WORK_DIR'):
print("XDG_WORK_DIR=\""+mountpoint+'"')
foundwork=True
else:
print(line, end='')
if not foundarchive:
with open(os.environ['HOME']+'/.config/user-dirs.dirs', 'a') as fh:
fh.write("XDG_ARCHIVE_DIR=\""+root+'"')
if not foundwork:
with open(os.environ['HOME']+'/.config/user-dirs.dirs', 'a') as fh:
fh.write("XDG_WORK_DIR=\""+mountpoint+'"')
2020-02-11 17:56:42 +01:00
2016-04-11 00:40:01 +02:00
FUSE(Passthrough(root), mountpoint, nothreads=True, foreground=True)
if __name__ == '__main__':
2020-02-11 15:14:26 +01:00
#main(sys.argv[2], sys.argv[1])
root = os.environ['HOME']+'/archive'
2020-02-11 17:56:42 +01:00
mountpoint = os.environ['HOME']+'/Work'
2020-02-11 15:14:26 +01:00
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)