Adding macros current_user_id & current_username (#2582)

* Adding macros current_user_id & current_username

* Addressing comments
This commit is contained in:
Maxime Beauchemin 2017-04-10 21:42:25 -07:00 committed by GitHub
parent 5e4fca4ea4
commit 493ba18362

View File

@ -4,15 +4,16 @@ from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import inspect
from jinja2.sandbox import SandboxedEnvironment
from flask import request
from datetime import datetime, timedelta
from dateutil.relativedelta import relativedelta
import inspect
import random
import time
import uuid
import random
from jinja2.sandbox import SandboxedEnvironment
from flask import request, g
from dateutil.relativedelta import relativedelta
from superset import app
@ -40,6 +41,18 @@ def url_param(param, default=None):
return request.args.get(param, default)
def current_user_id():
"""The id of the user who is currently logged in"""
if g.user:
return g.user.id
def current_username():
"""The username of the user who is currently logged in"""
if g.user:
return g.user.username
class BaseTemplateProcessor(object):
"""Base class for database-specific jinja context
@ -67,6 +80,8 @@ class BaseTemplateProcessor(object):
self.schema = table.schema
self.context = {
'url_param': url_param,
'current_user_id': current_user_id,
'current_username': current_username,
}
self.context.update(kwargs)
self.context.update(BASE_CONTEXT)