Fix email reports (#9695)

* Fix email reports

* Address comments and inherit from str

Co-authored-by: bogdan kyryliuk <bogdankyryliuk@dropbox.com>
This commit is contained in:
Bogdan
2020-05-01 07:42:58 -07:00
committed by GitHub
parent 9d3da79c4b
commit 858082f975
2 changed files with 5 additions and 5 deletions

View File

@@ -29,17 +29,17 @@ from superset.models.helpers import AuditMixinNullable, ImportMixin
metadata = Model.metadata # pylint: disable=no-member
class ScheduleType(enum.Enum):
class ScheduleType(str, enum.Enum):
slice = "slice"
dashboard = "dashboard"
class EmailDeliveryType(enum.Enum):
class EmailDeliveryType(str, enum.Enum):
attachment = "Attachment"
inline = "Inline"
class SliceEmailReportFormat(enum.Enum):
class SliceEmailReportFormat(str, enum.Enum):
visualization = "Visualization"
data = "Raw data"

View File

@@ -156,7 +156,7 @@ class EmailScheduleView(
class DashboardEmailScheduleView(
EmailScheduleView
): # pylint: disable=too-many-ancestors
schedule_type = ScheduleType.dashboard.value
schedule_type = ScheduleType.dashboard
schedule_type_model = Dashboard
add_title = _("Schedule Email Reports for Dashboards")
@@ -215,7 +215,7 @@ class DashboardEmailScheduleView(
class SliceEmailScheduleView(EmailScheduleView): # pylint: disable=too-many-ancestors
schedule_type = ScheduleType.slice.value
schedule_type = ScheduleType.slice
schedule_type_model = Slice
add_title = _("Schedule Email Reports for Charts")
edit_title = add_title