def do_p_edit(self, line): """Edit a project. p_edit <project name>""" session = db.getSession() project = dbutils.getOrCreateProject(line, createIfNeeded=False) if not project: raise YokadiException("Project does not exist.") # Create project line projectLine = parseutils.createLine(project.name, "", project.getKeywordDict()) # Edit line = tui.editLine(projectLine) # Update project projectName, garbage, keywordDict = parseutils.parseLine(line) if garbage: raise BadUsageException("Cannot parse line, got garbage (%s)" % garbage) if not dbutils.createMissingKeywords(list(keywordDict.keys())): return try: project.name = projectName project.setKeywordDict(keywordDict) session.merge(project) session.commit() except IntegrityError: session.rollback() raise YokadiException("A project named %s already exists. Please find another name" % projectName)
def do_t_add_keywords(self, line): """Add keywords to an existing task t_add_keywords <id> <@keyword1> <@keyword2>[=<value>]... """ tokens = parseutils.simplifySpaces(line).split(" ", 1) if len(tokens) < 2: raise YokadiException( "You should give at least two arguments: <task id> <keyword>") task = dbutils.getTaskFromId(tokens[0]) garbage, keywordFilters = parseutils.extractKeywords(tokens[1]) newKwDict = parseutils.keywordFiltersToDict(keywordFilters) if garbage: raise YokadiException( "Cannot parse line, got garbage (%s). Maybe you forgot to add @ before keyword ?" % garbage) if not dbutils.createMissingKeywords(list(newKwDict.keys())): # User cancel keyword creation return kwDict = task.getKeywordDict() kwDict.update(newKwDict) task.setKeywordDict(kwDict) self.session.merge(task) self.session.commit()
def do_p_edit(self, line): """Edit a project. p_edit <project name>""" project = dbutils.getOrCreateProject(line, createIfNeeded=False) if not project: raise YokadiException("Project does not exist.") # Create project line projectLine = parseutils.createLine(project.name, "", project.getKeywordDict()) # Edit line = tui.editLine(projectLine) # Update project projectName, garbage, keywordDict = parseutils.parseLine(line) if garbage: raise BadUsageException("Cannot parse line, got garbage (%s)" % garbage) if not dbutils.createMissingKeywords(keywordDict.keys()): return try: project.name = projectName except DuplicateEntryError: raise YokadiException( "A project named %s already exists. Please find another name" % projectName) project.setKeywordDict(keywordDict)
def updateTaskFromVTodo(task, vTodo): """Update a yokadi task with an ical VTODO object @param task: yokadi task (db.Task object) @param vTodo: ical VTODO (icalendar.Calendar.Todo object)""" for yokadiAttribute, icalAttribute in list( YOKADI_ICAL_ATT_MAPPING.items()): attr = vTodo.get(icalAttribute) if attr: # Convert ical type (vDates, vInt..) to sql alchemy understandable type (datetime, int...) attr = icalutils.convertIcalType(attr) if yokadiAttribute == "title": # Remove (id) attr = re.sub("\s?\(%s\)" % task.id, "", attr) if yokadiAttribute == "doneDate": # A done date defined indicate that task is done task.status = "done" # BUG: Done date is UTC, we must compute local time for yokadi if yokadiAttribute == "urgency": if attr == icalutils.yokadiUrgencyToIcalPriority(task.urgency): # Priority does not change - don't update it continue else: # Priority has changed, we need to update urgency attr = icalutils.icalPriorityToYokadiUrgency(int(attr)) # Update attribute setattr(task, yokadiAttribute, attr) # Update keywords from categories if vTodo.get("categories"): if isinstance(vTodo.get("categories"), (list)): categories = vTodo.get("categories") else: categories = vTodo.get("categories").split(",") keywords = ["@%s" % k for k in categories] garbage, keywordFilters = parseutils.extractKeywords( " ".join(keywords)) newKwDict = parseutils.keywordFiltersToDict(keywordFilters) if garbage: print("Got garbage while parsing categories: %s" % garbage) dbutils.createMissingKeywords(list(newKwDict.keys()), interactive=False) task.setKeywordDict(newKwDict)
def updateTaskFromVTodo(task, vTodo): """Update a yokadi task with an ical VTODO object @param task: yokadi task (db.Task object) @param vTodo: ical VTODO (icalendar.Calendar.Todo object)""" for yokadiAttribute, icalAttribute in YOKADI_ICAL_ATT_MAPPING.items(): attr = vTodo.get(icalAttribute) if attr: # Convert ical type (vDates, vInt..) to sqlobjectunderstandable type (datetime, int...) attr = icalutils.convertIcalType(attr) if yokadiAttribute == "title": # Remove (id) attr = re.sub("\s?\(%s\)" % task.id, "", attr) if yokadiAttribute == "doneDate": # A done date defined indicate that task is done task.status = "done" # BUG: Done date is UTC, we must compute local time for yokadi if yokadiAttribute == "urgency": if attr == icalutils.yokadiUrgencyToIcalPriority(task.urgency): # Priority does not change - don't update it continue else: # Priority has changed, we need to update urgency attr = icalutils.icalPriorityToYokadiUrgency(int(attr)) # Update attribute setattr(task, yokadiAttribute, attr) # Update keywords from categories if vTodo.get("categories"): if isinstance(vTodo.get("categories"), (list)): categories = vTodo.get("categories") else: categories = vTodo.get("categories").split(",") keywords = ["@%s" % k for k in categories] garbage, keywordFilters = parseutils.extractKeywords(" ".join(keywords)) newKwDict = parseutils.keywordFiltersToDict(keywordFilters) if garbage: print "Got garbage while parsing categories: %s" % garbage dbutils.createMissingKeywords(newKwDict.keys(), interactive=False) task.setKeywordDict(newKwDict)
def do_p_add(self, line): """Add new project. Will prompt to create keywords if they do not exist. p_add <projectName> [@<keyword1>] [@<keyword2>]""" if not line: print "Give at least a project name !" return projectName, garbage, keywordDict = parseutils.parseLine(line) if garbage: raise BadUsageException("Cannot parse line, got garbage (%s)" % garbage) try: project = Project(name=projectName) except DuplicateEntryError: raise YokadiException("A project named %s already exists. Please find another name" % projectName) print "Added project '%s'" % projectName if not dbutils.createMissingKeywords(keywordDict.keys()): return None project.setKeywordDict(keywordDict)
def do_p_add(self, line): """Add new project. Will prompt to create keywords if they do not exist. p_add <projectName> [@<keyword1>] [@<keyword2>]""" if not line: print "Give at least a project name !" return projectName, garbage, keywordDict = parseutils.parseLine(line) if garbage: raise BadUsageException("Cannot parse line, got garbage (%s)" % garbage) try: project = Project(name=projectName) except DuplicateEntryError: raise YokadiException( "A project named %s already exists. Please find another name" % projectName) print "Added project '%s'" % projectName if not dbutils.createMissingKeywords(keywordDict.keys()): return None project.setKeywordDict(keywordDict)
def do_t_add_keywords(self, line): """Add keywords to an existing task t_add_keywords <id> <@keyword1> <@keyword2>[=<value>]... """ tokens = parseutils.simplifySpaces(line).split(" ", 1) if len(tokens) < 2: raise YokadiException("You should give at least two arguments: <task id> <keyword>") task = dbutils.getTaskFromId(tokens[0]) garbage, keywordFilters = parseutils.extractKeywords(tokens[1]) newKwDict = parseutils.keywordFiltersToDict(keywordFilters) if garbage: raise YokadiException("Cannot parse line, got garbage (%s). Maybe you forgot to add @ before keyword ?" % garbage) if not dbutils.createMissingKeywords(newKwDict.keys()): # User cancel keyword creation return kwDict = task.getKeywordDict() kwDict.update(newKwDict) task.setKeywordDict(kwDict)
def do_p_add(self, line): """Add new project. Will prompt to create keywords if they do not exist. p_add <projectName> [@<keyword1>] [@<keyword2>]""" if not line: print("Give at least a project name !") return projectName, garbage, keywordDict = parseutils.parseLine(line) session = db.getSession() if garbage: raise BadUsageException("Cannot parse line, got garbage (%s)" % garbage) try: project = Project(name=projectName) session.add(project) session.commit() except IntegrityError: session.rollback() raise YokadiException("A project named %s already exists. Please find another name" % projectName) print("Added project '%s'" % projectName) if not dbutils.createMissingKeywords(list(keywordDict.keys())): return None project.setKeywordDict(keywordDict) session.merge(project) session.commit()
def _t_edit(self, line, keywordEditor=None): """Code shared by t_edit and bug_edit. if keywordEditor is not None it will be called after editing the task. Returns the modified task if OK, None if cancelled""" def editComplete(text, state): """ Specific completer for the edit prompt. This subfunction should stay here because it needs to access to cmd members""" if state == 0: origline = readline.get_line_buffer() line = origline.lstrip() stripped = len(origline) - len(line) begidx = readline.get_begidx() - stripped endidx = readline.get_endidx() - stripped if begidx > 0: self.completion_matches = projectAndKeywordCompleter( "", text, line, begidx, endidx, shift=1) else: self.completion_matches = [] try: return self.completion_matches[state] except IndexError: return None task = self.getTaskFromId(line) if self.cryptoMgr.isEncrypted(task.title): self.cryptoMgr.force_decrypt = True # Decryption must be turned on to edit title = self.cryptoMgr.decrypt(task.title) # Create task line keywordDict = task.getKeywordDict() userKeywordDict, keywordDict = dbutils.splitKeywordDict(keywordDict) taskLine = parseutils.createLine("", title, userKeywordDict) oldCompleter = readline.get_completer( ) # Backup previous completer to restore it in the end readline.set_completer(editComplete) # Switch to specific completer # Edit try: while True: print("(Press Ctrl+C to cancel)") try: line = tui.editLine(taskLine) if not line.strip(): tui.warning("Missing title") continue except KeyboardInterrupt: print() print("Cancelled") return None _, title, userKeywordDict = parseutils.parseLine( task.project.name + " " + line) if self.cryptoMgr.isEncrypted(task.title): title = self.cryptoMgr.encrypt(title) if dbutils.createMissingKeywords(userKeywordDict.keys()): # We were able to create missing keywords if there were any, # we can now exit the edit loop break finally: readline.set_completer(oldCompleter) keywordDict.update(userKeywordDict) if keywordEditor: keywordEditor(keywordDict) task.title = title task.setKeywordDict(keywordDict) return task
def _t_edit(self, line, keywordEditor=None): """Code shared by t_edit and bug_edit. if keywordEditor is not None it will be called after editing the task. Returns the modified task if OK, None if cancelled""" def editComplete(text, state): """ Specific completer for the edit prompt. This subfunction should stay here because it needs to access to cmd members""" if state == 0: origline = readline.get_line_buffer() line = origline.lstrip() stripped = len(origline) - len(line) begidx = readline.get_begidx() - stripped endidx = readline.get_endidx() - stripped if begidx > 0: self.completion_matches = projectAndKeywordCompleter("", text, line, begidx, endidx, shift=1) else: self.completion_matches = [] try: return self.completion_matches[state] except IndexError: return None task = self.getTaskFromId(line) # Create task line keywordDict = task.getKeywordDict() userKeywordDict, keywordDict = dbutils.splitKeywordDict(keywordDict) taskLine = parseutils.createLine("", task.title, userKeywordDict) oldCompleter = readline.get_completer() # Backup previous completer to restore it in the end readline.set_completer(editComplete) # Switch to specific completer # Edit try: while True: print("(Press Ctrl+C to cancel)") try: line = tui.editLine(taskLine) if not line.strip(): tui.warning("Missing title") continue except KeyboardInterrupt: print() print("Cancelled") return None _, title, userKeywordDict = parseutils.parseLine(task.project.name + " " + line) if dbutils.createMissingKeywords(userKeywordDict.keys()): # We were able to create missing keywords if there were any, # we can now exit the edit loop break finally: readline.set_completer(oldCompleter) keywordDict.update(userKeywordDict) if keywordEditor: keywordEditor(keywordDict) task.title = title task.setKeywordDict(keywordDict) return task