timeline_version migration

migrate.py

import models
import datetime
import time
from google.appengine.ext import ndb

# Upgrade sessions and set timeline_version to 1
#sessions = models.DBSession.all().run()
#for s in sessions:
#  if s.timeline_version:
#    continue
#  s.timeline_version = 1
#  s.put()

# Upgrade TimelineEvents and set timeline_version to 1 and sequence_number to utc_timestamp(start):
tles = models.TimelineEvent.query()
futs = []
for tle in tles:
  modded = False
  if not tle.timeline_version:
    tle.timeline_version = 1
    modded = True

  if not tle.sequence_number:
    # UTC timestamp (# of milliseconds since 1970)
    tle.sequence_number = int(time.mktime(tle.start.timetuple())*1e3 + tle.start.microsecond/1e3)
    modded = True

  if not modded:
    continue

  # Need to special-case handle any TextProperty()s, but we'll hack for the drawing_args case specifically:
  drawing_args = None
  if hasattr(tle, 'drawing_args'):
    drawing_args = getattr(tle, 'drawing_args')
    delattr(tle, 'drawing_args')

    # Add back the drawing_args through the proper channels:
    if drawing_args:
      drawing_args_dict = drawing_args.to_dict()
      safe_drawing_args = models.SlideAnnotationEventArgs(**drawing_args_dict)
      tle.drawing_args = safe_drawing_args

  futs.append(tle.put_async())
  if len(futs) >= 50:
    ndb.Future.wait_all(futs)
    for fut in futs:
      fut.get_result()
    futs = []

if futs is not None and len(futs) > 0:
  ndb.Future.wait_all(futs)
  for fut in futs:
    fut.get_result()

sequence_numbers.py

import models
import datetime
import time
from google.appengine.ext import ndb

# Upgrade TimelineEvents and set sequence_number to utc_timestamp(start):
tles = models.TimelineEvent.query()
for tle in tles:
  modded = False
  if not tle.sequence_number:
    # UTC timestamp (# of milliseconds since 1970)
    tle.sequence_number = int(time.mktime(tle.start.timetuple())*1e3 + tle.start.microsecond/1e3)
    modded = True

  if not modded:
    continue

  # Need to special-case handle any TextProperty()s, but we'll hack for the drawing_args case specifically:
  drawing_args = None
  if hasattr(tle, 'drawing_args'):
    drawing_args = getattr(tle, 'drawing_args')
    delattr(tle, 'drawing_args')

    # Add back the drawing_args through the proper channels:
    if drawing_args:
      drawing_args_dict = drawing_args.to_dict()
      safe_drawing_args = models.SlideAnnotationEventArgs(**drawing_args_dict)
      tle.drawing_args = safe_drawing_args

  tle.put()