Superset實(shí)現(xiàn)動態(tài)SQL查詢功能
使用自定義參數(shù)方式實(shí)現(xiàn) superset 實(shí)現(xiàn)SQL動態(tài)查詢
1、啟用參數(shù)：config.py 設(shè)置"ENABLE_TEMPLATE_PROCESSING": True
2、當(dāng)前superset v1.2版本支持的參數(shù)包括：
{{ current_username() }} 當(dāng)前登錄用戶名
{{ current_username(add_to_cache_keys=False) }} 不從緩存中獲取登錄用戶名，默認(rèn)從緩存獲取
{{ current_user_id()}} 當(dāng)前登錄用戶ID
{{ current_user_id(add_to_cache_keys=False) }}不從緩存中獲取登錄用戶ID，默認(rèn)從緩存獲取
{{ url_param('custom_variable') }} url 參數(shù)，比如127.0.0.1：8001\dashboard?abc=123，參數(shù)就是{{ url_param('abc') }} 結(jié)果就是123
{{ cache_key_wrapper() }} 還沒有弄明白啥用
{{ filter_values("字段名") }} 獲取dashboard filter_box組件對某個字段的篩選結(jié)果
{{ from_dttm }} 獲取dashboard filter_box組件日期篩選的開始時(shí)間
{{ to_dttm }} 獲取dashboard filter_box組件日期篩選的結(jié)束時(shí)間
{{ get_filters() }} 暫時(shí)沒有弄明白
除此之外，還可以自定義參數(shù)，自定義參數(shù)方法：
①修改superset/jinja_context.py文件，修改三個地方：
regex = re.compile(
r"\{\{.*("
r"current_user_id\(.*\)|"
r"current_username\(.*\)|"
r"current_userroles\(.*\)|"
r"isadmin\(.*\)|"
r"cache_key_wrapper\(.*\)|"
r"url_param\(.*\)"
r").*\}\}"
)
↑↑↑↑注意此處的current_userroles 和isadmin 是我自定義的，源文件沒有
def current_user_id(self, add_to_cache_keys: bool = True) -> Optional[int]:
"""
Return the user ID of the user who is currently logged in.
:param add_to_cache_keys: Whether the value should be included in the cache key
:returns: The user ID
"""
if hasattr(g, "user") and g.user:
if add_to_cache_keys:
self.cache_key_wrapper(g.user.get_id())
return g.user.get_id()
return None
def current_username(self, add_to_cache_keys: bool = True) -> Optional[str]:
"""
Return the username of the user who is currently logged in.
:param add_to_cache_keys: Whether the value should be included in the cache key
:returns: The username
"""
if g.user and hasattr(g.user, "username"):
if add_to_cache_keys:
self.cache_key_wrapper(g.user.username)
return g.user.username
return None
def current_userroles(self, add_to_cache_keys: bool = True) -> Optional[str]:
"""
Return the roles of the user who is currently logged in.
:param add_to_cache_keys: Whether the value should be included in the cache key
:returns: The userroles
"""
if g.user and hasattr(g.user, "roles"):
if add_to_cache_keys:
user_roles = "/".join([role.name.lower() for role in list(g.user.roles)])
self.cache_key_wrapper(user_roles)
print(user_roles)
return user_roles
"""admin in user_roles"""
return None
def isadmin(self, add_to_cache_keys: bool = True) -> Optional[str]:
"""
Return the roles of the user who is currently logged in.
:param add_to_cache_keys: Whether the value should be included in the cache key
:returns: The userroles
"""
if g.user and hasattr(g.user, "roles"):
if add_to_cache_keys:
user_roles = [role.name.lower() for role in list(g.user.roles)]
return "admin" in user_roles
return None
↑↑↑↑仿照系統(tǒng)自帶的current_username 編造自己的函數(shù)，我寫了current_userroles 和isadmin
class JinjaTemplateProcessor(BaseTemplateProcessor):
def set_context(self, **kwargs: Any) -> None:
super().set_context(**kwargs)
extra_cache = ExtraCache(self._extra_cache_keys)
self._context.update(
{
"url_param": partial(safe_proxy, extra_cache.url_param),
"current_user_id": partial(safe_proxy, extra_cache.current_user_id),
"current_username": partial(safe_proxy, extra_cache.current_username),
"current_userroles": partial(safe_proxy, extra_cache.current_userroles),
"isadmin": partial(safe_proxy, extra_cache.isadmin),
"cache_key_wrapper": partial(safe_proxy, extra_cache.cache_key_wrapper),
"filter_values": partial(safe_proxy, filter_values),
}
)
↑↑↑↑仿照系統(tǒng)自帶的current_username 編造自己的函數(shù)，我寫了current_userroles 和isadmin
就是這3個地方，但是注意，自己在第二步早的函數(shù)，返回值必須是：
ALLOWED_TYPES = (
NONE_TYPE,
"bool",
"str",
"unicode",
"i
否則會提示錯誤，或者自己修改這個types，我是轉(zhuǎn)換，比如上面那個g.user.roles 返回的結(jié)果就不是上面類型，導(dǎo)致我一直不成功，最后修改了下，才可以
3、判斷是否自定義成功：
在superset sql lab中執(zhí)行如下代碼，如果能被解析，就說明成功
4、應(yīng)用案例：
在dataset里面，動態(tài)訪問數(shù)據(jù)源，數(shù)據(jù)源添加where語句：select * from sales where salesname =' {{current_username()}}'
dashboard里面，通過獲取篩選器的結(jié)果，然后獲取其他表應(yīng)當(dāng)顯示的數(shù)據(jù)范圍：
select DATE,risktype,sum(num) as num from
(SELECT date , customerid,product,risktype ,count(*) as num
from v_superset_forecast_risk group by date , customerid,product,risktype ) a
join
(select distinct customer_code,product from v_superset_access
where name='{{ current_username() }}' )access
on a.customerid=access.customer_code
and a.product=access.product
and DATE_FORMAT(date,'%Y-%m')> DATE_FORMAT(date_sub(STR_TO_DATE(concat( {{ "'" + "', '".join(filter_values('yearmonthend')) + "'" }},'-01'), '%Y-%m-%d'), interval 12 month),'%Y-%m')
and DATE_FORMAT(date,'%Y-%m')<={{ "'" + "', '".join(filter_values('yearmonthend')) + "'" }}
group by DATE,risktype
因?yàn)閟ql里面可以使用jinja 表達(dá)式，比如判斷篩選當(dāng)前沒有篩選的時(shí)候，獲取什么數(shù)據(jù)
注意{% %} 內(nèi)部使用參數(shù)的時(shí)候，不需要加{{}}，否則報(bào)錯
通過篩選器實(shí)現(xiàn)模糊查詢
5、官方參考文檔：
https://superset.apache.org/docs/installation/sql-templating
官方?jīng)]有那么詳細(xì)，但是里面有一些我這里可能也沒有消化吸收掉，可以參考看下
總之，通過上面的自定義參數(shù)方法，和jinja表達(dá)式在sql中的應(yīng)用，可以實(shí)現(xiàn)動態(tài)查詢，解決一些無法通過頁面直接交互查詢結(jié)果顯示的內(nèi)容
另外如果你有其他應(yīng)用或者自定義上的思考，歡迎留言，相互學(xué)習(xí)
到此這篇關(guān)于Superset實(shí)現(xiàn)動態(tài)SQL查詢的文章就介紹到這了,更多相關(guān)Superset動態(tài)SQL查詢內(nèi)容請搜索本站以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持本站！
版權(quán)聲明：本站文章來源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非www.sddonglingsh.com所屬的服務(wù)器上建立鏡像，否則將依法追究法律責(zé)任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來，僅供學(xué)習(xí)參考，不代表本站立場，如有內(nèi)容涉嫌侵權(quán)，請聯(lián)系alex-e#qq.com處理。
關(guān)注官方微信