mirror of
https://github.com/9001/copyparty.git
synced 2025-10-31 03:53:31 +00:00
0a9a807772
if a hook relocates a file into a folder where that same file exists with the same filename, the filename-collision-avoidance would kick in, generating a new filename and another copy
131 lines
3.9 KiB
Python
131 lines
3.9 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import json
|
|
import os
|
|
import re
|
|
import sys
|
|
|
|
|
|
_ = r"""
|
|
relocate/redirect incoming uploads according to file extension or name
|
|
|
|
example usage as global config:
|
|
--xbu j,c1,bin/hooks/reloc-by-ext.py
|
|
|
|
parameters explained,
|
|
xbu = execute before upload
|
|
j = this hook needs upload information as json (not just the filename)
|
|
c1 = this hook returns json on stdout, so tell copyparty to read that
|
|
|
|
example usage as a volflag (per-volume config):
|
|
-v srv/inc:inc:r:rw,ed:c,xbu=j,c1,bin/hooks/reloc-by-ext.py
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
(share filesystem-path srv/inc as volume /inc,
|
|
readable by everyone, read-write for user 'ed',
|
|
running this plugin on all uploads with the params explained above)
|
|
|
|
example usage as a volflag in a copyparty config file:
|
|
[/inc]
|
|
srv/inc
|
|
accs:
|
|
r: *
|
|
rw: ed
|
|
flags:
|
|
xbu: j,c1,bin/hooks/reloc-by-ext.py
|
|
|
|
note: this could also work as an xau hook (after-upload), but
|
|
because it doesn't need to read the file contents its better
|
|
as xbu (before-upload) since that's safer / less buggy,
|
|
and only xbu works with up2k (dragdrop into browser)
|
|
"""
|
|
|
|
|
|
PICS = "avif bmp gif heic heif jpeg jpg jxl png psd qoi tga tif tiff webp"
|
|
VIDS = "3gp asf avi flv mkv mov mp4 mpeg mpeg2 mpegts mpg mpg2 nut ogm ogv rm ts vob webm wmv"
|
|
MUSIC = "aac aif aiff alac amr ape dfpwm flac m4a mp3 ogg opus ra tak tta wav wma wv"
|
|
|
|
|
|
def main():
|
|
inf = json.loads(sys.argv[1])
|
|
vdir, fn = os.path.split(inf["vp"])
|
|
|
|
try:
|
|
fn, ext = fn.rsplit(".", 1)
|
|
except:
|
|
# no file extension; pretend it's "bin"
|
|
ext = "bin"
|
|
|
|
ext = ext.lower()
|
|
|
|
# this function must end by printing the action to perform;
|
|
# that's handled by the print(json.dumps(... at the bottom
|
|
#
|
|
# the action can contain the following keys:
|
|
# "vp" is the folder URL to move the upload to,
|
|
# "ap" is the filesystem-path to move it to (but "vp" is safer),
|
|
# "fn" overrides the final filename to use
|
|
|
|
##
|
|
## some example actions to take; pick one by
|
|
## selecting it inside the print at the end:
|
|
##
|
|
|
|
# move all uploads to one specific folder
|
|
into_junk = {"vp": "/junk"}
|
|
|
|
# create a subfolder named after the filetype and move it into there
|
|
into_subfolder = {"vp": ext}
|
|
|
|
# move it into a toplevel folder named after the filetype
|
|
into_toplevel = {"vp": "/" + ext}
|
|
|
|
# move it into a filetype-named folder next to the target folder
|
|
into_sibling = {"vp": "../" + ext}
|
|
|
|
# move images into "/just/pics", vids into "/just/vids",
|
|
# music into "/just/tunes", and anything else as-is
|
|
if ext in PICS.split():
|
|
by_category = {"vp": "/just/pics"}
|
|
elif ext in VIDS.split():
|
|
by_category = {"vp": "/just/vids"}
|
|
elif ext in MUSIC.split():
|
|
by_category = {"vp": "/just/tunes"}
|
|
else:
|
|
by_category = {} # no action
|
|
|
|
# now choose the default effect to apply; can be any of these:
|
|
# into_junk into_subfolder into_toplevel into_sibling by_category
|
|
effect = into_sibling
|
|
|
|
##
|
|
## but we can keep going, adding more speicifc rules
|
|
## which can take precedence, replacing the fallback
|
|
## effect we just specified:
|
|
##
|
|
|
|
fn = fn.lower() # lowercase filename to make this easier
|
|
|
|
if "screenshot" in fn:
|
|
effect = {"vp": "/ss"}
|
|
if "mpv_" in fn:
|
|
effect = {"vp": "/anishots"}
|
|
elif "debian" in fn or "biebian" in fn:
|
|
effect = {"vp": "/linux-ISOs"}
|
|
elif re.search(r"ep(isode |\.)?[0-9]", fn):
|
|
effect = {"vp": "/podcasts"}
|
|
|
|
# regex lets you grab a part of the matching
|
|
# text and use that in the upload path:
|
|
m = re.search(r"\b(op|ed)([^a-z]|$)", fn)
|
|
if m:
|
|
# the regex matched; use "anime-op" or "anime-ed"
|
|
effect = {"vp": "/anime-" + m[1]}
|
|
|
|
# aaand DO IT
|
|
print(json.dumps({"reloc": effect}))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|