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

Close

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

Download this file

013-update-ordinals.py    68 lines (55 with data), 2.2 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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
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
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 chunked_project_iterator({}):
for project in some_projects:
c.project = project
mounts = project.ordered_mounts(include_search=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()
PAGESIZE=1024
def chunked_project_iterator(q_project):
'''shamelessly copied from refresh-all-repos.py'''
page = 0
while True:
results = (M.Project.query
.find(q_project)
.skip(PAGESIZE*page)
.limit(PAGESIZE)
.all())
if not results: break
yield results
page += 1
if __name__ == '__main__':
main()