def report_update_layout_end():
data = request_data()
provider = ReportProvider(_write_session)
layout_provider = ReportLayoutProvider(_write_session)
provider.update_layout_end(data['id'], data['layout'],
layout_provider.all())
return provider.detail(data['id'])
def __init__(self,
session: Session,
task: Task,
layout: str,
part: str = 'valid',
name: str = 'img_segment',
max_img_size: Tuple[int, int] = None,
stack_type: str = 'vertical',
main_metric: str = 'dice',
plot_count: int = 0,
colors: List[Tuple] = None):
self.session = session
self.task = task
self.layout = layout
self.part = part
self.name = name or 'img_segment'
self.max_img_size = max_img_size
self.stack_type = stack_type
self.main_metric = main_metric
self.colors = colors
self.plot_count = plot_count
self.dag_provider = DagProvider(session)
self.report_provider = ReportProvider(session)
self.layout_provider = ReportLayoutProvider(session)
self.task_provider = TaskProvider(session)
self.report_img_provider = ReportImgProvider(session)
self.report_task_provider = ReportTasksProvider(session)
self.report_series_provider = ReportSeriesProvider(session)
self.project = self.task_provider.project(task.id).id
self.layout = self.layout_provider.by_name(layout)
self.layout_dict = yaml_load(self.layout.content)
self.create_base()
def report_layouts():
data = request_data()
provider = ReportLayoutProvider(_read_session)
options = PaginatorOptions(**data['paginator'])
res = provider.get(data, options)
return res
def __init__(self,
session: Session,
task: Task,
layout: str,
part: str = 'valid',
name: str = 'img_classify',
max_img_size: Tuple[int, int] = None,
main_metric: str = 'accuracy',
plot_count: int = 0):
self.session = session
self.task = task
self.layout = layout
self.part = part
self.name = name or 'img_classify'
self.max_img_size = max_img_size
self.main_metric = main_metric
self.plot_count = plot_count
self.dag_provider = DagProvider(session)
self.report_provider = ReportProvider(session)
self.layout_provider = ReportLayoutProvider(session)
self.task_provider = TaskProvider(session)
self.report_img_provider = ReportImgProvider(session)
self.report_task_provider = ReportTasksProvider(session)
self.report_series_provider = ReportSeriesProvider(session)
self.project = self.task_provider.project(task.id).id
self.layout = self.layout_provider.by_name(layout)
self.layout_dict = yaml_load(self.layout.content)
def create_providers(self):
self.provider = TaskProvider(self.session)
self.report_provider = ReportProvider(self.session)
self.report_tasks_provider = ReportTasksProvider(self.session)
self.report_layout_provider = ReportLayoutProvider(self.session)
self.project_provider = ProjectProvider(self.session)
self.storage = Storage(self.session)
self.dag_provider = DagProvider(self.session)
def report_add_end():
data = request_data()
provider = ReportProvider(_write_session)
layouts = ReportLayoutProvider(_write_session).all()
layout = layouts[data['layout']]
report = Report(name=data['name'],
project=data['project'],
config=yaml_dump(layout))
provider.add(report)
def create_providers(self):
self.log_info('create_providers')
self.provider = TaskProvider(self.session)
self.report_provider = ReportProvider(self.session)
self.report_tasks_provider = ReportTasksProvider(self.session)
self.report_layout_provider = ReportLayoutProvider(self.session)
self.project_provider = ProjectProvider(self.session)
self.storage = Storage(self.session,
logger=self.logger,
component=self.component)
self.dag_provider = DagProvider(self.session)
def report_layout_edit():
data = request_data()
provider = ReportLayoutProvider(_write_session)
layout = provider.by_name(data['name'])
layout.last_modified = now()
if 'content' in data and data['content'] is not None:
data_loaded = yaml_load(data['content'])
ReportLayoutInfo(data_loaded)
layout.content = data['content']
if 'new_name' in data and data['new_name'] is not None:
layout.name = data['new_name']
provider.commit()
def upgrade(migrate_engine):
folder = os.path.dirname(__file__)
session = Session.create_session(connection_string=migrate_engine.url)
provider = ReportLayoutProvider(session)
try:
files = os.path.join(folder, '002', 'report_layout', '*.yml')
for path in glob(files):
name = str(os.path.basename(path).split('.')[0])
text = open(path).read()
provider.add(ReportLayout(name=name,
content=text,
last_modified=now()),
commit=False)
provider.commit()
except Exception:
provider.rollback()
raise
def report_layout_remove():
data = request_data()
provider = ReportLayoutProvider(_write_session)
provider.remove(data['name'], key_column='name')
def report_layout_add():
data = request_data()
provider = ReportLayoutProvider(_write_session)
layout = ReportLayout(name=data['name'], content='', last_modified=now())
provider.add(layout)
def report_add_start():
return {
'projects': ProjectProvider(_read_session).get()['data'],
'layouts': ReportLayoutProvider(_read_session).get()['data']
}
def downgrade(migrate_engine):
session = Session.create_session(connection_string=migrate_engine.url)
provider = ReportLayoutProvider(session)
provider.session.query(ReportLayout).delete(synchronize_session=False)
provider.session.commit()
