pt/superset
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2024-09-12 00:29:39 -04:00
Code Issues Packages Projects Releases Wiki Activity

Display the first partition. (#2425)

Browse Source
This commit is contained in:
Bogdan 2017-03-16 15:52:35 -07:00 committed by GitHub
parent 36deb8da71
commit 3d77a12aa9
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
superset/db_engine_specs.py
View File

@ -329,7 +329,7 @@ class PrestoEngineSpec(BaseEngineSpec):
full_table_name = "{}.{}".format(schema_name, table_name)
pql = cls._partition_query(full_table_name)
col_name, latest_part = cls.latest_partition(
table_name, schema_name, database)
table_name, schema_name, database, show_first=True)
return {
'partitions': {
'cols': cols,
@ -423,7 +423,7 @@ class PrestoEngineSpec(BaseEngineSpec):
return df.to_records(index=False)[0][0]
@classmethod
def latest_partition(cls, table_name, schema, database):
def latest_partition(cls, table_name, schema, database, show_first=False):
"""Returns col name and the latest (max) partition value for a table
:param table_name: the name of the table
@ -432,6 +432,9 @@ class PrestoEngineSpec(BaseEngineSpec):
:type schema: str
:param database: database query will be run against
:type database: models.Database
:param show_first: displays the value for the first partitioning key
if there are many partitioning keys
:type show_first: bool
>>> latest_partition('foo_table')
'2018-01-01'
@ -440,7 +443,7 @@ class PrestoEngineSpec(BaseEngineSpec):
if len(indexes[0]['column_names']) < 1:
raise SupersetTemplateException(
"The table should have one partitioned field")
elif len(indexes[0]['column_names']) > 1:
elif not show_first and len(indexes[0]['column_names']) > 1:
raise SupersetTemplateException(
"The table should have a single partitioned field "
"to use this function. You may want to use "