Mercurial > p > roundup > code

annotate scripts/copy-user.py @ 7665:6f841a55eabf

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
fix: database error handling during import of a non-user item The code to handle the case of retired and active users causing errors during import can get called on other db errors as well (e.g. out of memory in postgresql). The code that trys to detect/repair the out of order user case hard coded the db.user class. This crashed when a db error occurred when loading another data object (e.g. msg). Fix the crash by using the proper db object class for the object class being loaded. Credit to Norbert Schlemmer for finding this.
author John Rouillard <rouilj@ieee.org>
date Wed, 18 Oct 2023 15:49:53 -0400
parents 64b05e24dbd8
children 2a6c3eb4e059
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
1 #!/usr/bin/env python
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
2 # Copyright (C) 2003 by Intevation GmbH
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
3 # Author:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
4 # Thomas Arendsen Hein <thomas@intevation.de>
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
5 #
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
6 # This program is free software dual licensed under the GPL (>=v2)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
7 # and the Roundup Licensing (see COPYING.txt in the roundup distribution).
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
8
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
9 """
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
10 copy-user <instance-home> <instance-home> <userid> [<userid>...]
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
11
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
12 Copy one or more Roundup users from one tracker instance to another.
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
13 Example:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
14 copy-user /roundup/tracker1 /roundup/tracker2 `seq 3 10` 14 16
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
15 (copies users 3, 4, 5, 6, 7, 8, 9, 10, 14 and 16)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
16 """
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
17
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
18 from __future__ import print_function
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
19 import sys
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
20 import roundup.instance
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
21
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
22
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
23 def copy_user(home1, home2, *userids):
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
24 """Copy users which are listed by userids from home1 to home2"""
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
25
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
26 copyattribs = ['username', 'password', 'address', 'realname', 'phone',
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
27 'organisation', 'alternate_addresses', 'roles', 'timezone']
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
28
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
29 try:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
30 instance1 = roundup.instance.open(home1)
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
31 print("Opened source instance: %s" % home1)
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
32 except:
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
33 print("Can't open source instance: %s" % home1)
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
34 sys.exit(1)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
35
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
36 try:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
37 instance2 = roundup.instance.open(home2)
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
38 print("Opened target instance: %s" % home2)
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
39 except:
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
40 print("Can't open target instance: %s" % home2)
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
41 sys.exit(1)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
42
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
43 db1 = instance1.open('admin')
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
44 db2 = instance2.open('admin')
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
45
4781
6e9b9743de89 Implementation for:
John Rouillard <rouilj@ieee.org>
parents: 4570
diff changeset
46 db1.tx_Source = "cli"
6e9b9743de89 Implementation for:
John Rouillard <rouilj@ieee.org>
parents: 4570
diff changeset
47 db2.tx_Source = "cli"
6e9b9743de89 Implementation for:
John Rouillard <rouilj@ieee.org>
parents: 4570
diff changeset
48
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
49 userlist = db1.user.list()
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
50 for userid in userids:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
51 try:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
52 userid = str(int(userid))
5248
198b6e810c67 Use Python-3-compatible 'as' syntax for except statements
Eric S. Raymond <esr@thyrsus.com>
parents: 4781
diff changeset
53 except ValueError as why:
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
54 print("Not a numeric user id: %s Skipping ..." % (userid,))
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
55 continue
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
56 if userid not in userlist:
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
57 print("User %s not in source instance. Skipping ..." % userid)
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
58 continue
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
59
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
60 user = {}
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
61 for attrib in copyattribs:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
62 value = db1.user.get(userid, attrib)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
63 if value:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
64 user[attrib] = value
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
65 try:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
66 db2.user.lookup(user['username'])
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
67 print("User %s: Username '%s' exists in target instance. Skipping ..." % (userid, user['username']))
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
68 continue
5248
198b6e810c67 Use Python-3-compatible 'as' syntax for except statements
Eric S. Raymond <esr@thyrsus.com>
parents: 4781
diff changeset
69 except KeyError as why:
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
70 pass
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
71 print("Copying user %s (%s) ..." % (userid, user['username']))
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
72 db2.user.create(**user)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
73
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
74 db2.commit()
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
75 db2.close()
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
76 print("Closed target instance.")
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
77 db1.close()
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
78 print("Closed source instance.")
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
79
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
80
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
81 if __name__ == "__main__":
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
82 if len(sys.argv) < 4:
5376
64b05e24dbd8 Python 3 preparation: convert print to a function.
Joseph Myers <jsm@polyomino.org.uk>
parents: 5248
diff changeset
83 print(__doc__)
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
84 sys.exit(1)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
85 else:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
86 copy_user(*sys.argv[1:])
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
87
Roundup Issue Tracker: http://roundup-tracker.org/