import datetime from lib.tools import getConfig import logging from zombietweet import tClient from models.zombie import Zombie from models.schedule import Schedule config = getConfig() now = datetime.datetime.utcnow() todo = Schedule.objects(start_date__lte=now, end_date__gte=now) logging.basicConfig(level=logging.DEBUG) logging.debug("Fetching data at %s" % now) fetched = tClient.fetch(todo) for f in fetched: found = Zombie.objects(oid=f['oid']) if len(found) == 0: try: zz = Zombie() zz.create_zombie(f) except: pass
def count_schedule(self): from models.schedule import Schedule return Schedule.objects(uid=self.username).count()