class Users(Iterator): def __init__(self, hostname, auth): self.admin = AdminClient(hostname) self.auth = auth def next(self, seqid, channel): name = 'user-{}'.format(seqid) password = '******' logger.info('Creating new user {}, auth by {}'.format(name, self.auth)) self.admin.add_user(name=name, password=password, channels=[channel]) if self.auth == 'cookie': cookies = self.admin.create_session(name=name) return {'cookies': cookies} else: return {'name': name, 'password': password}
def __init__(self, hostname, auth): self.admin = AdminClient(hostname) self.auth = auth