from flask import Flask
from flask_jwt_login import JWT

app = Flask(__name__)	# create app object
jwt = JWT(app)			# initialize flask_jwt_login

from flask import Flask

app = Flask(__name__)
app.config.from_object('config.Config')

class Config(object):
    SECRET_KEY = 'random secret key for development'
    HASH_ALGORITHM = 'HS512' 
    # hash algorithm to use at encode and decode token
    JWT_COOKIE_NAME = 'token'
    # token name to be used
    
    # if you don't specify HASH_ALGORITHM or JWT_COOKIE_NAME,
    # they will have default value. (HS512 and token)

...
...

# initialize
jwt = JWT(app)

# user data class
# I want to recommend you to add hashed passsword data into token
class User():
	def __init__(self, id, pw, name):
		self.id = id
		self.pw = pw
		self.name = name

	def __repr__(self):
		return "User(id=%s, password=%s, name=%s)" % (self.id, self.pw, self.name)
		
# You have to write a function that check users' ids and passwords.
@jwt.authentication_handler
def authentication_handler(id, pw):
	for row in user_table:
		if row['id'] == id and row['pw'] == pw:
			return User(row['id'], row['pw'], row['name'])

	# if there is no matching user, returns None
	return None

from flask_jwt_login import process_login

@app.route('/some_url')
def some_function():
	token = process_login(request.form["id"], request.form["pw"])
	# this token will be the value returned from authentication handler
	
	response = make_response("sign in")
	response.set_cookie(TOKEN_NAME, token)
	return response

@jwt.unauthorized_handler
def unauthorized_handler():
	# if authentication failed, this handler will be performed
	return 'Unauthorized Access', 501

# this url is only accessed by users who have a valid token.
@app.route("/protected")
@login_required
def protected():
	return "Protected Page. name :" + get_current_user()["name"]