import logging
from ming.orm import ThreadLocalORMSession
from allura.lib import utils
from allura import model as M
from forgesvn import model as SM
log = logging.getLogger(__name__)
def kill_tree(repo, commit_id, path, tree):
'''They were arboring terrorists, I swear.'''
M.repo.Tree.query.remove(dict(_id=tree._id))
for tree_rec in tree.tree_ids:
tid = repo._tree_oid(commit_id, path + '/' + tree_rec.name)
child_tree = M.repo.Tree.query.get(_id=tid)
if child_tree:
print ' Found {0}'.format(path + '/' + tree_rec.name)
kill_tree(repo, commit_id, path + '/' + tree_rec.name, child_tree)
else:
print ' Missing {0}'.format(path + '/' + tree_rec.name)
def main():
for chunk in utils.chunked_find(SM.Repository):
for r in chunk:
print 'Processing {0}'.format(r)
all_commit_ids = r._impl.all_commit_ids()
if all_commit_ids:
for commit in M.repo.Commit.query.find({'_id':{'$in':all_commit_ids}}):
if commit.tree_id and M.repo.Tree.query.get(_id=commit.tree_id):
kill_tree(r._impl, commit._id, '', commit.tree)
ThreadLocalORMSession.flush_all()
ThreadLocalORMSession.close_all()
if __name__ == '__main__':
main()