Mercurial > p > roundup > code

annotate scripts/copy-user.py @ 5118:57452bc6d989

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
issue2550853 - better error handling and cleanup on some postgres tests by Stuart McGraw. My postgres tests passed both before and after I applied his patch, but the code change seems rational. issue2086536 - back_postgresql: fixing pg_command and prefering psycopg2. Patch done by Philipp Gortan (mephinet). His patch also improves handling of retryable errors. Applied and edited by John Rouillard. Edits included removing support for psycopg1. See: https://sourceforge.net/p/roundup/mailman/message/32855027/ for rational for dropping it Again all 137 postgres tests pass.
author John Rouillard <rouilj@ieee.org>
date Sat, 02 Jul 2016 21:15:23 -0400
parents 6e9b9743de89
children 198b6e810c67
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
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
18 import sys
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
19 import roundup.instance
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
20
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 def copy_user(home1, home2, *userids):
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
23 """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
24
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
25 copyattribs = ['username', 'password', 'address', 'realname', 'phone',
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
26 'organisation', 'alternate_addresses', 'roles', 'timezone']
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
27
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
28 try:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
29 instance1 = roundup.instance.open(home1)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
30 print "Opened source instance: %s" % home1
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
31 except:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
32 print "Can't open source instance: %s" % home1
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
33 sys.exit(1)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
34
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
35 try:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
36 instance2 = roundup.instance.open(home2)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
37 print "Opened target instance: %s" % home2
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
38 except:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
39 print "Can't open target instance: %s" % home2
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
40 sys.exit(1)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
41
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
42 db1 = instance1.open('admin')
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
43 db2 = instance2.open('admin')
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
44
4781
6e9b9743de89 Implementation for:
John Rouillard <rouilj@ieee.org>
parents: 4570
diff changeset
45 db1.tx_Source = "cli"
6e9b9743de89 Implementation for:
John Rouillard <rouilj@ieee.org>
parents: 4570
diff changeset
46 db2.tx_Source = "cli"
6e9b9743de89 Implementation for:
John Rouillard <rouilj@ieee.org>
parents: 4570
diff changeset
47
1943
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
48 userlist = db1.user.list()
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
49 for userid in userids:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
50 try:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
51 userid = str(int(userid))
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
52 except ValueError, why:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
53 print "Not a numeric user id: %s Skipping ..." % (userid,)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
54 continue
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
55 if userid not in userlist:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
56 print "User %s not in source instance. Skipping ..." % userid
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
57 continue
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
58
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
59 user = {}
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
60 for attrib in copyattribs:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
61 value = db1.user.get(userid, attrib)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
62 if value:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
63 user[attrib] = value
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
64 try:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
65 db2.user.lookup(user['username'])
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
66 print "User %s: Username '%s' exists in target instance. Skipping ..." % (userid, user['username'])
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
67 continue
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
68 except KeyError, why:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
69 pass
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
70 print "Copying user %s (%s) ..." % (userid, user['username'])
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
71 db2.user.create(**user)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
72
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
73 db2.commit()
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
74 db2.close()
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
75 print "Closed target instance."
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
76 db1.close()
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
77 print "Closed source instance."
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
78
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 if __name__ == "__main__":
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
81 if len(sys.argv) < 4:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
82 print __doc__
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
83 sys.exit(1)
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
84 else:
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
85 copy_user(*sys.argv[1:])
bd9e2e998e9d added contributed script for copying users between trackers
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
86
Roundup Issue Tracker: http://roundup-tracker.org/