Learn how easy it is to sync an existing GitHub or Google Code repo to a SourceForge project! See Demo

Close

[d13b9c]: scripts / migrations / 013-update-ordinals.py Maximize Restore History

Download this file

013-update-ordinals.py    53 lines (43 with data), 1.8 kB

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import sys
import logging
from pylons import c
from ming.orm import session
from ming.orm.ormsession import ThreadLocalORMSession
from allura import model as M
from allura.lib import utils
log = logging.getLogger('update-ordinals')
log.addHandler(logging.StreamHandler(sys.stdout))
def main():
test = sys.argv[-1] == 'test'
num_projects_examined = 0
log.info('Examining all projects for mount order.')
for some_projects in utils.chunked_find(M.Project):
for project in some_projects:
c.project = project
mounts = project.ordered_mounts(include_hidden=True)
# ordered_mounts() means duplicate ordinals (if any) will be next to each other
duplicates_found = False
prev_ordinal = None
for mount in mounts:
if mount['ordinal'] == prev_ordinal:
duplicates_found = True
break
prev_ordinal = mount['ordinal']
if duplicates_found:
if test:
log.info('Would renumber mounts for project "%s".' % project.shortname)
else:
log.info('Renumbering mounts for project "%s".' % project.shortname)
for i, mount in enumerate(mounts):
if 'ac' in mount:
mount['ac'].options['ordinal'] = i
elif 'sub' in mount:
mount['sub'].ordinal = i
ThreadLocalORMSession.flush_all()
num_projects_examined += 1
session(project).clear()
log.info('%s projects examined.' % num_projects_examined)
ThreadLocalORMSession.flush_all()
ThreadLocalORMSession.close_all()
if __name__ == '__main__':
main()