import dash import time import uuid from dash import dcc from dash.dependencies import Input, Output, State, ALL from dash.exceptions import PreventUpdate import feffery_antd_components as fac import feffery_utils_components as fuc from server import app from utils.common import validate_data_not_empty from api.role import get_role_list_api, get_role_detail_api, add_role_api, edit_role_api, delete_role_api, export_role_list_api from api.menu import get_menu_tree_api @app.callback( output=dict( role_table_data=Output('role-list-table', 'data', allow_duplicate=True), role_table_pagination=Output('role-list-table', 'pagination', allow_duplicate=True), role_table_key=Output('role-list-table', 'key'), role_table_selectedrowkeys=Output('role-list-table', 'selectedRowKeys'), api_check_token_trigger=Output('api-check-token', 'data', allow_duplicate=True) ), inputs=dict( search_click=Input('role-search', 'nClicks'), refresh_click=Input('role-refresh', 'nClicks'), pagination=Input('role-list-table', 'pagination'), operations=Input('role-operations-store', 'data') ), state=dict( role_name=State('role-role_name-input', 'value'), role_key=State('role-role_key-input', 'value'), status_select=State('role-status-select', 'value'), create_time_range=State('role-create_time-range', 'value'), button_perms=State('role-button-perms-container', 'data') ), prevent_initial_call=True ) def get_role_table_data(search_click, refresh_click, pagination, operations, role_name, role_key, status_select, create_time_range, button_perms): """ 获取角色表格数据回调（进行表格相关增删查改操作后均会触发此回调） """ create_time_start = None create_time_end = None if create_time_range: create_time_start = create_time_range[0] create_time_end = create_time_range[1] query_params = dict( role_name=role_name, role_key=role_key, status=status_select, create_time_start=create_time_start, create_time_end=create_time_end, page_num=1, page_size=10 ) triggered_id = dash.ctx.triggered_id if triggered_id == 'role-list-table': query_params = dict( role_name=role_name, role_key=role_key, status=status_select, create_time_start=create_time_start, create_time_end=create_time_end, page_num=pagination['current'], page_size=pagination['pageSize'] ) if search_click or refresh_click or pagination or operations: table_info = get_role_list_api(query_params) if table_info['code'] == 200: table_data = table_info['data']['rows'] table_pagination = dict( pageSize=table_info['data']['page_size'], current=table_info['data']['page_num'], showSizeChanger=True, pageSizeOptions=[10, 30, 50, 100], showQuickJumper=True, total=table_info['data']['total'] ) for item in table_data: if item['status'] == '0': item['status'] = dict(checked=True, disabled=item['role_id'] == 1) else: item['status'] = dict(checked=False, disabled=item['role_id'] == 1) item['key'] = str(item['role_id']) if item['role_id'] == 1: item['operation'] = [] else: item['operation'] = fac.AntdSpace( [ fac.AntdButton( '修改', id={ 'type': 'role-operation-table', 'operation': 'edit', 'index': str(item['role_id']) }, type='link', icon=fac.AntdIcon( icon='antd-edit' ), style={ 'padding': 0 } ) if 'system:role:edit' in button_perms else [], fac.AntdButton( '删除', id={ 'type': 'role-operation-table', 'operation': 'delete', 'index': str(item['role_id']) }, type='link', icon=fac.AntdIcon( icon='antd-delete' ), style={ 'padding': 0 } ) if 'system:role:remove' in button_perms else [], fac.AntdPopover( fac.AntdButton( '更多', type='link', icon=fac.AntdIcon( icon='antd-more' ), style={ 'padding': 0 } ), content=fac.AntdSpace( [ fac.AntdButton( '数据权限', id={ 'type': 'role-operation-table', 'operation': 'datascope', 'index': str(item['role_id']) }, type='text', block=True, icon=fac.AntdIcon( icon='antd-check-circle' ), style={ 'padding': 0 } ), fac.AntdButton( '分配用户', id={ 'type': 'role-operation-table', 'operation': 'allocation', 'index': str(item['role_id']) }, type='text', block=True, icon=fac.AntdIcon( icon='antd-user' ), style={ 'padding': 0 } ), ], direction='vertical' ), placement='bottomRight' ) if 'system:role:edit' in button_perms else [] ] ) return dict( role_table_data=table_data, role_table_pagination=table_pagination, role_table_key=str(uuid.uuid4()), role_table_selectedrowkeys=None, api_check_token_trigger={'timestamp': time.time()} ) return dict( role_table_data=dash.no_update, role_table_pagination=dash.no_update, role_table_key=dash.no_update, role_table_selectedrowkeys=dash.no_update, api_check_token_trigger={'timestamp': time.time()} ) raise PreventUpdate # 重置角色搜索表单数据回调 app.clientside_callback( ''' (reset_click) => { if (reset_click) { return [null, null, null, null, {'type': 'reset'}] } return window.dash_clientside.no_update; } ''', [Output('role-role_name-input', 'value'), Output('role-role_key-input', 'value'), Output('role-status-select', 'value'), Output('role-create_time-range', 'value'), Output('role-operations-store', 'data')], Input('role-reset', 'nClicks'), prevent_initial_call=True ) # 隐藏/显示角色搜索表单回调 app.clientside_callback( ''' (hidden_click, hidden_status) => { if (hidden_click) { return [ !hidden_status, hidden_status ? '隐藏搜索' : '显示搜索' ] } return window.dash_clientside.no_update; } ''', [Output('role-search-form-container', 'hidden'), Output('role-hidden-tooltip', 'title')], Input('role-hidden', 'nClicks'), State('role-search-form-container', 'hidden'), prevent_initial_call=True ) @app.callback( Output({'type': 'role-operation-button', 'operation': 'edit'}, 'disabled'), Input('role-list-table', 'selectedRowKeys'), prevent_initial_call=True ) def change_role_edit_button_status(table_rows_selected): """ 根据选择的表格数据行数控制编辑按钮状态回调 """ outputs_list = dash.ctx.outputs_list if outputs_list: if table_rows_selected: if len(table_rows_selected) > 1 or '1' in table_rows_selected: return True return False return True return dash.no_update @app.callback( Output({'type': 'role-operation-button', 'operation': 'delete'}, 'disabled'), Input('role-list-table', 'selectedRowKeys'), prevent_initial_call=True ) def change_role_delete_button_status(table_rows_selected): """ 根据选择的表格数据行数控制删除按钮状态回调 """ outputs_list = dash.ctx.outputs_list if outputs_list: if table_rows_selected: if '1' in table_rows_selected: return True return False return True return dash.no_update @app.callback( Output('role-menu-perms', 'expandedKeys', allow_duplicate=True), Input('role-menu-perms-radio-fold-unfold', 'checked'), State('role-menu-store', 'data'), prevent_initial_call=True ) def fold_unfold_role_menu(fold_unfold, menu_info): """ 新增和编辑表单中展开/折叠checkbox回调 """ if menu_info: default_expanded_keys = [] for item in menu_info: if item.get('parent_id') == 0: default_expanded_keys.append(str(item.get('menu_id'))) if fold_unfold: return default_expanded_keys else: return [] return dash.no_update @app.callback( Output('role-menu-perms', 'checkedKeys', allow_duplicate=True), Input('role-menu-perms-radio-all-none', 'checked'), State('role-menu-store', 'data'), prevent_initial_call=True ) def all_none_role_menu_mode(all_none, menu_info): """ 新增和编辑表单中全选/全不选checkbox回调 """ if menu_info: default_expanded_keys = [] for item in menu_info: if item.get('parent_id') == 0: default_expanded_keys.append(str(item.get('menu_id'))) if all_none: return [str(item.get('menu_id')) for item in menu_info] else: return [] return dash.no_update @app.callback( [Output('role-menu-perms', 'checkStrictly'), Output('role-menu-perms', 'checkedKeys', allow_duplicate=True)], Input('role-menu-perms-radio-parent-children', 'checked'), State('current-role-menu-store', 'data'), prevent_initial_call=True ) def change_role_menu_mode(parent_children, current_role_menu): """ 新增和编辑表单中父子联动checkbox回调 """ checked_menu = [] if parent_children: if current_role_menu: for item in current_role_menu: has_children = False for other_item in current_role_menu: if other_item['parent_id'] == item['menu_id']: has_children = True break if not has_children: checked_menu.append(str(item.get('menu_id'))) return [False, checked_menu] else: if current_role_menu: checked_menu = [str(item.get('menu_id')) for item in current_role_menu if item] or [] return [True, checked_menu] @app.callback( output=dict( modal_visible=Output('role-modal', 'visible', allow_duplicate=True), modal_title=Output('role-modal', 'title'), form_value=Output({'type': 'role-form-value', 'index': ALL, 'required': ALL}, 'value'), form_label_validate_status=Output({'type': 'role-form-label', 'index': ALL, 'required': True}, 'validateStatus', allow_duplicate=True), form_label_validate_info=Output({'type': 'role-form-label', 'index': ALL, 'required': True}, 'help', allow_duplicate=True), menu_perms_tree=Output('role-menu-perms', 'treeData'), menu_perms_expandedkeys=Output('role-menu-perms', 'expandedKeys', allow_duplicate=True), menu_perms_checkedkeys=Output('role-menu-perms', 'checkedKeys', allow_duplicate=True), menu_perms_halfcheckedkeys=Output('role-menu-perms', 'halfCheckedKeys', allow_duplicate=True), role_menu=Output('role-menu-store', 'data'), current_role_menu=Output('current-role-menu-store', 'data'), api_check_token_trigger=Output('api-check-token', 'data', allow_duplicate=True), edit_row_info=Output('role-edit-id-store', 'data'), modal_type=Output('role-operations-store-bk', 'data') ), inputs=dict( operation_click=Input({'type': 'role-operation-button', 'operation': ALL}, 'nClicks'), button_click=Input({'type': 'role-operation-table', 'operation': ALL, 'index': ALL}, 'nClicks') ), state=dict( selected_row_keys=State('role-list-table', 'selectedRowKeys') ), prevent_initial_call=True ) def add_edit_role_modal(operation_click, button_click, selected_row_keys): """ 显示新增或编辑角色弹窗回调 """ trigger_id = dash.ctx.triggered_id if trigger_id.operation in ['add', 'edit']: # 获取所有输出表单项对应value的index form_value_list = [x['id']['index'] for x in dash.ctx.outputs_list[2]] # 获取所有输出表单项对应label的index form_label_list = [x['id']['index'] for x in dash.ctx.outputs_list[3]] menu_params = dict(menu_name='', type='role') tree_info = get_menu_tree_api(menu_params) if tree_info.get('code') == 200: tree_data = tree_info['data'] if trigger_id.type == 'role-operation-button' and trigger_id.operation == 'add': role_info = dict(role_name=None, role_key=None, role_sort=None, status='0', remark=None) return dict( modal_visible=True, modal_title='新增角色', form_value=[role_info.get(k) for k in form_value_list], form_label_validate_status=[None] * len(form_label_list), form_label_validate_info=[None] * len(form_label_list), menu_perms_tree=tree_data[0], menu_perms_expandedkeys=[], menu_perms_checkedkeys=None, menu_perms_halfcheckedkeys=None, role_menu=tree_data[1], current_role_menu=None, api_check_token_trigger={'timestamp': time.time()}, edit_row_info=None, modal_type={'type': 'add'} ) elif trigger_id.operation == 'edit': if trigger_id.type == 'role-operation-button': role_id = int(','.join(selected_row_keys)) else: role_id = int(trigger_id.index) role_info_res = get_role_detail_api(role_id=role_id) if role_info_res['code'] == 200: role_info = role_info_res['data'] checked_menu = [] checked_menu_all = [] if role_info.get('menu')[0]: for item in role_info.get('menu'): checked_menu_all.append(str(item.get('menu_id'))) has_children = False for other_item in role_info.get('menu'): if other_item['parent_id'] == item['menu_id']: has_children = True break if not has_children: checked_menu.append(str(item.get('menu_id'))) half_checked_menu = [x for x in checked_menu_all if x not in checked_menu] return dict( modal_visible=True, modal_title='编辑角色', form_value=[role_info.get('role').get(k) for k in form_value_list], form_label_validate_status=[None] * len(form_label_list), form_label_validate_info=[None] * len(form_label_list), menu_perms_tree=tree_data[0], menu_perms_expandedkeys=[], menu_perms_checkedkeys=checked_menu, menu_perms_halfcheckedkeys=half_checked_menu, role_menu=tree_data[1], current_role_menu=role_info.get('menu'), api_check_token_trigger={'timestamp': time.time()}, edit_row_info=role_info.get('role') if role_info else None, modal_type={'type': 'edit'} ) return dict( modal_visible=dash.no_update, modal_title=dash.no_update, form_value=[dash.no_update] * len(form_value_list), form_label_validate_status=[dash.no_update] * len(form_value_list), form_label_validate_info=[dash.no_update] * len(form_value_list), menu_perms_tree=dash.no_update, menu_perms_expandedkeys=dash.no_update, menu_perms_checkedkeys=dash.no_update, menu_perms_halfcheckedkeys=dash.no_update, role_menu=dash.no_update, current_role_menu=dash.no_update, api_check_token_trigger={'timestamp': time.time()}, edit_row_info=None, modal_type=None ) raise PreventUpdate @app.callback( output=dict( form_label_validate_status=Output({'type': 'role-form-label', 'index': ALL, 'required': True}, 'validateStatus', allow_duplicate=True), form_label_validate_info=Output({'type': 'role-form-label', 'index': ALL, 'required': True}, 'help', allow_duplicate=True), modal_visible=Output('role-modal', 'visible'), operations=Output('role-operations-store', 'data', allow_duplicate=True), api_check_token_trigger=Output('api-check-token', 'data', allow_duplicate=True), global_message_container=Output('global-message-container', 'children', allow_duplicate=True) ), inputs=dict( confirm_trigger=Input('role-modal', 'okCounts') ), state=dict( modal_type=State('role-operations-store-bk', 'data'), edit_row_info=State('role-edit-id-store', 'data'), form_value=State({'type': 'role-form-value', 'index': ALL, 'required': ALL}, 'value'), form_label=State({'type': 'role-form-value', 'index': ALL, 'required': True}, 'placeholder'), menu_checked_keys=State('role-menu-perms', 'checkedKeys'), menu_half_checked_keys=State('role-menu-perms', 'halfCheckedKeys'), parent_checked=State('role-menu-perms-radio-parent-children', 'checked') ), prevent_initial_call=True ) def role_confirm(confirm_trigger, modal_type, edit_row_info, form_value, form_label, menu_checked_keys, menu_half_checked_keys, parent_checked): """ 新增或编辑角色弹窗确认回调，实现新增或编辑操作 """ if confirm_trigger: # 获取所有输出表单项对应label的index form_label_output_list = [x['id']['index'] for x in dash.ctx.outputs_list[0]] # 获取所有输入表单项对应的value及label form_value_state = {x['id']['index']: x.get('value') for x in dash.ctx.states_list[2]} form_label_state = {x['id']['index']: x.get('value') for x in dash.ctx.states_list[3]} if all(validate_data_not_empty(item) for item in [form_value_state.get(k) for k in form_label_output_list]): menu_half_checked_keys = menu_half_checked_keys if menu_half_checked_keys else [] menu_checked_keys = menu_checked_keys if menu_checked_keys else [] if parent_checked: menu_perms = menu_half_checked_keys + menu_checked_keys else: menu_perms = menu_checked_keys params_add = form_value_state params_add['menu_id'] = ','.join(menu_perms) if menu_perms else None params_edit = params_add.copy() params_edit['role_id'] = edit_row_info.get('role_id') if edit_row_info else None api_res = {} modal_type = modal_type.get('type') if modal_type == 'add': api_res = add_role_api(params_add) if modal_type == 'edit': api_res = edit_role_api(params_edit) if api_res.get('code') == 200: if modal_type == 'add': return dict( form_label_validate_status=[None] * len(form_label_output_list), form_label_validate_info=[None] * len(form_label_output_list), modal_visible=False, operations={'type': 'add'}, api_check_token_trigger={'timestamp': time.time()}, global_message_container=fuc.FefferyFancyMessage('新增成功', type='success') ) if modal_type == 'edit': return dict( form_label_validate_status=[None] * len(form_label_output_list), form_label_validate_info=[None] * len(form_label_output_list), modal_visible=False, operations={'type': 'edit'}, api_check_token_trigger={'timestamp': time.time()}, global_message_container=fuc.FefferyFancyMessage('编辑成功', type='success') ) return dict( form_label_validate_status=[None] * len(form_label_output_list), form_label_validate_info=[None] * len(form_label_output_list), modal_visible=dash.no_update, operations=dash.no_update, api_check_token_trigger={'timestamp': time.time()}, global_message_container=fuc.FefferyFancyMessage('处理失败', type='error') ) return dict( form_label_validate_status=[None if validate_data_not_empty(form_value_state.get(k)) else 'error' for k in form_label_output_list], form_label_validate_info=[None if validate_data_not_empty(form_value_state.get(k)) else form_label_state.get(k) for k in form_label_output_list], modal_visible=dash.no_update, operations=dash.no_update, api_check_token_trigger=dash.no_update, global_message_container=fuc.FefferyFancyMessage('处理失败', type='error') ) raise PreventUpdate @app.callback( [Output('role-operations-store', 'data', allow_duplicate=True), Output('api-check-token', 'data', allow_duplicate=True), Output('global-message-container', 'children', allow_duplicate=True)], [Input('role-list-table', 'recentlySwitchDataIndex'), Input('role-list-table', 'recentlySwitchStatus'), Input('role-list-table', 'recentlySwitchRow')], prevent_initial_call=True ) def table_switch_role_status(recently_switch_data_index, recently_switch_status, recently_switch_row): """ 表格内切换角色状态回调 """ if recently_switch_data_index: if recently_switch_status: params = dict(role_id=int(recently_switch_row['key']), status='0', type='status') else: params = dict(role_id=int(recently_switch_row['key']), status='1', type='status') edit_button_result = edit_role_api(params) if edit_button_result['code'] == 200: return [ {'type': 'switch-status'}, {'timestamp': time.time()}, fuc.FefferyFancyMessage('修改成功', type='success') ] return [ {'type': 'switch-status'}, {'timestamp': time.time()}, fuc.FefferyFancyMessage('修改失败', type='error') ] raise PreventUpdate @app.callback( [Output('role-delete-text', 'children'), Output('role-delete-confirm-modal', 'visible'), Output('role-delete-ids-store', 'data')], [Input({'type': 'role-operation-button', 'operation': ALL}, 'nClicks'), Input({'type': 'role-operation-table', 'operation': ALL, 'index': ALL}, 'nClicks')], State('role-list-table', 'selectedRowKeys'), prevent_initial_call=True ) def role_delete_modal(operation_click, button_click, selected_row_keys): """ 显示删除角色二次确认弹窗回调 """ trigger_id = dash.ctx.triggered_id if trigger_id.operation == 'delete': if trigger_id.type == 'role-operation-button': role_ids = ','.join(selected_row_keys) else: if trigger_id.type == 'role-operation-table': role_ids = trigger_id.index else: raise PreventUpdate return [ f'是否确认删除角色编号为{role_ids}的角色？', True, {'role_ids': role_ids} ] raise PreventUpdate @app.callback( [Output('role-operations-store', 'data', allow_duplicate=True), Output('api-check-token', 'data', allow_duplicate=True), Output('global-message-container', 'children', allow_duplicate=True)], Input('role-delete-confirm-modal', 'okCounts'), State('role-delete-ids-store', 'data'), prevent_initial_call=True ) def role_delete_confirm(delete_confirm, role_ids_data): """ 删除角色弹窗确认回调，实现删除操作 """ if delete_confirm: params = role_ids_data delete_button_info = delete_role_api(params) if delete_button_info['code'] == 200: return [ {'type': 'delete'}, {'timestamp': time.time()}, fuc.FefferyFancyMessage('删除成功', type='success') ] return [ dash.no_update, {'timestamp': time.time()}, fuc.FefferyFancyMessage('删除失败', type='error') ] raise PreventUpdate @app.callback( [Output('role_to_allocated_user-modal', 'visible'), Output({'type': 'allocate_user-search', 'index': 'allocated'}, 'nClicks'), Output('allocate_user-role_id-container', 'data')], Input({'type': 'role-operation-table', 'operation': ALL, 'index': ALL}, 'nClicks'), State({'type': 'allocate_user-search', 'index': 'allocated'}, 'nClicks'), prevent_initial_call=True ) def role_to_allocated_user_modal(allocated_click, allocated_user_search_nclick): """ 显示角色分配用户弹窗回调 """ trigger_id = dash.ctx.triggered_id if trigger_id.operation == 'allocation': return [ True, allocated_user_search_nclick + 1 if allocated_user_search_nclick else 1, trigger_id.index ] raise PreventUpdate @app.callback( [Output('role-export-container', 'data', allow_duplicate=True), Output('role-export-complete-judge-container', 'data'), Output('api-check-token', 'data', allow_duplicate=True), Output('global-message-container', 'children', allow_duplicate=True)], Input('role-export', 'nClicks'), prevent_initial_call=True ) def export_role_list(export_click): """ 导出角色信息回调 """ if export_click: export_role_res = export_role_list_api({}) if export_role_res.status_code == 200: export_role = export_role_res.content return [ dcc.send_bytes(export_role, f'角色信息_{time.strftime("%Y%m%d%H%M%S", time.localtime())}.xlsx'), {'timestamp': time.time()}, {'timestamp': time.time()}, fuc.FefferyFancyMessage('导出成功', type='success') ] return [ dash.no_update, dash.no_update, {'timestamp': time.time()}, fuc.FefferyFancyMessage('导出失败', type='error') ] raise PreventUpdate @app.callback( Output('role-export-container', 'data', allow_duplicate=True), Input('role-export-complete-judge-container', 'data'), prevent_initial_call=True ) def reset_role_export_status(data): """ 导出完成后重置下载组件数据回调，防止重复下载文件 """ time.sleep(0.5) if data: return None raise PreventUpdate