mirror of
https://github.com/apache/superset.git
synced 2026-04-18 23:55:00 +00:00
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# pylint: disable=C,R,W
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
from flask_appbuilder.models.sqla.interface import SQLAInterface
|
|
from flask_babel import gettext as __
|
|
|
|
from superset import appbuilder
|
|
from superset.models.annotations import Annotation, AnnotationLayer
|
|
from .base import DeleteMixin, SupersetModelView
|
|
|
|
|
|
class AnnotationModelView(SupersetModelView, DeleteMixin): # noqa
|
|
datamodel = SQLAInterface(Annotation)
|
|
list_columns = ['layer', 'short_descr', 'start_dttm', 'end_dttm']
|
|
edit_columns = [
|
|
'layer', 'short_descr', 'long_descr', 'start_dttm', 'end_dttm']
|
|
add_columns = edit_columns
|
|
|
|
def pre_add(self, obj):
|
|
if not obj.layer:
|
|
raise Exception('Annotation layer is required.')
|
|
if not obj.start_dttm and not obj.end_dttm:
|
|
raise Exception('Annotation start time or end time is required.')
|
|
elif not obj.start_dttm:
|
|
obj.start_dttm = obj.end_dttm
|
|
elif not obj.end_dttm:
|
|
obj.end_dttm = obj.start_dttm
|
|
elif obj.end_dttm < obj.start_dttm:
|
|
raise Exception('Annotation end time must be no earlier than start time.')
|
|
|
|
def pre_update(self, obj):
|
|
self.pre_add(obj)
|
|
|
|
|
|
class AnnotationLayerModelView(SupersetModelView, DeleteMixin):
|
|
datamodel = SQLAInterface(AnnotationLayer)
|
|
list_columns = ['id', 'name']
|
|
edit_columns = ['name', 'descr']
|
|
add_columns = edit_columns
|
|
|
|
|
|
appbuilder.add_view(
|
|
AnnotationLayerModelView,
|
|
'Annotation Layers',
|
|
label=__('Annotation Layers'),
|
|
icon='fa-comment',
|
|
category='Manage',
|
|
category_label=__('Manage'),
|
|
category_icon='')
|
|
appbuilder.add_view(
|
|
AnnotationModelView,
|
|
'Annotations',
|
|
label=__('Annotations'),
|
|
icon='fa-comments',
|
|
category='Manage',
|
|
category_label=__('Manage'),
|
|
category_icon='')
|