12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058 |
- #============================================================================
- # This file is part of Pwman3.
- #
- # Pwman3 is free software; you can redistribute it and/or modify
- # it under the terms of the GNU General Public License, version 2
- # as published by the Free Software Foundation;
- #
- # Pwman3 is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- # GNU General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License
- # along with Pwman3; if not, write to the Free Software
- # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
- #============================================================================
- # Copyright (C) 2013 Oz Nahum <nahumoz@gmail.com>
- #============================================================================
- # pylint: disable=I0011
- """
- Define the base CLI interface for pwman3
- """
- from __future__ import print_function
- class HelpUI(object):
- """
- this class holds all the UI help functionality.
- in PwmanCliNew. The later inherits from this class
- and allows it to print help messages to the console.
- """
- def usage(self, string):
- print ("Usage: %s" % (string))
- def help_open(self):
- self.usage("open <ID>")
- print ("Launch default browser with 'xdg-open url',\n",
- "the url must contain http:// or https://.")
- def help_o(self):
- self.help_open()
- def help_copy(self):
- self.usage("copy <ID>")
- print ("Copy password to X clipboard (xsel required)")
- def help_cp(self):
- self.help_copy()
- def help_cls(self):
- self.usage("cls")
- print ("Clear the Screen from information.")
- def help_list(self):
- self.usage("list <tag> ...")
- print ("List nodes that match current or specified filter.",
- " l is an alias.")
- def help_EOF(self):
- self.help_exit()
- def help_delete(self):
- self.usage("delete <ID|tag> ...")
- print ("Deletes nodes. rm is an alias.")
- self._mult_id_help()
- def help_h(self):
- self.help_help()
- def help_help(self):
- self.usage("help [topic]")
- print ("Prints a help message for a command.")
- def help_e(self):
- self.help_edit()
- def help_n(self):
- self.help_new()
- def help_p(self):
- self.help_print()
- def help_l(self):
- self.help_list()
- def help_edit(self):
- self.usage("edit <ID|tag> ... ")
- print ("Edits a nodes.")
- self._mult_id_help()
- def help_import(self):
- self.usage("import [filename] ...")
- print ("Imports a nodes from a file.")
- def help_export(self):
- self.usage("export <ID|tag> ... ")
- print ("Exports a list of ids to an external format. If no IDs or",
- " tags are specified, then all nodes under the current",
- " filter are exported.")
- self._mult_id_help()
- def help_new(self):
- self.usage("new")
- print ("Creates a new node.,",
- "You can override default config settings the following way:\n",
- "pwman> n {'leetify':False, 'numerics':True}")
- def help_rm(self):
- self.help_delete()
- def help_print(self):
- self.usage("print <ID|tag> ...")
- print ("Displays a node. ")
- self._mult_id_help()
- def _mult_id_help(self):
- print("Multiple ids and nodes can be specified, separated by a space.",
- " A range of ids can be specified in the format n-N. e.g. ",
- " '10-20' would specify all nodes having ids from 10 to 20 ",
- " inclusive. Tags are considered one-by-one. e.g. 'foo 2 bar'",
- " would yield to all nodes with tag 'foo', node 2 and all ",
- " nodes with tag 'bar'.")
- def help_exit(self):
- self.usage("exit")
- print("Exits the application.")
- def help_save(self):
- self.usage("save [filename]")
- print("Saves the current configuration to [filename]. If no filename ",
- "is given, the configuration is saved to the file from which ",
- "the initial configuration was loaded.")
- def help_set(self):
- self.usage("set [configoption] [value]")
- print("Sets a configuration option. If no value is specified, the ",
- "current value for [configoption] is output. If neither ",
- "[configoption] nor [value] are specified, the whole current ",
- "configuration is output. [configoption] must be of the ",
- "format <section>.<option>")
- def help_passwd(self):
- self.usage("passwd")
- print("Changes the password on the database. ")
- def help_forget(self):
- self.usage("forget")
- print("Forgets the database password. Your password will need to ",
- "be reentered before accessing the database again.")
- def help_clear(self):
- self.usage("clear")
- print("Clears the filter criteria. ")
- def help_filter(self):
- self.usage("filter <tag> ...")
- print("Filters nodes on tag. Arguments can be zero or more tags. ",
- "Displays current tags if called without arguments.")
- def help_tags(self):
- self.usage("tags")
- print("Displays all tags in used in the database.")
- class BaseUI(object):
- """
- this class holds all the UI functionality
- in PwmanCliNew. The later inherits from this class
- and allows it to print messages to the console.
- """
- def _do_filter(self, args):
- pass
- def _tags(self, enc):
- """
- read tags from TAGS table in DB,
- this method has a working unittest
- """
- tags = self._db.listtags()
- if tags:
- _tags = [''] * len(tags)
- for t in tags:
- try:
- _tags.append(enc.decrypt(t))
- except (ValueError, Exception), e:
- _tags.append(t)
- del(e)
- _tags = filter(None, _tags)
- return _tags
- import pwman.exchange.importer as importer
- import pwman.util.generator as generator
- from pwman.data.nodes import Node
- from pwman.data.tags import Tag
- from pwman.util.crypto import CryptoEngine
- from pwman.util.crypto import zerome
- import pwman.util.config as config
- import re
- import sys
- import os
- import cmd
- import time
- import select as uselect
- import ast
- from pwman.ui import tools
- from pwman.ui.tools import CliMenu
- from pwman.ui.tools import CliMenuItem
- from colorama import Fore
- from pwman.ui.base import HelpUI, BaseUI
- from pwman.ui.tools import CLICallback
- from pwman.data.nodes import NewNode
- from pwman.ui.tools import CMDLoop
- import getpass
- from pwman.data.tags import TagNew
- try:
- import readline
- _readline_available = True
- except ImportError, e: # pragma: no cover
- _readline_available = False
- def get_pass_conf():
- numerics = config.get_value("Generator", "numerics").lower() == 'true'
- # TODO: allow custom leetifying through the config
- leetify = config.get_value("Generator", "leetify").lower() == 'true'
- special_chars = config.get_value("Generator", "special_chars"
- ).lower() == 'true'
- return numerics, leetify, special_chars
- # pylint: disable=R0904
- class PwmanCliOld(cmd.Cmd, HelpUI, BaseUI):
- """
- UI class for MacOSX
- """
- def error(self, exception):
- if (isinstance(exception, KeyboardInterrupt)):
- print('')
- else:
- print("Error: {0} ".format(exception))
- def do_exit(self, args):
- """exit the ui"""
- self._db.close()
- return True
- def get_ids(self, args):
- """
- Command can get a single ID or
- a range of IDs, with begin-end.
- e.g. 1-3 , will get 1 to 3.
- """
- ids = []
- rex = re.compile("^(?P<begin>\d+)(?:-(?P<end>\d+))?$")
- rex = rex.match(args)
- if hasattr(rex, 'groupdict'):
- try:
- begin = int(rex.groupdict()['begin'])
- end = int(rex.groupdict()['end'])
- if not end > begin:
- print("Start node should be smaller than end node")
- return ids
- ids += range(begin, end+1)
- return ids
- except TypeError:
- ids.append(int(begin))
- else:
- print("Could not understand your input...")
- return ids
- def get_filesystem_path(self, default="", reader=raw_input):
- return tools.getinput("Enter filename: ", default, reader=reader)
- def get_username(self, default="", reader=raw_input):
- return tools.getinput("Username: ", default, reader)
- def get_password(self, argsgiven, numerics=False, leetify=False,
- symbols=False, special_signs=False, reader=raw_input):
- """
- in the config file:
- numerics -> numerics
- leetify -> symbols
- special_chars -> special_signs
- """
- # TODO: replace this code with tools.getpassword
- if argsgiven == 1:
- length = tools.getinput("Password length (default 7): ", "7")
- length = len(length)
- password, dumpme = generator.generate_password(length, length,
- True, leetify,
- numerics,
- special_signs)
- print ("New password: %s" % (password))
- return password
- # no args given
- password = tools.getpassword("Password (Blank to generate): ",
- tools._defaultwidth, False, reader)
- if not password:
- length = tools.getinput("Password length (default 7): ", "7")
- if length:
- length = int(length)
- else:
- length = 7
- password, dumpme = generator.generate_password(length, length,
- True, leetify,
- numerics,
- special_signs)
- print ("New password: %s" % (password))
- return password
- def get_url(self, default="", reader=raw_input):
- return tools.getinput("Url: ", default, reader)
- def get_notes(self, default="", reader=raw_input):
- return tools.getinput("Notes: ", default, reader)
- def get_tags(self, default=None):
- """read node tags from user"""
- defaultstr = ''
- if default:
- for t in default:
- defaultstr += "%s " % (t.get_name())
- else:
- tags = self._db.currenttags()
- for t in tags:
- defaultstr += "%s " % (t.get_name())
- strings = []
- tags = self._db.listtags(True)
- for t in tags:
- strings.append(t.get_name())
- def complete(text, state):
- count = 0
- for s in strings:
- if s.startswith(text):
- if count == state:
- return s
- else:
- count += 1
- taglist = tools.getinput("Tags: ", defaultstr, complete)
- tagstrings = taglist.split()
- tags = []
- for tn in tagstrings:
- tags.append(TagNew(tn))
- return tags
- def print_node(self, node):
- width = str(tools._defaultwidth)
- print ("Node %d." % (node._id))
- print ("%" + width + "s %s") % (tools.typeset("Username:", Fore.RED),
- node.get_username())
- print ("%" + width + "s %s") % (tools.typeset("Password:", Fore.RED),
- node.get_password())
- print ("%" + width + "s %s") % (tools.typeset("Url:", Fore.RED),
- node.get_url())
- print ("%" + width + "s %s") % (tools.typeset("Notes:", Fore.RED),
- node.get_notes())
- print (tools.typeset("Tags: ", Fore.RED)),
- for t in node.get_tags():
- print (" %s \n" % t.get_name()),
- def heardEnter():
- inpt, out, err = uselect.select([sys.stdin], [], [], 0.0001)
- for stream in inpt:
- if stream == sys.stdin:
- sys.stdin.readline()
- return True
- return False
- def waituntil_enter(somepredicate, timeout, period=0.25):
- mustend = time.time() + timeout
- while time.time() < mustend:
- cond = somepredicate()
- if cond:
- break
- time.sleep(period)
- self.do_cls('')
- flushtimeout = int(config.get_value("Global", "cls_timeout"))
- if flushtimeout > 0:
- print ("Type Enter to flush screen (autoflash in "
- "%d sec.)" % flushtimeout)
- waituntil_enter(heardEnter, flushtimeout)
- def do_tags(self, arg):
- tags = self._db.listtags()
- if len(tags) > 0:
- tags[0].get_name() # hack to get password request before output
- print ("Tags: "),
- if len(tags) == 0:
- print ("None"),
- for t in tags:
- print ("%s " % (t.get_name())),
- print
- def complete_filter(self, text, line, begidx, endidx):
- strings = []
- enc = CryptoEngine.get()
- if not enc.alive():
- return strings
- tags = self._db.listtags()
- for t in tags:
- name = t.get_name()
- if name.startswith(text):
- strings.append(t.get_name())
- return strings
- def do_filter(self, args):
- tagstrings = args.split()
- try:
- tags = []
- for ts in tagstrings:
- tags.append(Tag(ts))
- self._db.filter(tags)
- tags = self._db.currenttags()
- print ("Current tags: "),
- if len(tags) == 0:
- print ("None"),
- for t in tags:
- print ("%s " % (t.get_name())),
- print
- except Exception, e:
- self.error(e)
- def do_clear(self, args):
- try:
- self._db.clearfilter()
- except Exception, e:
- self.error(e)
- def do_edit(self, arg):
- ids = self.get_ids(arg)
- for i in ids:
- try:
- i = int(i)
- node = self._db.getnodes([i])[0]
- menu = CliMenu()
- print ("Editing node %d." % (i))
- menu.add(CliMenuItem("Username", self.get_username,
- node.get_username,
- node.set_username))
- menu.add(CliMenuItem("Password", self.get_password,
- node.get_password,
- node.set_password))
- menu.add(CliMenuItem("Url", self.get_url,
- node.get_url,
- node.set_url))
- menu.add(CliMenuItem("Notes", self.get_notes,
- node.get_notes,
- node.set_notes))
- menu.add(CliMenuItem("Tags", self.get_tags,
- node.get_tags,
- node.set_tags))
- menu.run()
- self._db.editnode(i, node)
- # when done with node erase it
- zerome(node._password)
- except Exception, e:
- self.error(e)
- def do_import(self, arg):
- try:
- args = arg.split()
- if len(args) == 0:
- types = importer.Importer.types()
- intype = tools.select("Select filetype:", types)
- imp = importer.Importer.get(intype)
- infile = tools.getinput("Select file:")
- imp.import_data(self._db, infile)
- else:
- for i in args:
- types = importer.Importer.types()
- intype = tools.select("Select filetype:", types)
- imp = importer.Importer.get(intype)
- imp.import_data(self._db, i)
- except Exception, e:
- self.error(e)
- def do_export(self, arg):
- print('Not implemented...')
- def do_new(self, args):
- """
- can override default config settings the following way:
- Pwman3 0.2.1 (c) visit: http://github.com/pwman3/pwman3
- pwman> n {'leetify':False, 'numerics':True, 'special_chars':True}
- Password (Blank to generate):
- """
- errmsg = ("could not parse config override, please input some"
- " kind of dictionary, e.g.: n {'leetify':False, "
- " numerics':True, 'special_chars':True}")
- try:
- username = self.get_username()
- if args:
- try:
- args = ast.literal_eval(args)
- except Exception:
- raise Exception(errmsg)
- if not isinstance(args, dict):
- raise Exception(errmsg)
- password = self.get_password(1, **args)
- else:
- numerics = config.get_value("Generator",
- "numerics").lower() == 'true'
- # TODO: allow custom leetifying through the config
- leetify = config.get_value("Generator",
- "leetify").lower() == 'true'
- special_chars = config.get_value("Generator",
- "special_chars").lower() == \
- 'true'
- password = self.get_password(0,
- numerics=numerics,
- symbols=leetify,
- special_signs=special_chars)
- url = self.get_url()
- notes = self.get_notes()
- node = Node(username, password, url, notes)
- tags = self.get_tags()
- node.set_tags(tags)
- self._db.addnodes([node])
- print ("Password ID: %d" % (node.get_id()))
- except Exception, e:
- self.error(e)
- def do_print(self, arg):
- for i in self.get_ids(arg):
- try:
- node = self._db.getnodes([i])
- self.print_node(node[0])
- except Exception, e: # pragma: no cover
- self.error(e)
- def do_delete(self, arg):
- ids = self.get_ids(arg)
- try:
- nodes = self._db.getnodes(ids)
- for n in nodes:
- b = tools.getyesno("Are you sure you want to delete '%s@%s'?"
- % (n.get_username(), n.get_url()), False)
- if b is True:
- self._db.removenodes([n])
- print ("%s@%s deleted" % (n.get_username(), n.get_url()))
- except Exception, e:
- self.error(e)
- def do_list(self, args):
- """
- TODO: in order to make this code testable
- The functionality in this method should
- go to a method that returns a string.
- This method should only do the printing.
- """
- if len(args.split()) > 0:
- self.do_clear('')
- self.do_filter(args)
- try:
- if sys.platform != 'win32':
- rows, cols = tools.gettermsize()
- else:
- rows, cols = 18, 80
- nodeids = self._db.listnodes()
- nodes = self._db.getnodes(nodeids)
- cols -= 8
- i = 0
- for n in nodes:
- tags = n.get_tags()
- tagstring = ''
- first = True
- for t in tags:
- if not first:
- tagstring += ", "
- else:
- first = False
- tagstring += t.get_name()
- name = "%s@%s" % (n.get_username(), n.get_url())
- name_len = cols * 2 / 3
- tagstring_len = cols / 3
- if len(name) > name_len:
- name = name[:name_len - 3] + "..."
- if len(tagstring) > tagstring_len:
- tagstring = tagstring[:tagstring_len - 3] + "..."
- fmt = "%%5d. %%-%ds %%-%ds" % (name_len, tagstring_len)
- print (tools.typeset(fmt % (n.get_id(), name, tagstring),
- Fore.YELLOW, False))
- i += 1
- if i > rows - 2:
- i = 0
- c = tools.getonechar("Press <Space> for more, "
- "or 'Q' to cancel")
- if c == 'q':
- break
- except Exception, e:
- self.error(e)
- def do_forget(self, args):
- try:
- enc = CryptoEngine.get()
- enc.forget()
- except Exception, e:
- self.error(e)
- def do_passwd(self, args):
- try:
- key = self._db.changepassword()
- self._db._save(key)
- except Exception, e:
- self.error(e)
- def do_set(self, args):
- argstrs = args.split()
- try:
- if len(argstrs) == 0:
- conf = config.get_conf()
- for s in conf.keys():
- for n in conf[s].keys():
- print ("%s.%s = %s" % (s, n, conf[s][n]))
- elif len(argstrs) == 1:
- r = re.compile("(.+)\.(.+)")
- m = r.match(argstrs[0])
- if m is None or len(m.groups()) != 2:
- print ("Invalid option format")
- self.help_set()
- return
- print ("%s.%s = %s" % (m.group(1), m.group(2),
- config.get_value(m.group(1),
- m.group(2))))
- elif len(argstrs) == 2:
- r = re.compile("(.+)\.(.+)")
- m = r.match(argstrs[0])
- if m is None or len(m.groups()) != 2:
- print ("Invalid option format")
- self.help_set()
- return
- config.set_value(m.group(1), m.group(2), argstrs[1])
- else:
- self.help_set()
- except Exception, e:
- self.error(e)
- def do_save(self, args):
- argstrs = args.split()
- try:
- if len(argstrs) > 0:
- config.save(argstrs[0])
- else:
- config.save()
- print ("Config saved.")
- except Exception, e:
- self.error(e)
- def do_cls(self, args):
- os.system('clear')
- def do_copy(self, args):
- if self.hasxsel:
- ids = self.get_ids(args)
- if len(ids) > 1:
- print ("Can copy only 1 password at a time...")
- return None
- try:
- node = self._db.getnodes(ids)
- tools.text_to_clipboards(node[0].get_password())
- print ("copied password for {}@{} clipboard".format(
- node[0].get_username(), node[0].get_url()))
- print ("erasing in 10 sec...")
- time.sleep(10)
- tools.text_to_clipboards("")
- except Exception, e:
- self.error(e)
- else:
- print ("Can't copy to clipboard, no xsel found in the system!")
- def do_open(self, args):
- ids = self.get_ids(args)
- if not args:
- self.help_open()
- return
- if len(ids) > 1:
- print ("Can open only 1 link at a time ...")
- return None
- try:
- node = self._db.getnodes(ids)
- url = node[0].get_url()
- tools.open_url(url)
- except Exception, e:
- self.error(e)
- def postloop(self):
- try:
- readline.write_history_file(self._historyfile)
- except Exception:
- pass
- def __init__(self, db, hasxsel):
- """
- initialize CLI interface, set up the DB
- connecion, see if we have xsel ...
- """
- _dbwarning = "\n*** WARNNING: You are using the old database format" \
- + " which is unsecure." \
- + " This database format is in hold. No bugs are fixed. Please " \
- + " upgrade your database." \
- + " Check the help (pwman3 -h) or look at the manpage which" \
- + " explains how to proceed. ***"
- print (_dbwarning)
- sys.exit(1)
- class BaseCommands(PwmanCliOld):
- """
- Inherit from the old class, override
- all the methods related to tags, and
- newer Node format, so backward compatability is kept...
- Commands defined here, can have aliases definded in Aliases.
- You can define the aliases here too, but it makes
- the class code really long and unclear.
- """
- def do_copy(self, args):
- if self.hasxsel:
- ids = self.get_ids(args)
- if len(ids) > 1:
- print ("Can copy only 1 password at a time...")
- return None
- try:
- node = self._db.getnodes(ids)
- tools.text_to_clipboards(node[0].password)
- print ("copied password for {}@{} clipboard".format(
- node[0].username, node[0].url))
- print ("erasing in 10 sec...")
- time.sleep(10)
- tools.text_to_clipboards("")
- except Exception, e:
- self.error(e)
- else:
- print ("Can't copy to clipboard, no xsel found in the system!")
- def do_open(self, args):
- ids = self.get_ids(args)
- if not args:
- self.help_open()
- return
- if len(ids) > 1:
- print ("Can open only 1 link at a time ...")
- return None
- try:
- node = self._db.getnodes(ids)
- url = node[0].url
- tools.open_url(url)
- except Exception, e:
- self.error(e)
- def do_edit(self, arg, menu=None):
- ids = self.get_ids(arg)
- for i in ids:
- try:
- i = int(i)
- node = self._db.getnodes([i])[0]
- if not menu:
- menu = CMDLoop()
- print ("Editing node %d." % (i))
- menu.add(CliMenuItem("Username", self.get_username,
- node.username,
- node.username))
- menu.add(CliMenuItem("Password", self.get_password,
- node.password,
- node.password))
- menu.add(CliMenuItem("Url", self.get_url,
- node.url,
- node.url))
- menunotes = CliMenuItem("Notes", self.get_notes,
- node.notes,
- node.notes)
- menu.add(menunotes)
- menu.add(CliMenuItem("Tags", self.get_tags,
- node.tags,
- node.tags))
- menu.run(node)
- self._db.editnode(i, node)
- # when done with node erase it
- zerome(node._password)
- except Exception, e:
- self.error(e)
- def print_node(self, node):
- width = str(tools._defaultwidth)
- print ("Node %d." % (node._id))
- print (("%" + width + "s %s") % (tools.typeset("Username:", Fore.RED),
- node.username))
- print (("%" + width + "s %s") % (tools.typeset("Password:", Fore.RED),
- node.password))
- print (("%" + width + "s %s") % (tools.typeset("Url:", Fore.RED),
- node.url))
- print (("%" + width + "s %s") % (tools.typeset("Notes:", Fore.RED),
- node.notes))
- print (tools.typeset("Tags: ", Fore.RED)),
- for t in node.tags:
- print (" %s " % t)
- print()
- def heardEnter():
- i, o, e = uselect.select([sys.stdin], [], [], 0.0001)
- for s in i:
- if s == sys.stdin:
- sys.stdin.readline()
- return True
- return False
- def waituntil_enter(somepredicate, timeout, period=0.25):
- mustend = time.time() + timeout
- while time.time() < mustend:
- cond = somepredicate()
- if cond:
- break
- time.sleep(period)
- self.do_cls('')
- try:
- flushtimeout = int(config.get_value("Global", "cls_timeout"))
- except ValueError:
- flushtimeout = 10
- if flushtimeout > 0:
- print ("Type Enter to flush screen (autoflash in "
- "%d sec.)" % flushtimeout)
- waituntil_enter(heardEnter, flushtimeout)
- def do_tags(self, arg):
- enc = CryptoEngine.get()
- if not enc.alive():
- enc._getcipher()
- print ("Tags: \n",)
- t = self._tags(enc)
- print ('\n'.join(t))
- def get_tags(self, default=None, reader=raw_input):
- """read tags from user"""
- defaultstr = ''
- if default:
- for t in default:
- defaultstr += "%s " % (t)
- else:
- # tags = self._db.currenttags()
- tags = self._db._filtertags
- for t in tags:
- defaultstr += "%s " % (t)
- # strings = []
- tags = self._db.listtags(True)
- # for t in tags:
- # strings.append(t.get_name())
- # strings.append(t)
- strings = [t for t in tags]
- def complete(text, state):
- count = 0
- for s in strings:
- if s.startswith(text):
- if count == state:
- return s
- else:
- count += 1
- taglist = tools.getinput("Tags: ", defaultstr, completer=complete,
- reader=reader)
- tagstrings = taglist.split()
- tags = [TagNew(tn) for tn in tagstrings]
- return tags
- def do_list(self, args):
- if len(args.split()) > 0:
- self.do_clear('')
- self.do_filter(args)
- try:
- if sys.platform != 'win32':
- rows, cols = tools.gettermsize()
- else:
- rows, cols = 18, 80 # fix this !
- nodeids = self._db.listnodes()
- nodes = self._db.getnodes(nodeids)
- cols -= 8
- i = 0
- for n in nodes:
- tags = n.tags
- tags = filter(None, tags)
- tagstring = ''
- first = True
- for t in tags:
- if not first:
- tagstring += ", "
- else:
- first = False
- tagstring += t
- name = "%s@%s" % (n.username, n.url)
- name_len = cols * 2 / 3
- tagstring_len = cols / 3
- if len(name) > name_len:
- name = name[:name_len - 3] + "..."
- if len(tagstring) > tagstring_len:
- tagstring = tagstring[:tagstring_len - 3] + "..."
- fmt = "%%5d. %%-%ds %%-%ds" % (name_len, tagstring_len)
- formatted_entry = tools.typeset(fmt % (n._id,
- name, tagstring),
- Fore.YELLOW, False)
- print (formatted_entry)
- i += 1
- if i > rows - 2:
- i = 0
- c = tools.getonechar("Press <Space> for more,"
- " or 'Q' to cancel")
- if c.lower() == 'q':
- break
- except Exception, e:
- self.error(e)
- def do_filter(self, args):
- tagstrings = args.split()
- try:
- tags = [TagNew(ts) for ts in tagstrings]
- self._db.filter(tags)
- tags = self._db.currenttags()
- print ("Current tags: ",)
- if len(tags) == 0:
- print ("None",)
- for t in tags:
- print ("%s " % t)
- print
- except Exception, e:
- self.error(e)
- def do_new(self, args):
- """
- can override default config settings the following way:
- Pwman3 0.2.1 (c) visit: http://github.com/pwman3/pwman3
- pwman> n {'leetify':False, 'numerics':True, 'special_chars':True}
- Password (Blank to generate):
- """
- errmsg = ("could not parse config override, please input some"
- " kind of dictionary, e.g.: n {'leetify':False, "
- " numerics':True, 'special_chars':True}")
- try:
- username = self.get_username()
- if args:
- try:
- args = ast.literal_eval(args)
- except Exception:
- raise Exception(errmsg)
- if not isinstance(args, dict):
- raise Exception(errmsg)
- password = self.get_password(argsgiven=1, **args)
- else:
- numerics, leet, s_chars = get_pass_conf()
- password = self.get_password(argsgiven=0,
- numerics=numerics,
- symbols=leet,
- special_signs=s_chars)
- url = self.get_url()
- notes = self.get_notes()
- node = NewNode()
- node.username = username
- node.password = password
- node.url = url
- node.notes = notes
- #node = NewNode(username, password, url, notes)
- node.tags = self.get_tags()
- self._db.addnodes([node])
- print ("Password ID: %d" % (node._id))
- # when done with node erase it
- zerome(password)
- except Exception, e:
- self.error(e)
- def do_print(self, arg):
- for i in self.get_ids(arg):
- try:
- node = self._db.getnodes([i])
- self.print_node(node[0])
- # when done with node erase it
- zerome(node[0]._password)
- except Exception, e:
- self.error(e)
- def do_delete(self, arg):
- ids = self.get_ids(arg)
- try:
- nodes = self._db.getnodes(ids)
- for n in nodes:
- try:
- b = tools.getyesno(("Are you sure you want to"
- " delete '%s@%s'?"
- ) % (n.username, n.url), False)
- except NameError:
- pass
- if b is True:
- self._db.removenodes([n])
- print ("%s@%s deleted" % (n.username, n.url))
- except Exception, e:
- self.error(e)
- def get_password(self, argsgiven, numerics=False, leetify=False,
- symbols=False, special_signs=False,
- reader=getpass.getpass, length=None):
- return tools.getpassword("Password (Blank to generate): ",
- reader=reader, length=length, leetify=leetify)
- class Aliases(BaseCommands, PwmanCliOld):
- """
- Define all the alias you want here...
- """
- def do_cp(self, args):
- self.do_copy(args)
- def do_e(self, arg):
- self.do_edit(arg)
- def do_EOF(self, args):
- return self.do_exit(args)
- def do_l(self, args):
- self.do_list(args)
- def do_ls(self, args):
- self.do_list(args)
- def do_p(self, arg):
- self.do_print(arg)
- def do_rm(self, arg):
- self.do_delete(arg)
- def do_o(self, args):
- self.do_open(args)
- def do_h(self, arg):
- self.do_help(arg)
- def do_n(self, arg):
- self.do_new(arg)
|