fix crash when a command doesn't have a docstring

This commit is contained in:
Romain Bignon 2012-02-02 08:44:00 +01:00
commit c009b26864

View file

@ -418,16 +418,18 @@ class ReplApplication(Cmd, ConsoleApplication):
doc = getattr(self, 'do_' + command).__doc__ doc = getattr(self, 'do_' + command).__doc__
except AttributeError: except AttributeError:
return None return None
if doc: if not doc:
doc = '\n'.join(line.strip() for line in doc.strip().split('\n')) return '%s' % command
if not doc.startswith(command):
doc = '%s\n\n%s' % (command, doc)
if short:
doc = doc.split('\n')[0]
if not doc.startswith(command):
doc = command
return doc doc = '\n'.join(line.strip() for line in doc.strip().split('\n'))
if not doc.startswith(command):
doc = '%s\n\n%s' % (command, doc)
if short:
doc = doc.split('\n')[0]
if not doc.startswith(command):
doc = command
return doc
def get_commands_doc(self): def get_commands_doc(self):
names = set(name for name in self.get_names() if name.startswith('do_')) names = set(name for name in self.get_names() if name.startswith('do_'))