Attachment 7642 Details for Bug 41162

migration script

migrate.py (text/x-python), 2.46 KB, created by Florian Best on 2016-05-09 10:49 CEST

(hide)

Description:

Filename:

MIME Type:

Creator: Florian Best

Created: 2016-05-09 10:49 CEST

Size: 2.46 KB

patch

obsolete

>#!/usr/bin/python2.7
>import ldap
>import sys
>
>from univention.admin import uldap, uexceptions
>from ucsschool.lib.schoolldap import SchoolSearchBase
>
>ldap_errors = (ldap.LDAPError, uexceptions.base,)
>
>
>class MigrationFailed(Exception):
>
>	def __init__(self, errors):
>		super(Exception, self).__init__(errors)
>		self.errors = errors
>
>	def _format(self):
>		yield '%d error(s) occurred' % (len(self.errors),)
>		for (etype, exc, etraceback), dn in self.errors:
>			yield '%s: %s: %s' % (etype.__name__, dn, exc)
>
>	def __str__(self):
>		return '\n'.join(self._format())
>
>
>def migrate(lo):
>	errors = []
>	try:
>		schools = lo.search('objectClass=ucsschoolOrganizationalUnit', attr=['ou'])
>	except ldap_errors:
>		errors.append((sys.exc_info(), None))
>		raise MigrationFailed(errors)
>
>	for dn, attrs in schools:
>		school = attrs['ou'][0]
>		yield 'Migrating users of school %r' % (school,)
>		base = SchoolSearchBase([school], dn=dn)
>		mapping = [
>			(base.students, 'ucsschoolStudent'),
>			(base.teachers, 'ucsschoolTeacher'),
>			(base.teachersAndStaff, ('ucsschoolTeacher', 'ucsschoolStaff')),
>			(base.staff, 'ucsschoolStaff'),
>			(base.admins, 'ucsschoolAdministrator'),
>		]
>		for container, object_classes in mapping:
>			yield 'Migrating users underneath of %r' % (container,)
>			if isinstance(object_classes, basestring):
>				object_classes = [object_classes]
>			try:
>				objects = lo.search(base=container, scope='one')
>			except uexceptions.noObject:
>				yield 'Container %r does not exists' % (container,)
>				continue
>			except ldap_errors:
>				errors.append((sys.exc_info(), container))
>				continue
>			for object_dn, object_attrs in objects:
>				old_schools = object_attrs.get('ucsschoolSchool', [])
>				old_object_classes = object_attrs.get('objectClass', [])
>				new_schools = set(old_schools) | set([school])
>				new_object_classes = set(old_object_classes) | set(object_classes)
>				yield 'Adding %r into school %r' % (object_dn, ', '.join(new_schools))
>				try:
>					lo.modify(object_dn, [
>						('ucsschoolSchool', list(old_schools), list(new_schools)),
>						('objectClass', list(old_object_classes), list(new_object_classes))
>					])
>				except ldap_errors:
>					errors.append((sys.exc_info(), object_dn))
>	if errors:
>		raise MigrationFailed(errors)
>
>
>if __name__ == '__main__':
>	try:
>		lo, po = uldap.getAdminConnection()
>	except IOError:
>		raise SystemExit('This script must be executed on a DC Master.')
>	try:
>		for message in migrate(lo):
>			print message
>	except MigrationFailed as exc:
>		raise SystemExit(str(exc))

Actions: View

Attachments on bug 41162: 7642