twisted-credentials

from twisted.internet import reactor
from twisted.internet.protocol import ServerFactory,Protocol
from twisted.protocols.basic import LineReceiver

from zope.interface import Interface,implements

from twisted.cred import checkers,credentials,portal


class IEchoAvatar(Interface):
	def logout(self):
		'''
		Interface 4 avatar interface
		'''

class EchoAvatar(object):
	implements(IEchoAvatar)

	def logout(self):
		print 'Logout 4 EchoAvatar'
		pass

class EchoRealm(object):
	implements(portal.IRealm)

	def requestAvatar(avatarId, mind, *interfaces):
		if IEchoAvatar in interfaces:
			avatar = EchoAvatar
			return IEchoAvatar, avatar, avatar.logout
		raise NotImplementedError('Only implements EchoAvatar Interface!')


class EchoProtocol(LineReceiver):
	def __init__(self):
		self._avatar = None
		realm = EchoRealm()
		self._portal = portal.Portal(realm)
		check = checkers.InMemoryUsernamePasswordDatabaseDontUse()
		check.addUser('user', 'pass')
		self._portal.registerChecker(check)

	def lineReceived(self, line):
		if not self._avatar:
			log_user = line.strip().split(':')
			if len(log_user) == 2:
				self.login(log_user[0], log_user[1])
			else:
				print 'please input user and password'
		else:
			self.sendLine(line)

	def login(self, user, password):
		d = self._portal.login(
			credentials.UsernamePassword(user,password),
			None,
			IEchoAvatar)
		d.addCallbacks(self._cbLoginSuccess, self._cbLoginFail)
	
	def _cbLoginSuccess(self,(interface, avatar, logout) ):
		self._avatar = avatar
		self.sendLine('Login success!')

	def _cbLoginFail(self, fail):
		self.sendLine('Login failed!')
		self.transport.loseConnection()

f = ServerFactory()
f.protocol = EchoProtocol
reactor.listenTCP(8000, f)
reactor.run()