mirror of
https://github.com/rwinkhart/sshyp.git
synced 2026-09-05 08:37:14 -04:00
Replace lists with tuples where applicable
This commit is contained in:
+23
-23
@@ -128,7 +128,7 @@ def pass_gen(): # prompts the user for necessary information to generate a pass
|
|||||||
else:
|
else:
|
||||||
break
|
break
|
||||||
_complexity = str(input('password complexity - simple (for compatibility) or complex (for security)? (s/C) '))
|
_complexity = str(input('password complexity - simple (for compatibility) or complex (for security)? (s/C) '))
|
||||||
if _complexity not in ['s', 'S']:
|
if _complexity not in ('s', 'S'):
|
||||||
_complexity = 'c'
|
_complexity = 'c'
|
||||||
_gen = string_gen(_complexity.lower(), _length)
|
_gen = string_gen(_complexity.lower(), _length)
|
||||||
return _gen
|
return _gen
|
||||||
@@ -236,7 +236,7 @@ def tweak(): # runs configuration wizard
|
|||||||
# config directory creation
|
# config directory creation
|
||||||
Path(expanduser('~/.config/sshyp/devices')).mkdir(mode=0o700, parents=True, exist_ok=True)
|
Path(expanduser('~/.config/sshyp/devices')).mkdir(mode=0o700, parents=True, exist_ok=True)
|
||||||
if not Path(f"{expanduser('~/.config/sshyp/tmp')}").exists():
|
if not Path(f"{expanduser('~/.config/sshyp/tmp')}").exists():
|
||||||
if uname()[0] in ['Haiku', 'FreeBSD']:
|
if uname()[0] in ('Haiku', 'FreeBSD'):
|
||||||
symlink('/tmp', expanduser('~/.config/sshyp/tmp'))
|
symlink('/tmp', expanduser('~/.config/sshyp/tmp'))
|
||||||
elif Path("/data/data/com.termux").exists():
|
elif Path("/data/data/com.termux").exists():
|
||||||
symlink('/data/data/com.termux/files/usr/tmp', expanduser('~/.config/sshyp/tmp'))
|
symlink('/data/data/com.termux/files/usr/tmp', expanduser('~/.config/sshyp/tmp'))
|
||||||
@@ -285,13 +285,13 @@ def tweak(): # runs configuration wizard
|
|||||||
_ssh_gen = (input(f"{_divider}make sure the ssh service on the remote server is running and properly "
|
_ssh_gen = (input(f"{_divider}make sure the ssh service on the remote server is running and properly "
|
||||||
f"configured\n\nsync support requires a unique ssh key - would you like to have this "
|
f"configured\n\nsync support requires a unique ssh key - would you like to have this "
|
||||||
f"automatically generated? (Y/n/o(ffline)) "))
|
f"automatically generated? (Y/n/o(ffline)) "))
|
||||||
if _ssh_gen.lower() not in ['n', 'o', 'offline']:
|
if _ssh_gen.lower() not in ('n', 'o', 'offline'):
|
||||||
Path(f"{expanduser('~')}/.ssh").mkdir(mode=0o700, exist_ok=True)
|
Path(f"{expanduser('~')}/.ssh").mkdir(mode=0o700, exist_ok=True)
|
||||||
run(['ssh-keygen', '-t', 'ed25519', '-f', expanduser('~/.ssh/sshyp')])
|
run(['ssh-keygen', '-t', 'ed25519', '-f', expanduser('~/.ssh/sshyp')])
|
||||||
elif _ssh_gen.lower() == 'n':
|
elif _ssh_gen.lower() == 'n':
|
||||||
print(f"\n\u001b[4;1mensure that the key file you are using is located at "
|
print(f"\n\u001b[4;1mensure that the key file you are using is located at "
|
||||||
f"{expanduser('~/.ssh/sshyp')}\u001b[0m")
|
f"{expanduser('~/.ssh/sshyp')}\u001b[0m")
|
||||||
elif _ssh_gen.lower() in ['o', 'offline']:
|
elif _ssh_gen.lower() in ('o', 'offline'):
|
||||||
_offline_mode = True
|
_offline_mode = True
|
||||||
print('\nsshyp has been set to offline mode - to enable syncing, run "sshyp tweak" again')
|
print('\nsshyp has been set to offline mode - to enable syncing, run "sshyp tweak" again')
|
||||||
|
|
||||||
@@ -347,7 +347,7 @@ def tweak(): # runs configuration wizard
|
|||||||
|
|
||||||
|
|
||||||
def print_info(): # prints help text based on argument
|
def print_info(): # prints help text based on argument
|
||||||
if arguments[0] in ['version', '-v']:
|
if arguments[0] in ('version', '-v'):
|
||||||
print('\nsshyp is a simple, self-hosted, sftp-synchronized password manager\nfor unix(-like) systems (haiku/'
|
print('\nsshyp is a simple, self-hosted, sftp-synchronized password manager\nfor unix(-like) systems (haiku/'
|
||||||
'freebsd/linux/termux)\n\nsshyp is a viable alternative to (and compatible with) pass/password-store\n')
|
'freebsd/linux/termux)\n\nsshyp is a viable alternative to (and compatible with) pass/password-store\n')
|
||||||
print(" .. \u001b[38;5;9m♥♥ ♥♥\u001b[0m ..\n .''.''/()\\ \u001b[38;5;13m"
|
print(" .. \u001b[38;5;9m♥♥ ♥♥\u001b[0m ..\n .''.''/()\\ \u001b[38;5;13m"
|
||||||
@@ -602,7 +602,7 @@ def add_entry(): # adds a new entry
|
|||||||
if Path(f"{directory}{_entry_name}.gpg").is_file():
|
if Path(f"{directory}{_entry_name}.gpg").is_file():
|
||||||
print(f"\n\u001b[38;5;9merror: entry (/{_entry_name}) already exists\u001b[0m\n")
|
print(f"\n\u001b[38;5;9merror: entry (/{_entry_name}) already exists\u001b[0m\n")
|
||||||
s_exit(3)
|
s_exit(3)
|
||||||
if arguments[arg_start_p] in ['note', '-n']: # note entry
|
if arguments[arg_start_p] in ('note', '-n'): # note entry
|
||||||
_shm_folder, _shm_entry = shm_gen()
|
_shm_folder, _shm_entry = shm_gen()
|
||||||
run([editor, f"{tmp_dir}{_shm_folder}/{_shm_entry}-n"])
|
run([editor, f"{tmp_dir}{_shm_folder}/{_shm_entry}-n"])
|
||||||
_notes = open(f"{tmp_dir}{_shm_folder}/{_shm_entry}-n", 'r').read()
|
_notes = open(f"{tmp_dir}{_shm_folder}/{_shm_entry}-n", 'r').read()
|
||||||
@@ -679,13 +679,13 @@ def edit(): # edits the contents of an entry
|
|||||||
s_exit(2)
|
s_exit(2)
|
||||||
_shm_folder, _shm_entry = shm_gen()
|
_shm_folder, _shm_entry = shm_gen()
|
||||||
determine_decrypt(directory + _entry_name, _shm_folder, _shm_entry)
|
determine_decrypt(directory + _entry_name, _shm_folder, _shm_entry)
|
||||||
if arguments[arg_start_p] in ['username', '-u']:
|
if arguments[arg_start_p] in ('username', '-u'):
|
||||||
_detail, _edit_line = str(input('username: ')), 1
|
_detail, _edit_line = str(input('username: ')), 1
|
||||||
elif arguments[arg_start_p] in ['password', '-p']:
|
elif arguments[arg_start_p] in ('password', '-p'):
|
||||||
_detail, _edit_line = str(input('password: ')), 0
|
_detail, _edit_line = str(input('password: ')), 0
|
||||||
elif arguments[arg_start_p] in ['url', '-l']:
|
elif arguments[arg_start_p] in ('url', '-l'):
|
||||||
_detail, _edit_line = str(input('url: ')), 2
|
_detail, _edit_line = str(input('url: ')), 2
|
||||||
if arguments[arg_start_p] in ['note', '-n']:
|
if arguments[arg_start_p] in ('note', '-n'):
|
||||||
_edit_line = 2
|
_edit_line = 2
|
||||||
_new_lines = optimized_edit(edit_note(_shm_folder, _shm_entry,
|
_new_lines = optimized_edit(edit_note(_shm_folder, _shm_entry,
|
||||||
open(f"{tmp_dir}{_shm_folder}/{_shm_entry}", 'r').readlines()), None, -1)
|
open(f"{tmp_dir}{_shm_folder}/{_shm_entry}", 'r').readlines()), None, -1)
|
||||||
@@ -705,7 +705,7 @@ def gen(): # generates a password for a new or an existing entry
|
|||||||
else:
|
else:
|
||||||
_entry_name = entry_name_fetch()
|
_entry_name = entry_name_fetch()
|
||||||
_shm_folder, _shm_entry = shm_gen()
|
_shm_folder, _shm_entry = shm_gen()
|
||||||
if arg_count > 1 and arguments[arg_start_p] in ['update', '-u']: # gen update
|
if arg_count > 1 and arguments[arg_start_p] in ('update', '-u'): # gen update
|
||||||
if not Path(f"{directory}{_entry_name}.gpg").is_file():
|
if not Path(f"{directory}{_entry_name}.gpg").is_file():
|
||||||
print(f"\n\u001b[38;5;9merror: entry (/{_entry_name}) does not exist\u001b[0m\n")
|
print(f"\n\u001b[38;5;9merror: entry (/{_entry_name}) does not exist\u001b[0m\n")
|
||||||
s_exit(2)
|
s_exit(2)
|
||||||
@@ -745,13 +745,13 @@ def copy_data(): # copies a specified field of an entry to the clipboard
|
|||||||
_shm_folder, _shm_entry = shm_gen()
|
_shm_folder, _shm_entry = shm_gen()
|
||||||
determine_decrypt(directory + _entry_name, _shm_folder, _shm_entry)
|
determine_decrypt(directory + _entry_name, _shm_folder, _shm_entry)
|
||||||
_copy_line, _index = open(f"{tmp_dir}{_shm_folder}/{_shm_entry}", 'r').readlines(), 0
|
_copy_line, _index = open(f"{tmp_dir}{_shm_folder}/{_shm_entry}", 'r').readlines(), 0
|
||||||
if arguments[arg_start_p] in ['username', '-u']:
|
if arguments[arg_start_p] in ('username', '-u'):
|
||||||
_index = 1
|
_index = 1
|
||||||
elif arguments[arg_start_p] in ['password', '-p']:
|
elif arguments[arg_start_p] in ('password', '-p'):
|
||||||
_index = 0
|
_index = 0
|
||||||
elif arguments[arg_start_p] in ['url', '-l']:
|
elif arguments[arg_start_p] in ('url', '-l'):
|
||||||
_index = 2
|
_index = 2
|
||||||
elif arguments[arg_start_p] in ['note', '-n']:
|
elif arguments[arg_start_p] in ('note', '-n'):
|
||||||
_index = 3
|
_index = 3
|
||||||
if 'WAYLAND_DISPLAY' in environ: # Wayland clipboard detection
|
if 'WAYLAND_DISPLAY' in environ: # Wayland clipboard detection
|
||||||
run(['wl-copy', _copy_line[_index].rstrip()])
|
run(['wl-copy', _copy_line[_index].rstrip()])
|
||||||
@@ -836,7 +836,7 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
elif (arg_count == 2 and arg_start == 0) or (arg_count == 3 and arg_start == 1):
|
elif (arg_count == 2 and arg_start == 0) or (arg_count == 3 and arg_start == 1):
|
||||||
if arguments[arg_start] == 'copy':
|
if arguments[arg_start] == 'copy':
|
||||||
if arguments[arg_start_p] in ['username', '-u', 'password', '-p', 'url', '-l', 'note', '-n']:
|
if arguments[arg_start_p] in ('username', '-u', 'password', '-p', 'url', '-l', 'note', '-n'):
|
||||||
try:
|
try:
|
||||||
success_flag = 1
|
success_flag = 1
|
||||||
copy_data()
|
copy_data()
|
||||||
@@ -844,20 +844,20 @@ if __name__ == "__main__":
|
|||||||
print(f"\n\u001b[38;5;9merror: field does not exist in entry\u001b[0m\n")
|
print(f"\n\u001b[38;5;9merror: field does not exist in entry\u001b[0m\n")
|
||||||
s_exit(2)
|
s_exit(2)
|
||||||
elif arguments[arg_start] == 'add':
|
elif arguments[arg_start] == 'add':
|
||||||
if arguments[arg_start_p] in ['note', '-n', 'password', '-p']:
|
if arguments[arg_start_p] in ('note', '-n', 'password', '-p'):
|
||||||
success_flag, sync_flag = True, True
|
success_flag, sync_flag = True, True
|
||||||
add_entry()
|
add_entry()
|
||||||
elif arguments[arg_start_p] in ['folder', '-f']:
|
elif arguments[arg_start_p] in ('folder', '-f'):
|
||||||
success_flag = 1
|
success_flag = 1
|
||||||
add_folder()
|
add_folder()
|
||||||
elif arguments[arg_start] == 'edit':
|
elif arguments[arg_start] == 'edit':
|
||||||
if arguments[arg_start_p] in ['rename', 'relocate', '-r']:
|
if arguments[arg_start_p] in ('rename', 'relocate', '-r'):
|
||||||
success_flag, sync_flag, silent_sync = True, True, True
|
success_flag, sync_flag, silent_sync = True, True, True
|
||||||
rename()
|
rename()
|
||||||
elif arguments[arg_start_p] in ['username', '-u', 'password', '-p', 'url', '-l', 'note', '-n']:
|
elif arguments[arg_start_p] in ('username', '-u', 'password', '-p', 'url', '-l', 'note', '-n'):
|
||||||
success_flag, sync_flag = True, True
|
success_flag, sync_flag = True, True
|
||||||
edit()
|
edit()
|
||||||
elif arguments[arg_start] == 'gen' and arguments[arg_start_p] in ['update', '-u']:
|
elif arguments[arg_start] == 'gen' and arguments[arg_start_p] in ('update', '-u'):
|
||||||
success_flag, sync_flag = True, True
|
success_flag, sync_flag = True, True
|
||||||
gen()
|
gen()
|
||||||
|
|
||||||
@@ -877,10 +877,10 @@ if __name__ == "__main__":
|
|||||||
success_flag = True
|
success_flag = True
|
||||||
whitelist_setup()
|
whitelist_setup()
|
||||||
elif arg_count > 1 and arguments[arg_start] == 'whitelist':
|
elif arg_count > 1 and arguments[arg_start] == 'whitelist':
|
||||||
if arguments[arg_start_p] in ['list', '-l']:
|
if arguments[arg_start_p] in ('list', '-l'):
|
||||||
success_flag = True
|
success_flag = True
|
||||||
whitelist_list()
|
whitelist_list()
|
||||||
elif arguments[arg_start_p] in ['add', 'del']:
|
elif arguments[arg_start_p] in ('add', 'del'):
|
||||||
success_flag = True
|
success_flag = True
|
||||||
whitelist_manage()
|
whitelist_manage()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user