Advertisement
thund3rPasteBin

Untitled

Jul 7th, 2018
113
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.59 KB | None | 0 0
  1. @app.route('/', methods=['GET'])
  2. def index():
  3.  
  4.  
  5.  
  6.  
  7. @app.route('/login', methods=['GET', 'POST'])
  8. def login():
  9.     check_host_alive(url_auth_server, url_auth_port)
  10.  
  11.     form = LoginForm(target=target)
  12.     if form.validate_on_submit():
  13.         username = form.login.data
  14.         password = form.password.data
  15.         target = form.target.data
  16.         app.logger.info("Validating form...")
  17.         auth_token = ValidUser(request, username, password)
  18.         app.logger.debug('Token: {}'.format(auth_token))
  19.  
  20.         if auth_token:
  21.             app.logger.debug("Target **********************")
  22.             app.logger.debug(target)
  23.             app.logger.debug(redirect_url())
  24.             resp = make_response(redirect(url_for('index')))
  25.             secure = True if app.debug is False else False
  26.             # Secure limits cookies to HTTPS traffic only.
  27.             # HttpOnly prevents JavaScript from reading the cookie
  28.             app.logger.info('Saving cookies...')
  29.  
  30.             resp.set_cookie('token', auth_token,
  31.                             secure=secure,
  32.                             httponly=True,
  33.                             expires=expire_date,
  34.                             )
  35.             resp.set_cookie('user_name', username,
  36.                             secure=secure,
  37.                             httponly=True,
  38.                             expires=expire_date,
  39.                             )
  40.  
  41.             # Set headers that will be received by the service for this request
  42.             app.logger.info('Setting headers')
  43.             resp.headers['REMOTE_USER'] = username
  44.             resp.headers['X-WEBAUTH-USER'] = username
  45.             resp.headers['X-Forwarded-User'] = username
  46.             app.logger.debug(resp.headers)
  47.             set_token_db_cache(username, auth_token)
  48.  
  49.             return resp
  50.         app.logger.warning('Wrong password! Please repeat!')
  51.         error = 'Invalid credentials.'
  52.         flash('{} Please repeat!'.format(error))
  53.  
  54.     return render_template('login.html', form=form)
  55.  
  56. if __name__ == '__main__':
  57.     handler = RotatingFileHandler('sol.log', maxBytes=10000, backupCount=3)
  58.     # app.logger = logging.getLogger(__name__)
  59.     handler.setLevel(logging.DEBUG)
  60.     formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
  61.     handler.setFormatter(formatter)
  62.     app.logger.addHandler(handler)
  63.     app.logger.info('test')
  64.     app.logger.debug("Current working directory {}".format(os.getcwd()))
  65.  
  66.     create_db()
  67.     app.run(host=listen_interface, port=AUTH_PORT)
  68.  
  69.         # TODO remove and create cache file
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement