Attachment 9634 Details for Bug 47329

migrate-recordUID

migrate-recordUID (text/x-python), 5.32 KB, created by Sönke Schwardt-Krummrich on 2018-08-22 11:29 CEST

(hide)

Description:

Filename:

MIME Type:

Creator: Sönke Schwardt-Krummrich

Created: 2018-08-22 11:29 CEST

Size: 5.32 KB

patch

obsolete

>#!/usr/bin/python
># -*- coding: utf-8 -*-
>#
># Copyright 2018 Univention GmbH
>#
># http://www.univention.de/
>#
># All rights reserved.
>#
># The source code of this program is made available
># under the terms of the GNU Affero General Public License version 3
># (GNU AGPL V3) as published by the Free Software Foundation.
>#
># Binary versions of this program provided by Univention to you as
># well as other copyrighted, protected or trademarked materials like
># Logos, graphics, fonts, specific documentations and configurations,
># cryptographic keys etc. are subject to a license agreement between
># you and Univention and not subject to the GNU AGPL V3.
>#
># In the case you use this program under the terms of the GNU AGPL V3,
># the program is provided in the hope that it will be useful,
># but WITHOUT ANY WARRANTY; without even the implied warranty of
># MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
># GNU Affero General Public License for more details.
>#
># You should have received a copy of the GNU Affero General Public
># License with the Debian GNU/Linux or Univention distribution in file
># /usr/share/common-licenses/AGPL-3; if not, see
># <http://www.gnu.org/licenses/>.
>#
>
>import os
>import csv
>from optparse import OptionParser
>from ldap.filter import filter_format
>from univention.config_registry import ConfigRegistry
>from univention.uldap import getAdminConnection
>from ucsschool.importer.models.import_user import ImportUser
>
>ucr = ConfigRegistry()
>ucr.load()
>
>
>def process_csv(options):  # type: (Any) -> None
>	lo = getAdminConnection()
>	reader = csv.reader(open(options.infile, "rb"), delimiter=',')
>
>	writer = csv.writer(open(options.outfile, 'w'), delimiter=';', quotechar='"', quoting=csv.QUOTE_ALL)
>	writer.writerow(['Firstname', 'Lastname', 'Birthday', 'OldRecordUID', 'NewRecordUID', 'DN', 'Comment'])
>
>	for line_cnt, row in enumerate(reader):
>		if line_cnt == 0 and options.ignore_header:
>			continue
>		firstname = row[options.col_firstname]
>		lastname = row[options.col_lastname]
>		birthday = row[options.col_birthday]
>		recordUID = row[options.col_recordUID]
>
>		if any(not(x.strip()) for x in (firstname, lastname, birthday, recordUID)):
>			writer.writerow([firstname, lastname, birthday, '', recordUID, '', 'ERROR: not all values are set'])
>			continue
>
>		firstname_normalized = ImportUser.normalize(firstname)
>		lastname_normalized = ImportUser.normalize(lastname)
>
>		filter_s = filter_format('(&(ucsschoolSourceUid=%)(|(&(givenName=%s)(sn=%s))(&(givenName=%s)(sn=%s)))(univentionBirthday=%s))', (
>			options.sourceUID,
>			firstname,
>			lastname,
>			firstname_normalized,
>			lastname_normalized,
>			birthday,
>		))
>		result = lo.search(filter=filter_s)
>		if not result:
>			writer.writerow([firstname, lastname, birthday, '', recordUID, '', 'ERROR: not found in LDAP'])
>		else:
>			comment = ''
>			if len(result) > 1:
>				comment = 'ERROR: found several users - using only the first one!'
>
>			dn, attrs = result[0]
>			oldRecordUID = attrs.get('ucsschoolRecordUID', [''])[0]
>			writer.writerow([firstname, lastname, birthday, oldRecordUID, recordUID, dn, comment])
>			if options.dry_run:
>				print 'DRY_RUN: Not updating %s' % (dn,)
>			else:
>				print 'Updating %s  (%s --> %s)' % (dn, oldRecordUID, recordUID)
># 				lo.modify(
># 					dn,
># 					[
># 						['ucsschoolRecordUID', [oldRecordUID], [recordUID]],
># 					])
>
>def main():  # type: () -> None
>	parser = OptionParser()
>	parser.add_option('--dry-run', action='store_true', dest='dry_run', default=False, help='perform a dry run')
>	parser.add_option('--column-firstname', action='store', dest='col_firstname', default=1, type='int', help='column number for firstname (counting starts with 0! Default is %default)')
>	parser.add_option('--column-lastname', action='store', dest='col_lastname', default=0, type='int', help='column number for lastname (counting starts with 0! Default is %default)')
>	parser.add_option('--column-birthday', action='store', dest='col_birthday', default=2, type='int', help='column number for birthday (counting starts with 0! Default is %default)')
>	parser.add_option('--column-recordUID', action='store', dest='col_recordUID', default=5, type='int', help='column number for recordUID (counting starts with 0! Default is %default)')
>	parser.add_option('--sourceUID', action='store', dest='sourceUID', default=None, help='sourceUID to be set')
>	parser.add_option('--ignore-header', action='store_true', dest='ignore_header', default=False, help='ignore first/header line of CSV file')
>	parser.add_option('-i', metavar='INFILE', action='store', dest='infile', default=None, help='path to input file')
>	parser.add_option('-o', metavar='OUTFILE', action='store', dest='outfile', default=None, help='path to output file')
>
>	(options, args) = parser.parse_args()
>
>	if not options.infile or not os.path.isfile(options.infile):
>		parser.error('Please specify a path to a input CSV file via argument -i <INFILE>')
>	if not options.outfile or os.path.isfile(options.outfile):
>		parser.error('Please specify a path for a output CSV file via argument -o <INFILE>. The file may not exist!')
>	if any(x is None for x in (options.col_firstname, options.col_lastname, options.col_birthday, options.col_recordUID)):
>		parser.error('Please specify column numbers for firstname, lastname, birthday and recordUID!')
>	if not options.sourceUID:
>		parser.error('Please specify --sourceUID.')
>	process_csv(options)
>
>
>if __name__ == '__main__':
>	main()

Actions: View

Attachments on bug 47329: 9634