def setUp(self):
self.orm.register(Calendar)
self.orm.register(DateValue)
ts = Calendar(name='MyCalendar').save()
for dt, value in izip(dates, values):
ts.add(dt, value)
ts.save()
def fill(self, update=False):
session = self.session()
c = yield session.add(Calendar(name=self.data.random_string()))
with session.begin() as t:
for dt, value in zip(self.data.dates, self.data.values):
t.add(DateValue(dt=dt, value=value))
yield t.on_result
items = t.saved[DateValue._meta]
with session.begin() as t:
if update:
c.data.update(items)
else:
for value in items:
c.data.add(value)
yield t.on_result
yield c
