clean-tle-versioning.py

import models

from lib.utils import AsyncBatchUpdateManager
from lib.utils import QueryBatchIterator

session = models.DBSession.get_by_key_name("18ceb0d6-52b7-4c67-b47c-ae72edfe4394")
print session.is_versioned
print ""

tles = models.TimelineEvent.query_unversioned(session)

mgr = AsyncBatchUpdateManager(batch_size=100)

updcount = 0
delcount = 0

def clean_action(tle):
  global updcount, delcount
  if tle.timeline_version is None: return
  if tle.timeline_version == 1:
    tle = tle.puttable()
    tle.timeline_version = None
    mgr.put(tle)
    updcount += 1
  if tle.timeline_version > 1:
    delcount += 1

QueryBatchIterator.iterate(tles, 100, clean_action)
mgr.complete()

print updcount
print delcount
print ""
print mgr.has_errors()