yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-24 08:58:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

lineage export improvement around selector (#20669)

Browse Source
This commit is contained in:
Ashish Gupta 2025-04-07 18:51:06 +05:30 committed by GitHub
parent b7d43e7ee2
commit e154dbbcc7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
openmetadata-ui/src/main/resources/ui/src/components/Lineage/Lineage.component.tsx
View File

@ -131,6 +131,7 @@ const Lineage = ({
<div <div
className="h-full relative lineage-container" className="h-full relative lineage-container"
data-testid="lineage-container" data-testid="lineage-container"
id="lineage-container" // ID is required for export PNG functionality
ref={reactFlowWrapper}> ref={reactFlowWrapper}>
{entityLineage && ( {entityLineage && (
<> <>

3
openmetadata-ui/src/main/resources/ui/src/constants/Export.constants.ts
View File

@ -15,3 +15,6 @@ export enum ExportTypes {
PNG = 'PNG', PNG = 'PNG',
PDF = 'PDF', PDF = 'PDF',
} }
export const LINEAGE_EXPORT_SELECTOR =
'#lineage-container .react-flow__viewport';

9
openmetadata-ui/src/main/resources/ui/src/context/LineageProvider/LineageProvider.tsx
View File

@ -56,7 +56,10 @@ import {
import LineageNodeRemoveButton from '../../components/Lineage/LineageNodeRemoveButton'; import LineageNodeRemoveButton from '../../components/Lineage/LineageNodeRemoveButton';
import { SourceType } from '../../components/SearchedData/SearchedData.interface'; import { SourceType } from '../../components/SearchedData/SearchedData.interface';
import { ROUTES } from '../../constants/constants'; import { ROUTES } from '../../constants/constants';
import { ExportTypes } from '../../constants/Export.constants'; import {
ExportTypes,
LINEAGE_EXPORT_SELECTOR,
} from '../../constants/Export.constants';
import { import {
ELEMENT_DELETE_STATE, ELEMENT_DELETE_STATE,
ZOOM_VALUE, ZOOM_VALUE,
@ -460,8 +463,8 @@ const LineageProvider = ({ children }: LineageProviderProps) => {
name: decodedFqn, name: decodedFqn,
exportTypes: [ExportTypes.CSV, ExportTypes.PNG], exportTypes: [ExportTypes.CSV, ExportTypes.PNG],
}), }),
documentSelector: '.react-flow__viewport', documentSelector: LINEAGE_EXPORT_SELECTOR,
viewport: getViewportForLineageExport(nodes), viewport: getViewportForLineageExport(nodes, LINEAGE_EXPORT_SELECTOR),
onExport: exportLineageData, onExport: exportLineageData,
}); });
} }

9
openmetadata-ui/src/main/resources/ui/src/utils/EntityLineageUtils.tsx
View File

@ -1717,10 +1717,11 @@ export const getViewportForBoundsReactFlow = (
return { x: translateX, y: translateY, zoom: scale }; return { x: translateX, y: translateY, zoom: scale };
}; };
export const getViewportForLineageExport = (nodes: Node[]): ExportViewport => { export const getViewportForLineageExport = (
const exportElement = document.querySelector( nodes: Node[],
'.react-flow__viewport' documentSelector: string
) as HTMLElement; ): ExportViewport => {
const exportElement = document.querySelector(documentSelector) as HTMLElement;
const imageWidth = exportElement.scrollWidth; const imageWidth = exportElement.scrollWidth;
const imageHeight = exportElement.scrollHeight; const imageHeight = exportElement.scrollHeight;