def run(self, edit): # get selected text (lineText, methodName) = self._get_selection() (objName, args) = self._parse_selection(methodName, lineText) # get all defenition of object in current view points = self._get_object_defs(objName) className = self.get_class_name(points, self.lineRegion) numOfArgs = self._get_numofargs(args) methodText = ClassMenager.get_method_text(methodName, numOfArgs) self.add_method(className, methodText, edit)
def on_done(self, fileName): root = self._get_base() path = os.path.join(root, fileName) path = os.path.normpath(path) classText = ClassMenager.get_class_text(self.className) assert(classText) fileUpdated = False if not os.path.exists(path): self._create_new_file(path, classText) fileUpdated = True fileView = self.window.open_file(path) if not fileUpdated: fEdit = fileView.begin_edit() # TODO: at the moment there is an assumption that file is empty # adding new text at the begining fileView.insert(fEdit, 0, "classText") fileView.end_edit(fEdit)