def add_consumer():
c = Consumer()
g.user.consumers.append(c)
db.session.commit()
return redirect(url_for('.home'))
def setup(self):
super(TestMain, self).setup()
self.cli = self.app.test_client()
self.user = User('test', '*****@*****.**', 'password')
self.consumer = Consumer('annotateit')
self.consumer.secret = 'secret'
db.session.add(self.user)
db.session.add(self.consumer)
db.session.commit()
def setup(self):
super(TestConsumer, self).setup()
c = Consumer('foo')
db.session.add(c)
db.session.commit()
print("Users in DB: " + str(users_count))
if users_count == 0:
print("Creating admin user... ")
u = User(username, email, password)
u.is_admin = True
db.session.add(u)
db.session.commit()
print("done.\n")
print("Creating primary consumer... ")
c = Consumer(ckey)
c.user_id = u.id
c.secret = csecret
db.session.add(c)
db.session.commit()
print("done.\n")
print("Primary consumer key: %s" % c.key)
print("Primary consumer secret: %s" % c.secret)
else:
print("Updating primary consumer... ")
u = User.query.filter(username='******').first()
c = Consumer.query.filter(user_id=u.id).first()
