Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- @app.route('/', methods=['GET'])
- def index():
- @app.route('/login', methods=['GET', 'POST'])
- def login():
- check_host_alive(url_auth_server, url_auth_port)
- form = LoginForm(target=target)
- if form.validate_on_submit():
- username = form.login.data
- password = form.password.data
- target = form.target.data
- app.logger.info("Validating form...")
- auth_token = ValidUser(request, username, password)
- app.logger.debug('Token: {}'.format(auth_token))
- if auth_token:
- app.logger.debug("Target **********************")
- app.logger.debug(target)
- app.logger.debug(redirect_url())
- resp = make_response(redirect(url_for('index')))
- secure = True if app.debug is False else False
- # Secure limits cookies to HTTPS traffic only.
- # HttpOnly prevents JavaScript from reading the cookie
- app.logger.info('Saving cookies...')
- resp.set_cookie('token', auth_token,
- secure=secure,
- httponly=True,
- expires=expire_date,
- )
- resp.set_cookie('user_name', username,
- secure=secure,
- httponly=True,
- expires=expire_date,
- )
- # Set headers that will be received by the service for this request
- app.logger.info('Setting headers')
- resp.headers['REMOTE_USER'] = username
- resp.headers['X-WEBAUTH-USER'] = username
- resp.headers['X-Forwarded-User'] = username
- app.logger.debug(resp.headers)
- set_token_db_cache(username, auth_token)
- return resp
- app.logger.warning('Wrong password! Please repeat!')
- error = 'Invalid credentials.'
- flash('{} Please repeat!'.format(error))
- return render_template('login.html', form=form)
- if __name__ == '__main__':
- handler = RotatingFileHandler('sol.log', maxBytes=10000, backupCount=3)
- # app.logger = logging.getLogger(__name__)
- handler.setLevel(logging.DEBUG)
- formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
- handler.setFormatter(formatter)
- app.logger.addHandler(handler)
- app.logger.info('test')
- app.logger.debug("Current working directory {}".format(os.getcwd()))
- create_db()
- app.run(host=listen_interface, port=AUTH_PORT)
- # TODO remove and create cache file
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement