def init_data(db): """Initializes DB with some dummy data.""" account1 = Account(name="Fermi Corp", website="http://fermigier.com/") db.session.add(account1) contact1 = Contact(first_name="Stefane", last_name="Fermigier", email="*****@*****.**") contact1.account = account1 contact2 = Contact(first_name="Paul", last_name="Dupont", email="*****@*****.**") contact2.account = account1 user1 = User(first_name="Stefane", last_name="Fermigier", email="*****@*****.**", password="******") photo_path = join(dirname(__file__), "..", "dummy_files", "mugshot.jpg") user1.photo = open(photo_path).read() group1 = Group(name="Group 1") group1.photo = open(photo_path).read() user1.join(group1) #db.session.add(contact1) #db.session.add(contact2) db.session.add(user1) db.session.commit() assert len(Contact.query.all()) == 2 assert len(Account.query.all()) == 1 assert len(User.query.all()) == 1 assert len(Group.query.all()) == 1
def load_groups(self): reader = self.get_reader("Groups.csv") for line in reader: d = {} for col in ['name', 'description']: d[col] = line[col] group = Group(**d) photo_path = join(dirname(__file__), "..", "group_photos", line['pic']) group.photo = open(photo_path).read() for user in self.users: group.members.append(user) group.admins.append(choice(self.users)) self.db.session.add(group)