diff options
-rw-r--r-- | paramiko/__init__.py | 13 |
1 files changed, 4 insertions, 9 deletions
diff --git a/paramiko/__init__.py b/paramiko/__init__.py index 4d970c5b..ceefe474 100644 --- a/paramiko/__init__.py +++ b/paramiko/__init__.py @@ -92,15 +92,10 @@ from hostkeys import HostKeys from config import SSHConfig # fix module names for epydoc -for x in (Transport, SecurityOptions, Channel, SFTPServer, SSHException, - PasswordRequiredException, BadAuthenticationType, ChannelFile, - SubsystemHandler, AuthHandler, RSAKey, DSSKey, SFTPError, - SFTP, SFTPClient, SFTPServer, Message, Packetizer, SFTPAttributes, - SFTPHandle, SFTPServerInterface, BufferedFile, Agent, AgentKey, - PKey, BaseSFTP, SFTPFile, ServerInterface, HostKeys, SSHClient, - MissingHostKeyPolicy, AutoAddPolicy, RejectPolicy, ChannelException, - SSHConfig, BadHostKeyException, AuthenticationException): - x.__module__ = 'paramiko' +for c in locals().values(): + if issubclass(type(c), type) or type(c).__name__ == 'classobj': + # classobj for exceptions :/ + c.__module__ = __name__ from common import AUTH_SUCCESSFUL, AUTH_PARTIALLY_SUCCESSFUL, AUTH_FAILED, \ OPEN_SUCCEEDED, OPEN_FAILED_ADMINISTRATIVELY_PROHIBITED, OPEN_FAILED_CONNECT_FAILED, \ |