from ucsschool.importer.utils.username_handler import UsernameHandler
import ucsschool.importer.utils.username_handler
ucsschool.importer.utils.username_handler.noObject = KeyError


class UsernameHandler(UsernameHandler):
	__doc__ = UsernameHandler.__doc__
	usernames = {}

	def add_to_ldap(self, username, first_number):
		self.usernames[username] = int(first_number)

	def get_next_number(self, username):
		return self.usernames[username]

	def get_and_raise_number(self, username):
		cur = self.get_next_number(username)
		self.usernames.setdefault(username, 0)
		self.usernames[username] += 1
		return str(cur)


import doctest
doctest.testmod()