Mercurial > p > roundup > code
annotate scripts/copy-user.py @ 5414:3fa026621f69
Python 3 preparation: comparisons.
Python 3 no longer has the cmp function, or cmp= arguments to sorting
functions / methods (key= must be used instead), and requires rich
comparison methods such as __lt__ to be defined instead of using
__cmp__. All of the comparison mechanisms supported in Python 3 are
also supported in Python 2.
This patch makes the corresponding changes in Roundup to use key
functions and rich comparison methods. In the case of the
JournalPassword and Permission classes, only __eq__ and __ne__ are
defined as I don't see ordered comparisons as useful there (and for
Permission, the old __cmp__ function didn't try to provide a valid
ordering). In the case of the Date class, I kept the __cmp__ method
and implemented the others in terms of it, to avoid excess
repetitiveness in duplicating implementation code for all six rich
comparison methods.
In roundup/admin.py, help_commands_html used operator.attrgetter to
produce the second argument of sorted() - which would be reasonable
for a key function, but the second argument is the cmp function in
Python 2, not a key function (and the key function must be a named
argument not a positional argument in Python 3). That function
appears to be completely unused, so I expect that code never worked.
This patch adds the missing key= to that sorted() call, but it would
also be reasonable to remove the unused function completely instead.
| author | Joseph Myers <jsm@polyomino.org.uk> |
|---|---|
| date | Wed, 25 Jul 2018 00:39:37 +0000 |
| parents | 64b05e24dbd8 |
| children | 2a6c3eb4e059 |
| 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 | 46 db1.tx_Source = "cli" |
| 47 db2.tx_Source = "cli" | |
| 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 |
