def f_open(folder):
if folder.lower() == "download":
os.system('xdg-open "%s"' % edit_config.down_addr())
elif folder.lower() == "upload":
for addr in edit_config.up_addr():
os.system('xdg-open "%s"' % addr)
else:
print("%s is an unrecognised argument for open" % folder)
def f_remove(drive, mode, addrs):
if mode == "local":
down_dir = edit_config.down_addr()
# Appending file/folder name to download directory
for addr in addrs:
f_path = os.path.join(down_dir, addr)
if not os.path.exists(f_path):
print("%s doesn't exist in %s" % (addr, down_dir))
else:
# use recursive removal if directory
if os.path.isdir(addr):
shutil.rmtree(f_path)
else:
os.remove(f_path)
print("%s removed from %s" % (addr, down_dir))
# Move file in GDrive to trash or delete permanently
elif mode == "remote":
for addr in addrs:
# check if file_id valid
if is_valid_id(drive, addr):
# file to be removed
r_file = drive.CreateFile({'id': addr})
f_name = r_file['title']
# delete permanently if in trash
if is_trash(drive, r_file['id']):
r_file.Delete()
print("%s deleted permanently" % f_name)
# move to trash
else:
r_file.Trash()
print(
"%s moved to GDrive trash. List files in trash by -lt parameter"
% f_name)
else:
print("%s is not a valid mode" % mode)
return
def f_up(drive, addr, fold_id):
# checks if the specified file/folder exists
if not os.path.exists(addr):
print(
"Specified file/folder doesn't exist, please remove from upload list using -config"
)
return
# pass the address to f_create and on success delete/move file/folder
if f_create(drive, addr, fold_id, str(file_add.get_f_name(addr)), False):
# remove file if Remove_Post_Upload is true, otherwise move to GDrive downloads
remove_post_upload = edit_config.read_config()['Remove_Post_Upload']
if remove_post_upload:
# use recursive removal if directory
if os.path.isdir(addr):
shutil.rmtree(addr)
# normal os removal for file
else:
os.remove(addr)
else:
shutil.move(addr, edit_config.down_addr())
else:
print("Upload unsuccessful, please try again!")
def f_list_local():
for f in os.listdir(edit_config.down_addr()):
print(f)
def main():
# set path for relativistic imports if not launched as package
try:
sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))
import file_add
import auth
import edit_config
import file_ops
import cron_handle
# using relativistic imports directly if launched as package
except ImportError:
from . import file_add
from . import auth
from . import edit_config
from . import file_ops
from . import cron_handle
GoogleAuth.DEFAULT_SETTINGS['client_config_file'] = file_add.client_secrets
gauth = auth.drive_auth(0) # parameter to reset GAccount permissions
drive = GoogleDrive(gauth)
arguments = sys.argv[1:]
# if function called without any arguments print version info
if len(arguments) == 0:
p_info("ver")
arg_index = 0
while True:
if arg_index >= len(arguments):
break
elif arguments[arg_index] == "-v" or arguments[
arg_index] == "-version" or arguments[arg_index] == "version":
p_info("ver")
elif arguments[arg_index] == "-h" or arguments[
arg_index] == "-help" or arguments[arg_index] == "help":
p_info("help")
elif arguments[arg_index] == "-re" or arguments[
arg_index] == "-reset" or arguments[arg_index] == "reset":
auth.reset_account()
elif arguments[arg_index] == "-st" or arguments[
arg_index] == "-start" or arguments[arg_index] == "start":
cron_handle.cron_process(drive, "start")
elif arguments[arg_index] == "-x" or arguments[
arg_index] == "-stop" or arguments[arg_index] == "stop":
cron_handle.cron_process(drive, "stop")
elif arguments[arg_index] == "-y" or arguments[
arg_index] == "-status" or arguments[arg_index] == "status":
cron_handle.cron_process(drive, "status")
elif arguments[arg_index] == "-c" or arguments[
arg_index] == "-config" or arguments[arg_index] == "config":
edit_config.write_config()
elif arguments[arg_index] == "-d" or arguments[
arg_index] == "-download" or arguments[arg_index] == "download":
arg_index += 1
if is_matching(arg_index, len(arguments)):
# download entire drive folder
if arguments[arg_index] == "all":
file_list = drive.ListFile({
'q':
"'root' in parents and trashed=false"
}).GetList()
for argument in file_list:
file_ops.f_down(drive, argument,
edit_config.down_addr())
# download only specified folder
else:
for argument in arguments[arg_index:len(arguments)]:
file_ops.f_down(drive, argument,
edit_config.down_addr())
arg_index = len(arguments) # all arguments used up by download
elif arguments[arg_index] == "-u" or arguments[
arg_index] == "-upload" or arguments[arg_index] == "upload":
arg_index += 1
if is_matching(arg_index, len(arguments)):
# adding to root folder hence None
file_ops.f_create(
drive, arguments[arg_index], None,
str(file_add.get_f_name(arguments[arg_index])), True)
elif arguments[arg_index] == "-s" or arguments[
arg_index] == "-share" or arguments[arg_index] == "share":
arg_index += 1
if is_matching(arg_index, len(arguments)):
file_ops.share_link(drive, arguments[arg_index], True)
elif arguments[arg_index] == "-r" or arguments[
arg_index] == "-remove" or arguments[arg_index] == "remove":
arg_index += 2
# in case of less arguments than required
if is_matching(arg_index, len(arguments)):
file_ops.f_remove(drive, arguments[arg_index - 1],
arguments[arg_index:len(arguments)])
arg_index = len(arguments)
elif arguments[arg_index] == "-o" or arguments[
arg_index] == "-open" or arguments[arg_index] == "open":
arg_index += 1
if is_matching(arg_index, len(arguments)):
file_ops.f_open(arguments[arg_index])
elif arguments[arg_index] == "-ls_files" or arguments[arg_index] == "-laf" or \
arguments[arg_index] == "ls_files":
arg_index += 1
if is_matching(arg_index, len(arguments)):
file_ops.f_list(drive, arguments[arg_index], 1)
elif arguments[arg_index] == "-ls" or arguments[
arg_index] == "-l" or arguments[arg_index] == "ls":
if (arg_index + 1) < len(arguments):
if arguments[arg_index + 1] == "remote":
arg_index += 1
file_ops.f_list(drive, "all", 0)
# list of files in downloads directory
elif arguments[arg_index + 1] == "local":
arg_index += 1
file_ops.f_list_local()
# no argument matching -ls
else:
file_ops.f_list(drive, "all", 0)
# no argument after -ls
else:
file_ops.f_list(drive, "all", 0)
elif arguments[arg_index] == "-ls_trash" or arguments[
arg_index] == "-lt" or arguments[arg_index] == "ls_trash":
file_ops.f_list(drive, "trash", 0)
elif arguments[arg_index] == "-ls_folder" or arguments[arg_index] == "-lf" or \
arguments[arg_index] == "ls_folder":
arg_index += 1 # increase arg_index to read the query argument
if is_matching(arg_index, len(arguments)):
file_ops.f_list(drive, arguments[arg_index], 0)
else:
print(
str(arguments[arg_index]) +
" is an unrecognised argument. Please report if you know this is an error"
".\n\n")
p_info("arg")
arg_index += 1