def auth_token(): ac = 'Access-Control-' headers = {} if 'origin' in request.headers: headers[ac + 'Allow-Origin'] = request.headers['origin'] headers[ac + 'Allow-Credentials'] = 'true' headers[ac + 'Expose-Headers'] = 'Location, Content-Type, Content-Length' if request.method == 'OPTIONS': headers[ ac + 'Allow-Headers'] = 'X-Requested-With, Content-Type, Content-Length' headers[ac + 'Allow-Methods'] = 'GET, OPTIONS' headers[ac + 'Max-Age'] = '86400' if g.user: c = Consumer.fetch('annotateit') payload = { 'consumerKey': c.key, 'userId': g.user.username, 'ttl': c.ttl } if g.user.is_admin: payload['admin'] = True token = auth.encode_token(payload, c.secret) return Response(token, headers=headers, mimetype='text/plain') else: return Response('Please go to {0} to log in!'.format(request.host_url), status=401, headers=headers, mimetype='text/plain')
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 auth_token(): ac = 'Access-Control-' headers = {} if 'origin' in request.headers: headers[ac + 'Allow-Origin'] = request.headers['origin'] headers[ac + 'Allow-Credentials'] = 'true' headers[ac + 'Expose-Headers'] = 'Location, Content-Type, Content-Length' if request.method == 'OPTIONS': headers[ac + 'Allow-Headers'] = 'X-Requested-With, Content-Type, Content-Length' headers[ac + 'Allow-Methods'] = 'GET, OPTIONS' headers[ac + 'Max-Age'] = '86400' if g.user: c = Consumer.fetch('annotateit') payload = {'consumerKey': c.key, 'userId': g.user.username, 'ttl': c.ttl} if g.user.is_admin: payload['admin'] = True token = auth.encode_token(payload, c.secret) return Response(token, headers=headers, mimetype='text/plain') else: return Response('Please go to {0} to log in!'.format(request.host_url), status=401, headers=headers, mimetype='text/plain')
def test_key(self): c = Consumer.fetch('foo') h.assert_equal(c.key, 'foo')
def consumer(self): if not hasattr(self, '_consumer'): self._consumer = Consumer.fetch('annotateit') return self._consumer
def consumer(self): if not hasattr(self, '_consumer'): self._consumer = Consumer.fetch('annotateit') # Allen change to admin return self._consumer
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()
def setup(self): super(TestConsumer, self).setup() c = Consumer('foo') db.session.add(c) db.session.commit()
def test_default_ttl(self): c = Consumer.fetch('foo') h.assert_equal(c.ttl, 86400)
def test_secret(self): c = Consumer.fetch('foo') assert c.secret, 'Consumer secret should be set!'
password = '' while not password: password = getpass("Admin password: "******"Primary consumer key [annotateit]: ").strip() if not ckey: ckey = 'annotateit' with app.test_request_context(): print("\nCreating admin user... ", end="") u = User(username, email, password) u.is_admin = True db.session.add(u) db.session.commit() print("done.") print("Creating primary consumer... ", end="") c = Consumer(ckey) c.user_id = u.id db.session.add(c) db.session.commit() print("done.\n") print("Primary consumer secret: %s" % c.secret)