139 lines
4.3 KiB
Python
Executable File
139 lines
4.3 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 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 '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):
|
|
folders = sorted(os.listdir(config['data_folder']))
|
|
table = []
|
|
table.append( ('Path','Share','Size','Unit') )
|
|
for folder in folders:
|
|
full_path = os.path.join(config['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(config['data_folder'], share['path'])
|
|
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 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]
|
|
|
|
share_file = config['shares_file']
|
|
print("creating backup %s"%(share_file+".bkp",))
|
|
copyfile(share_file, share_file+".bkp")
|
|
with open(share_file,'wt') as fp:
|
|
json.dump(shares, fp, indent = 2, sort_keys = True)
|
|
print("Removed %s from %s"%(name, share_file))
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='Flees share manager')
|
|
parser.add_argument('-c','--config', action="store", dest="config", default = "data/config.json",
|
|
help = "Your current config.json file")
|
|
|
|
subparsers = parser.add_subparsers(help='sub-command help', dest='subparser_name')
|
|
|
|
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")
|
|
|
|
parser_folders = subparsers.add_parser('folders', help = "List folders and share names")
|
|
|
|
parser_remove = subparsers.add_parser('remove', help = "Remove a share")
|
|
parser_remove.add_argument(dest="name")
|
|
|
|
|
|
opts = parser.parse_args()
|
|
|
|
if os.path.exists(opts.config):
|
|
config = json.load(open(opts.config,'rt'))
|
|
else:
|
|
print("config file does not exist!")
|
|
sys.exit(1)
|
|
|
|
if os.path.exists(config['shares_file']):
|
|
shares = json.load(open(config['shares_file'],'rt'))
|
|
else:
|
|
print("shares_file does not exist!")
|
|
sys.exit(1)
|
|
|
|
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)
|
|
|
|
|
|
|