def add_test_works(): print 'Creating Works...' from david.modules.works.model import cat2type, Work from david.modules.artist.model import Artist with app.test_client() as c: # send a fake request, so `current_app` can be the app rv = c.get('/') artist1 = Artist() artist1.name = '佟大为' artist1.uid = 'tong' artist2 = Artist() artist2.name = '关悦' artist2.uid = 'guan' db.session.add_all([artist1, artist2]) db.session.commit() artist1, artist2 = Artist.query.all() artist1.desc = artist2.desc = lorem one_year = datetime.timedelta(days=366) one_month = datetime.timedelta(days=30) for cat, model in cat2type.items(): pubdate = datetime.date.today() + one_year for i in range(20): pubdate -= one_year for i in range(7): pubdate -= one_month obj = Work() obj.title = '%s作品%s' % (model.__name__, i + 1) obj.cat = cat obj.pubdate = pubdate obj.desc = lorem obj.artist_id = artist1.id if i % 2 else artist2.id obj.owner_id = User.query.first().id db.session.add(obj) db.session.commit() print 'Done.' print
def artist(self): return Artist.get(self.artist_id)