micke
/
passui
1
0
Fork 0
Code Issues 3 Pull Requests Releases 1 Wiki Activity

Add docstrings

Browse Source
main
Micke Nordin 3 years ago
parent cffef46e80
commit 353b19d8b3
1 changed files with 119 additions and 24 deletions
Show Stats Download Patch File Download Diff File

143
src/passui
Unescape View File

@ -14,6 +14,11 @@ class PassUi(wx.Frame):
The wx.Frame for passui
"""
def __init__(self, *args, **kw):
"""__init__.
:param args:
:param kw:
"""
super().__init__(*args, **kw)
# Find the top directory and set that to current dirtectory
self.topdir = os.environ.get('PASSWORD_STORE_DIR')
@ -31,6 +36,7 @@ class PassUi(wx.Frame):
self.add_buttons()
def add_buttons(self):
"""add_buttons."""
self.sizer.Clear(delete_windows=True)
self.add_tools()
if self.curdir != self.topdir:
@ -41,7 +47,7 @@ class PassUi(wx.Frame):
index = 0
for cpath in self.cur_paths:
if cpath != self.curdir:
label = '🗀 ' + os.path.basename(os.path.normpath(cpath))
label = '🗀 ' + os.path.basename(os.path.normpath(cpath))
btn = wx.Button(self.pnl, label=label)
self.sizer.Add(btn, 0, wx.EXPAND) # pylint: disable=no-member
self.Bind(wx.EVT_BUTTON,
@ -64,6 +70,7 @@ class PassUi(wx.Frame):
self.sizer.Layout()
def add_push_pull(self):
"""add_push_pull."""
pushbtn = wx.Button(self.pnl, label="Push to remote")
self.sizer.Add(pushbtn, 0, wx.EXPAND) # pylint: disable=no-member
self.Bind(wx.EVT_BUTTON, lambda event: pass_push(event), pushbtn)
@ -72,6 +79,10 @@ class PassUi(wx.Frame):
self.Bind(wx.EVT_BUTTON, lambda event: pass_pull(event), pullbtn)
def add_tools(self, index=None):
"""add_tools.
:param index:
"""
btn = wx.Button(self.pnl, label="Show tools")
font = btn.GetFont().MakeBold()
btn.SetFont(font)
@ -81,14 +92,27 @@ class PassUi(wx.Frame):
btn)
def back_button_clicked(self, event, index=None):
"""back_button_clicked.
:param event:
:param index:
"""
if index:
self.show_password_dialog(index)
else:
self.add_buttons()
def delete_password(self, event, index):
"""delete_password.
:param event:
:param index:
"""
path = self.get_pass_path_from_index(index, "password")
dlg = wx.MessageDialog(self.pnl, "Delete " + path + "?", "Are you sure?", style=wx.CANCEL|wx.CANCEL_DEFAULT|wx.OK)
dlg = wx.MessageDialog(self.pnl,
"Delete " + path + "?",
"Are you sure?",
style=wx.CANCEL | wx.CANCEL_DEFAULT | wx.OK)
dlg.SetOKCancelLabels("&Yes", "&Don't delete")
reply = dlg.ShowModal()
if reply == wx.ID_CANCEL:
@ -101,6 +125,11 @@ class PassUi(wx.Frame):
self.back_button_clicked(event)
def get_pass_path_from_index(self, index, pathtype="path"):
"""get_pass_path_from_index.
:param index:
:param pathtype:
"""
result = ""
if pathtype == "password":
result = self.cur_passwords[index]
@ -109,6 +138,7 @@ class PassUi(wx.Frame):
return result.replace(self.topdir, '').replace('.gpg', '')
def get_pass_passwords(self):
"""get_pass_passwords."""
passwords = []
for mfile in os.listdir(self.curdir):
if mfile.endswith(".gpg"): # and os.path.isfile(mfile):
@ -117,6 +147,7 @@ class PassUi(wx.Frame):
return passwords
def get_pass_paths(self):
"""get_pass_paths."""
dirs = []
if self.curdir != self.topdir:
dirs.append(self.curdir)
@ -128,6 +159,10 @@ class PassUi(wx.Frame):
return dirs
def make_back_button(self, index=None):
"""make_back_button.
:param index:
"""
if index != None:
label = self.get_pass_path_from_index(index, "password")
else:
@ -138,6 +173,7 @@ class PassUi(wx.Frame):
return btn
def move_up(self):
"""move_up."""
if self.curdir == self.topdir:
return True
if not os.path.isdir(self.curdir):
@ -146,9 +182,19 @@ class PassUi(wx.Frame):
return True
def password_button_clicked(self, event, index):
"""password_button_clicked.
:param event:
:param index:
"""
self.show_password_dialog(index)
def path_button_clicked(self, event, path=None):
"""path_button_clicked.
:param event:
:param path:
"""
if path == None:
path = os.path.abspath(os.path.join(self.curdir, os.pardir))
self.curdir = path
@ -156,18 +202,34 @@ class PassUi(wx.Frame):
self.cur_passwords = self.get_pass_passwords()
self.add_buttons()
def save_to_pass(self, event, path, text, name = None):
def save_to_pass(self, event, path, text, name=None):
"""save_to_pass.
:param event:
:param path:
:param text:
:param name:
"""
fullpath = os.path.dirname(self.topdir + '/' + path.lstrip('/'))
if name != None:
path = name.GetLineText(0)
fullpath = os.path.dirname(self.topdir + '/' + path.lstrip('/'))
filename = fullpath + '.gpg'
if os.path.exists(fullpath) or os.path.exists(filename):
dlg = wx.MessageDialog(self.pnl, "Path: " + path + " allready exists! Please update password path", "Error", style=wx.OK)
dlg = wx.MessageDialog(
self.pnl,
"Path: " + path +
" allready exists! Please update password path",
"Error",
style=wx.OK)
dlg.ShowModal()
reply = wx.ID_CANCEL
else:
dlg = wx.MessageDialog(self.pnl, "Save to " + path + "?", "Are you sure?", style=wx.CANCEL|wx.CANCEL_DEFAULT|wx.OK)
dlg = wx.MessageDialog(self.pnl,
"Save to " + path + "?",
"Are you sure?",
style=wx.CANCEL | wx.CANCEL_DEFAULT
| wx.OK)
dlg.SetOKCancelLabels("&Yes", "&Don't save")
reply = dlg.ShowModal()
if reply == wx.ID_CANCEL:
@ -185,24 +247,26 @@ class PassUi(wx.Frame):
self.back_button_clicked(event)
def show_new_dialog(self, event):
"""show_new_dialog.
:param event:
"""
self.sizer.Clear(delete_windows=True)
passpath = self.curdir.replace(self.topdir, '')
btn = self.make_back_button()
self.sizer.Add(btn, 0, wx.EXPAND) # pylint: disable=no-member
self.Bind(
wx.EVT_BUTTON,
lambda event, path=self.curdir: self.path_button_clicked(event, path),
btn)
self.Bind(wx.EVT_BUTTON,
lambda event, path=self.curdir: self.path_button_clicked(
event, path),
btn)
name_sizer = wx.BoxSizer(orient=wx.VERTICAL) # pylint: disable=no-member
name_sizer.Add(wx.StaticText(self.pnl, -1, "Password path:"), 0, wx.EXPAND) # pylint: disable=no-member
name_sizer.Add(wx.StaticText(self.pnl, -1, "Password path:"), 0,
wx.EXPAND) # pylint: disable=no-member
pw_sizer = wx.BoxSizer(orient=wx.VERTICAL) # pylint: disable=no-member
pw_sizer.Add(wx.StaticText(self.pnl, -1, "Password:"), 0, wx.EXPAND) # pylint: disable=no-member
name = wx.TextCtrl(self.pnl,
value=passpath,
style=wx.TE_DONTWRAP)
text = wx.TextCtrl(self.pnl,
style=wx.TE_MULTILINE | wx.TE_DONTWRAP)
name = wx.TextCtrl(self.pnl, value=passpath, style=wx.TE_DONTWRAP)
text = wx.TextCtrl(self.pnl, style=wx.TE_MULTILINE | wx.TE_DONTWRAP)
sbtn = wx.Button(self.pnl, label="Save password")
name_sizer.Add(name, 0, wx.EXPAND) # pylint: disable=no-member
pw_sizer.Add(text, 0, wx.EXPAND) # pylint: disable=no-member
@ -210,14 +274,18 @@ class PassUi(wx.Frame):
self.sizer.Add(pw_sizer, 0, wx.EXPAND) # pylint: disable=no-member
self.sizer.Add(sbtn, 0, wx.EXPAND) # pylint: disable=no-member
self.Bind(wx.EVT_BUTTON,
lambda event, path=passpath, mtext=text, mname=name: self.save_to_pass(
event, path, mtext, mname),
lambda event, path=passpath, mtext=text, mname=name: self.
save_to_pass(event, path, mtext, mname),
sbtn)
self.pnl.SetupScrolling()
self.sizer.Layout()
name.SetFocus()
def show_password_dialog(self, index):
"""show_password_dialog.
:param index:
"""
self.sizer.Clear(delete_windows=True)
self.add_tools(index)
@ -253,6 +321,11 @@ class PassUi(wx.Frame):
self.sizer.Layout()
def show_password(self, event, index):
"""show_password.
:param event:
:param index:
"""
self.sizer.Clear(delete_windows=True)
self.add_tools(index)
passpath = self.get_pass_path_from_index(index, "password")
@ -284,6 +357,11 @@ class PassUi(wx.Frame):
self.sizer.Layout()
def show_tools(self, event, index=None):
"""show_tools.
:param event:
:param index:
"""
self.sizer.Clear(delete_windows=True)
btn = wx.Button(self.pnl, label="Go back")
font = btn.GetFont().MakeBold()
@ -297,16 +375,19 @@ class PassUi(wx.Frame):
font = nbtn.GetFont().MakeBold()
nbtn.SetFont(font)
self.sizer.Add(nbtn, 0, wx.EXPAND) # pylint: disable=no-member
self.Bind(
wx.EVT_BUTTON,
lambda event: self.show_new_dialog(event),
nbtn)
self.Bind(wx.EVT_BUTTON, lambda event: self.show_new_dialog(event),
nbtn)
self.add_push_pull()
self.pnl.SetupScrolling()
self.sizer.Layout()
def copy_to_clipboard(event, text):
"""copy_to_clipboard.
:param event:
:param text:
"""
password = text.split('\n')[0]
command1 = ['/bin/echo', password]
command2 = ['/usr/bin/wl-copy']
@ -315,21 +396,37 @@ def copy_to_clipboard(event, text):
def get_password_from_path(passpath):
"""get_password_from_path.
:param passpath:
"""
result = run_command(['/usr/bin/pass', 'show', passpath])
password = result[0].decode()
return password
def pass_pull(event):
"""pass_pull.
:param event:
"""
result = run_command(['/usr/bin/pass', 'git', 'pull'])
def pass_push(event):
"""pass_push.
:param event:
"""
result = run_command(['/usr/bin/pass', 'git', 'push'])
def run_command(command1, command2=None):
"""Run a command on system and capture result"""
"""Run a command on system and capture result
:param command1:
:param command2:
"""
process1 = subprocess.Popen(command1,
shell=False,
stdout=subprocess.PIPE,
@ -345,8 +442,6 @@ def run_command(command1, command2=None):
return process1.communicate()
if __name__ == '__main__':
# When this module is run (not imported) then create the app, the
# frame, show it, and start the event loop.

Write Preview
Loading…
Cancel
Save