349 lines
12 KiB
Python
Executable File
349 lines
12 KiB
Python
Executable File
#!/usr/bin/env python
|
|
import argparse,json,sys,os
|
|
from shutil import copyfile
|
|
from tabulate import tabulate
|
|
import hashlib
|
|
|
|
def get_folder_size(path):
|
|
|
|
total_size = 0
|
|
for dirpath, dirnames, filenames in os.walk(path):
|
|
for f in filenames:
|
|
fp = os.path.join(dirpath, f)
|
|
total_size += os.path.getsize(fp)
|
|
return total_size
|
|
|
|
def get_or_no(key,d,no):
|
|
if key in d:
|
|
return d[key]
|
|
return no
|
|
|
|
def get_direct_token(share, filename):
|
|
if not 'pass_hash' in share:
|
|
return None
|
|
return hashlib.sha1(
|
|
share['pass_hash'].encode('utf-8') + filename.encode('utf-8')
|
|
).hexdigest()
|
|
|
|
|
|
def get_root_path(opts):
|
|
root_folder = os.path.dirname(
|
|
os.path.dirname(
|
|
os.path.abspath(
|
|
opts.config
|
|
)
|
|
)
|
|
)
|
|
return root_folder
|
|
|
|
def file_size_human(num):
|
|
for x in ['B','KB','MB','GB','TB']:
|
|
if num < 1024.0:
|
|
if x=='B':
|
|
return "%d %s" % (num, x)
|
|
return "%3.1f %s" % (num, x)
|
|
num /= 1024.0
|
|
|
|
def list_shares(shares,opts):
|
|
table = []
|
|
table.append(('Name', 'Path','Public','Password','Upload','Overwrite','Direct','Expire'))
|
|
for share in shares:
|
|
public = get_or_no('public',share, False)
|
|
password = 'pass_hash' in share or 'pass_plain' in share
|
|
if opts.show_password:
|
|
if not password:
|
|
password = ""
|
|
if 'pass_plain' in share:
|
|
password = hashlib.sha1(share['pass_plain'].encode('utf-8')).hexdigest()
|
|
if 'pass_hash' in share:
|
|
password = share['pass_hash']
|
|
upload = get_or_no('upload',share, False)
|
|
overwrite = get_or_no('overwrite',share, True)
|
|
direct = get_or_no('direct_links',share, False) if password else False
|
|
expire = get_or_no('expire',share, "-")
|
|
table.append((
|
|
share['name'],
|
|
share['path']+"/",
|
|
public,
|
|
password,
|
|
upload,
|
|
overwrite,
|
|
direct,
|
|
expire
|
|
))
|
|
print(tabulate(table, headers = "firstrow"))
|
|
|
|
|
|
def list_folders(shares,config):
|
|
if 'data_folder' not in config:
|
|
print("data_folder not defined in config")
|
|
sys.exit(1)
|
|
data_folder = os.path.join(config['__root_path__'], config['data_folder'])
|
|
folders = sorted(os.listdir(data_folder))
|
|
table = []
|
|
table.append( ('Path','Share','Size','Unit') )
|
|
for folder in folders:
|
|
full_path = os.path.join(data_folder, folder)
|
|
if not os.path.isdir(full_path):
|
|
continue
|
|
share_name = "[unused by any share]"
|
|
for share in shares:
|
|
share_path = os.path.join(data_folder, share['path'])
|
|
if not os.path.exists(share_path):
|
|
break
|
|
if os.path.samefile(full_path, share_path):
|
|
share_name = share['name']
|
|
break
|
|
(size_num, size_unit) = file_size_human(get_folder_size(
|
|
full_path
|
|
)).split(" ")
|
|
table.append((
|
|
folder,
|
|
share_name,
|
|
size_num,
|
|
size_unit
|
|
))
|
|
print(tabulate(table, headers = "firstrow"))
|
|
|
|
def add_share(shares, config, opts):
|
|
|
|
share = {
|
|
'name': opts.name,
|
|
'path': opts.path,
|
|
'public': opts.public,
|
|
'upload': opts.upload,
|
|
'overwrite': opts.overwrite,
|
|
'direct_links': opts.direct,
|
|
}
|
|
if opts.plain:
|
|
share.update({
|
|
'pass_plain': opts.plain
|
|
})
|
|
if opts.hashed:
|
|
share.update({
|
|
'pass_hash': hashlib.sha1(opts.hashed).hexdigest()
|
|
})
|
|
if opts.expire:
|
|
try:
|
|
date_object = datetime.strptime(opts.expire,"%Y-%m-%d %H:%M")
|
|
except ValueError as e:
|
|
print(e)
|
|
sys.exit(1)
|
|
share.update({
|
|
'expire': opts.expire
|
|
})
|
|
|
|
if opts.insert:
|
|
shares.append(share)
|
|
shares_file = os.path.join(config['__root_path__'], opts.shares_file)
|
|
if os.path.exists(shares_file):
|
|
print("creating backup %s"%(shares_file+".bkp",))
|
|
copyfile(
|
|
shares_file,
|
|
shares_file+".bkp"
|
|
)
|
|
with open(shares_file,'wt') as fp:
|
|
json.dump(shares, fp, indent = 2, sort_keys = True)
|
|
print("Wrote file %s"%(shares_file,))
|
|
print("Add share: %s"%( opts.name, ))
|
|
else:
|
|
print("Share not saved anywhere. Save with -i")
|
|
print(json.dumps(share, indent = 2, sort_keys = True))
|
|
|
|
|
|
def remove_share(shares,config,opts):
|
|
name = opts.name
|
|
share = [share for share in shares if share['name'] == name]
|
|
for share_ in share:
|
|
print("Removing share: %s"%( name, ))
|
|
print(json.dumps(share_, indent = 2, sort_keys = True))
|
|
|
|
shares = [share for share in shares if share['name'] != name]
|
|
|
|
shares_file = os.path.join(config['__root_path__'], opts.shares_file)
|
|
print("creating backup %s"%(shares_file+".bkp",))
|
|
copyfile(shares_file, shares_file+".bkp")
|
|
with open(shares_file,'wt') as fp:
|
|
json.dump(shares, fp, indent = 2, sort_keys = True)
|
|
print("Removed %s from %s"%(name, shares_file))
|
|
|
|
|
|
def print_rest_api(shares, config, opts):
|
|
if 'public_url' not in config:
|
|
print("Set public_url variable in your config.json")
|
|
sys.exit(1)
|
|
|
|
shares = [share for share in shares if share['name'] == opts.name]
|
|
if len(shares) == 0:
|
|
print("No such share %s"%( opts.name, ))
|
|
sys.exit(1)
|
|
share = shares[0]
|
|
|
|
if opts.type == "list":
|
|
print("Link to list contents of the share:")
|
|
print("%s/list/%s"%(
|
|
config['public_url'],
|
|
share['name']
|
|
))
|
|
return
|
|
if not 'pass_hash' in share:
|
|
print("REST API enabled only if pass_hash is set for share")
|
|
sys.exit(1)
|
|
if opts.type == "login":
|
|
print("Link to automatically login in the share:")
|
|
print("%s/list/%s/%s"%(
|
|
config['public_url'],
|
|
share['name'],
|
|
share['pass_hash']
|
|
))
|
|
elif opts.type == "upload":
|
|
if 'upload' not in share or not share['upload']:
|
|
print("Uploading not allowed to this share")
|
|
sys.exit(0)
|
|
|
|
print("Link to upload file to the share:")
|
|
print("curl -F file=@'the_file_name.ext' %s/upload/%s/%s"%(
|
|
config['public_url'],
|
|
share['name'],
|
|
share['pass_hash']
|
|
))
|
|
elif opts.type == "download":
|
|
print("Links to download files:")
|
|
share_path = os.path.join(
|
|
config['__root_path__'],
|
|
config['data_folder'],
|
|
share['path']
|
|
)
|
|
if not os.path.exists(share_path):
|
|
print("no files")
|
|
sys.exit(0)
|
|
for filename in sorted(os.listdir(share_path)):
|
|
print("%s/download/%s/%s/%s"%(
|
|
config['public_url'],
|
|
share['name'],
|
|
share['pass_hash'],
|
|
filename
|
|
))
|
|
elif opts.type == "direct":
|
|
if 'direct_links' not in share or not share['direct_links']:
|
|
print("Direct downloading not allowed in this share")
|
|
sys.exit(0)
|
|
print("Links to direct download files:")
|
|
share_path = os.path.join(
|
|
config['__root_path__'],
|
|
config['data_folder'],
|
|
share['path']
|
|
)
|
|
if not os.path.exists(share_path):
|
|
print("no files")
|
|
sys.exit(0)
|
|
for filename in sorted(os.listdir(share_path)):
|
|
print("%s/direct/%s/%s/%s"%(
|
|
config['public_url'],
|
|
share['name'],
|
|
get_direct_token(share,filename),
|
|
filename
|
|
))
|
|
elif opts.type == "zip":
|
|
print("ZIP download:")
|
|
print("%s/zip/%s/%s"%(
|
|
config['public_url'],
|
|
share['name'],
|
|
share['pass_hash']
|
|
))
|
|
|
|
|
|
|
|
def parse_options():
|
|
config_default = os.path.realpath(
|
|
os.path.join(
|
|
os.path.dirname(
|
|
os.path.realpath(__file__)
|
|
),
|
|
"..",
|
|
"data",
|
|
"config.json"
|
|
)
|
|
)
|
|
|
|
parser = argparse.ArgumentParser(description='Flees share manager')
|
|
parser.add_argument('-c','--config', action="store", dest="config", default = config_default,
|
|
help = "Your current config.json file [%(default)s]")
|
|
parser.add_argument('-s','--shares', action="store", dest="shares_file", default = None,
|
|
help = "shares.json you want to use. Defaults to what config.json defines")
|
|
|
|
subparsers = parser.add_subparsers(help='sub-command help', dest='subparser_name')
|
|
## list shares
|
|
parser_list = subparsers.add_parser('list', help = "List shares")
|
|
parser_list.add_argument('-P', action="store_true", dest="show_password", default = False,
|
|
help = "Display hashed passwords")
|
|
## list folders
|
|
parser_folders = subparsers.add_parser('folders', help = "List folders and share names")
|
|
## Remove
|
|
parser_remove = subparsers.add_parser('remove', help = "Remove a share")
|
|
parser_remove.add_argument(dest="name")
|
|
## Add
|
|
parser_add = subparsers.add_parser('add', help = "Add a share")
|
|
parser_add.add_argument('-n','--name', action="store", dest="name", required = True)
|
|
parser_add.add_argument('-p','--path', action="store", dest="path", required = True,
|
|
help= "path relative to data folder"
|
|
)
|
|
parser_add.add_argument('-P','--public', action="store_true", dest="public", default = False)
|
|
parser_add.add_argument('-u','--upload', action="store_true", dest="upload", default = False)
|
|
parser_add.add_argument('-o','--overwrite', action="store_false", dest="overwrite", default = True,
|
|
help = "Disable file overwrites")
|
|
parser_add.add_argument('-d','--direct', action="store_true", dest="direct", default = False,
|
|
help = "Allow direct file sharing (password hash included in URL)")
|
|
parser_add.add_argument('--pass-plain', action="store", dest="plain", default = False)
|
|
parser_add.add_argument('--pass-hash', action="store", dest="hashed", default = False)
|
|
parser_add.add_argument('-e','--expire', action="store", dest="expire", default = False,
|
|
help = "expire date in format '%Y-%m-%d %H:%M' ex. '2018-12-24 21:00'"
|
|
)
|
|
parser_add.add_argument('-i','--insert', action="store_true", dest="insert", default = False,
|
|
help = "Insert new share directly in the shares.json file"
|
|
)
|
|
## REST
|
|
parser_rest = subparsers.add_parser('rest', help = "Display REST API links")
|
|
parser_rest.add_argument(dest="name", help = "Name of the share")
|
|
parser_rest.add_argument(dest="type", help = "Type of command",
|
|
choices = ['list','login','upload','download','direct','zip']
|
|
)
|
|
|
|
return parser.parse_args()
|
|
|
|
if __name__ == "__main__":
|
|
opts = parse_options()
|
|
config = {}
|
|
if os.path.exists(opts.config):
|
|
config = json.load(open(opts.config,'rt'))
|
|
config['__root_path__'] = get_root_path(opts)
|
|
else:
|
|
print("config file %s does not exist!"%(opts.config,))
|
|
sys.exit(1)
|
|
|
|
if opts.shares_file:
|
|
config['shares_file'] = opts.shares_file
|
|
if 'shares_file' in config:
|
|
# if not from command line, read from config
|
|
opts.shares_file = config['shares_file']
|
|
|
|
if os.path.exists(os.path.join(config['__root_path__'],config['shares_file'])):
|
|
shares = json.load(open(os.path.join(config['__root_path__'],config['shares_file']),'rt'))
|
|
else:
|
|
print("shares_file %s does not exist!"%(os.path.join(config['__root_path__'],config['shares_file'])))
|
|
shares = []
|
|
|
|
if opts.subparser_name == 'list':
|
|
list_shares(shares,opts)
|
|
elif opts.subparser_name == 'folders':
|
|
list_folders(shares,config)
|
|
elif opts.subparser_name == 'remove':
|
|
remove_share(shares,config,opts)
|
|
elif opts.subparser_name == 'add':
|
|
add_share(shares,config,opts)
|
|
elif opts.subparser_name == 'rest':
|
|
print_rest_api(shares,config,opts)
|
|
|
|
|
|
|