workdirfs/workdirfs.py

238 lines
8 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-12 16:28:31 +01:00
import argparse
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
2020-02-12 01:27:35 +01:00
class WorkdirFS(Operations):
2020-02-12 16:28:31 +01:00
def __init__(self, args):
self.args = args
2016-04-11 00:40:01 +02:00
# Helpers
# =======
def _full_path(self, partial):
2020-02-12 16:28:31 +01:00
today = datetime.now() - timedelta(hours=self.args.timeoffset)
if self.args.yearlydir:
path = os.path.join(self.args.archive,"workdir",today.strftime("%Y"))
if self.args.monthlydir:
2020-02-12 00:19:51 +01:00
path = os.path.join(path, today.strftime("%m"))
else:
2020-02-12 16:28:31 +01:00
path = os.path.join(self.args.archive, "workdir")
2020-02-12 00:19:51 +01:00
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.
2020-02-12 16:28:31 +01:00
return os.path.relpath(pathname, self.args.archive)
2016-04-11 00:40:01 +02:00
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 16:28:31 +01:00
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)):
2020-02-12 17:16:53 +01:00
print("""Directory is empty -> remove it (not now implemented for
testpurpose)""",
2020-02-12 00:19:51 +01:00
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
2020-02-12 16:28:31 +01:00
def main(args):
2020-02-12 01:27:35 +01:00
#FUSE(WorkdirFS(root), mountpoint, nothreads=True, foreground=True)
2020-02-12 16:28:31 +01:00
check_dir(args.archive)
check_dir(args.mountpoint)
cleanup_dirs(args.archive)
2020-02-12 00:19:51 +01:00
# 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'):
2020-02-12 16:28:31 +01:00
print("XDG_ARCHIVE_DIR=\""+args.archive+'"\n')
2020-02-12 00:19:51 +01:00
foundarchive=True
elif line.startswith('XDG_WORK_DIR'):
2020-02-12 16:28:31 +01:00
print("XDG_WORK_DIR=\""+args.mountpoint+'"\n')
2020-02-12 00:19:51 +01:00
foundwork=True
else:
print(line, end='')
if not foundarchive:
2020-02-12 00:29:40 +01:00
with open(os.environ['HOME']+'/.config/user-dirs.dirs', 'a') as fh:
2020-02-12 16:28:31 +01:00
fh.write("XDG_ARCHIVE_DIR=\""+args.archive+'"\n')
2020-02-12 00:19:51 +01:00
if not foundwork:
2020-02-12 00:29:40 +01:00
with open(os.environ['HOME']+'/.config/user-dirs.dirs', 'a') as fh:
2020-02-12 16:28:31 +01:00
fh.write("XDG_WORK_DIR=\""+args.mountpoint+'"\n')
2020-02-11 17:56:42 +01:00
2020-02-12 16:28:31 +01:00
# start FUSE filesystem
FUSE(WorkdirFS(args), args.mountpoint, nothreads=True, foreground=True)
2016-04-11 00:40:01 +02:00
if __name__ == '__main__':
2020-02-11 15:14:26 +01:00
#main(sys.argv[2], sys.argv[1])
2020-02-12 16:28:31 +01:00
parser = argparse.ArgumentParser()
parser.add_argument("-a", "--archive",
default=os.environ['HOME']+'/archive', help="Path to archivedir-base")
parser.add_argument("-m", "--mountpoint",
default=os.environ['HOME']+'/Work', help='Path to Workdir')
parser.add_argument("-t", "--timeoffset", type=int, default=2, help="""If you're working
all day till 3 o'clock in the morning, set it to 4, so next day
archive-dir will be created 4 hours after midnight. You have 1h
tolerance, if you're working one day a little bit longer""")
parser.add_argument("-y", "--yearlydir", action="store_false")
parser.add_argument("-M", "--monthlydir", action="store_false")
args = parser.parse_args()
print(args)
#root = os.environ['HOME']+'/archive'
#mountpoint = os.environ['HOME']+'/Work'
main(args)