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


class EventBeacon(ndb.Expando):
    pass


def drop_event_beacons():
    query = EventBeacon.query()
    total_deleted = 0
    while True:
        keys = query.fetch(1000, keys_only=True)
        if not keys:
            break
        ndb.delete_multi(keys)
        total_deleted += len(keys)
        print "total deleted: %s" % total_deleted


def drop_events():
    query = Event.query(Event.timestamp < datetime.datetime(2014, 9, 1, 0, 0, 0))
    total_deleted = 0
    while True:
        keys = query.fetch(1000, keys_only=True)
        if not keys:
            break
        ndb.delete_multi(keys)
        total_deleted += len(keys)
        print "total deleted: %s" % total_deleted