def iter_data():
for data in iter_site_cafes():
yield Cafe(**data)
for data in iter_user_cafes():
yield Cafe(**data)
for data in iter_cafe_topics():
yield CafeTopic(**data)
def add_cafe_topic(self, title, content, user_id, cafe_id):
t = Topic(title=title, content=content, user_id=user_id)
db.session.add(t)
db.session.flush()
ct = CafeTopic(
cafe_id=cafe_id,
topic_id=t.id,
user_id=user_id,
status=CafeTopic.STATUS_PUBLIC,
)
db.session.add(ct)
return t
def create_public_topic(self):
cafe = Cafe(
name=u'pub', slug='pub', user_id=1,
permission=Cafe.PERMISSION_PUBLIC, status=9,
)
db.session.add(cafe)
topic = Topic(title=u'hello', content='', user_id=1)
db.session.add(topic)
db.session.flush()
ct = CafeTopic(cafe_id=cafe.id, topic_id=topic.id, user_id=1)
db.session.add(ct)
db.session.commit()
return topic
