feat(components): gs-sequences-by-location: add button to download th…

…e data (#625) resolves #624
GenSpectrum · Jan 7, 2025 · 865f0a2 · 865f0a2
1 parent 0cdde85
commit 865f0a2
Show file tree

Hide file tree

Showing 6 changed files with 102 additions and 3 deletions.
diff --git a/components/src/preact/map/sequences-by-location.tsx b/components/src/preact/map/sequences-by-location.tsx
@@ -4,9 +4,15 @@ import z from 'zod';
 
 import { SequencesByLocationMap } from './sequences-by-location-map';
 import { SequencesByLocationTable } from './sequences-by-location-table';
-import { type MapLocationData, MapLocationDataType } from '../../query/computeMapLocationData';
+import {
+    type EnhancedLocationsTableData,
+    type MapLocationData,
+    MapLocationDataType,
+} from '../../query/computeMapLocationData';
+import { type AggregateData } from '../../query/queryAggregateData';
 import { querySequencesByLocationData } from '../../query/querySequencesByLocationData';
 import { LapisUrlContext } from '../LapisUrlContext';
+import { CsvDownloadButton } from '../components/csv-download-button';
 import { ErrorBoundary } from '../components/error-boundary';
 import { Fullscreen } from '../components/fullscreen';
 import Info, { InfoComponentCode, InfoHeadline1, InfoParagraph } from '../components/info';
@@ -120,16 +126,27 @@ const SequencesByLocationMapTabs: FunctionComponent<SequencesByLocationMapTabsPr
 
     const tabs = originalComponentProps.views.map((view) => getTab(view));
 
-    return <Tabs tabs={tabs} toolbar={<Toolbar originalComponentProps={originalComponentProps} />} />;
+    return (
+        <Tabs
+            tabs={tabs}
+            toolbar={<Toolbar originalComponentProps={originalComponentProps} tableData={data.tableData} />}
+        />
+    );
 };
 
 type ToolbarProps = {
     originalComponentProps: SequencesByLocationProps;
+    tableData: AggregateData | EnhancedLocationsTableData;
 };
 
-const Toolbar: FunctionComponent<ToolbarProps> = ({ originalComponentProps }) => {
+const Toolbar: FunctionComponent<ToolbarProps> = ({ originalComponentProps, tableData }) => {
     return (
         <div class='flex flex-row'>
+            <CsvDownloadButton
+                className='mx-1 btn btn-xs'
+                getData={() => tableData}
+                filename='sequences_by_location.csv'
+            />
             <SequencesByLocationMapInfo originalComponentProps={originalComponentProps} />
             <Fullscreen />
         </div>

diff --git a/...-of-visualizatio-63683-ion--germany-table-only-should-match-snapshot-1-chromium-linux.txt b/...-of-visualizatio-63683-ion--germany-table-only-should-match-snapshot-1-chromium-linux.txt
@@ -0,0 +1,20 @@
+count,division,proportion
+109799,,0.33325037407543423
+50650,Baden-Wuerttemberg,0.15372755168007673
+50102,North Rhine Westphalia,0.15206431972902673
+28546,Bavaria,0.08663981619465884
+25459,Saxony,0.0772704785433973
+10343,Rheinland-Pfalz,0.03139198552866799
+9832,Schleswig-Holstein,0.029841052085261884
+9633,Hamburg,0.029237068219825846
+6198,Niedersachsen,0.01881151757775154
+4969,Hesse,0.015081386067093805
+4911,Saarland,0.014905350568625011
+4747,Saxony-Anhalt,0.014407595021230488
+3846,Berlin,0.011672974605361798
+3778,Brandenburg,0.011466588158881142
+3238,Thuringia,0.00982763696624064
+1748,Mecklenburg-Vorpommern,0.005305345712473329
+1672,Bremen,0.005074678507583185
+7,Germany,0.000021245663608302804
+1,Baden-Württemberg,0.0000030350948011861152
diff --git a/...d-of-visualizatio-63683-ion--germany-table-only-should-match-snapshot-1-firefox-linux.txt b/...d-of-visualizatio-63683-ion--germany-table-only-should-match-snapshot-1-firefox-linux.txt
@@ -0,0 +1,20 @@
+count,division,proportion
+109799,,0.33325037407543423
+50650,Baden-Wuerttemberg,0.15372755168007673
+50102,North Rhine Westphalia,0.15206431972902673
+28546,Bavaria,0.08663981619465884
+25459,Saxony,0.0772704785433973
+10343,Rheinland-Pfalz,0.03139198552866799
+9832,Schleswig-Holstein,0.029841052085261884
+9633,Hamburg,0.029237068219825846
+6198,Niedersachsen,0.01881151757775154
+4969,Hesse,0.015081386067093805
+4911,Saarland,0.014905350568625011
+4747,Saxony-Anhalt,0.014407595021230488
+3846,Berlin,0.011672974605361798
+3778,Brandenburg,0.011466588158881142
+3238,Thuringia,0.00982763696624064
+1748,Mecklenburg-Vorpommern,0.005305345712473329
+1672,Bremen,0.005074678507583185
+7,Germany,0.000021245663608302804
+1,Baden-Württemberg,0.0000030350948011861152
diff --git a/...-of-visualizatio-da706-es-by-location--germany-should-match-snapshot-1-chromium-linux.txt b/...-of-visualizatio-da706-es-by-location--germany-should-match-snapshot-1-chromium-linux.txt
@@ -0,0 +1,20 @@
+count,division,proportion,isShownOnMap
+109799,,0.33325037407543423,false
+50650,Baden-Wuerttemberg,0.15372755168007673,true
+50102,North Rhine Westphalia,0.15206431972902673,true
+28546,Bavaria,0.08663981619465884,true
+25459,Saxony,0.0772704785433973,true
+10343,Rheinland-Pfalz,0.03139198552866799,true
+9832,Schleswig-Holstein,0.029841052085261884,true
+9633,Hamburg,0.029237068219825846,true
+6198,Niedersachsen,0.01881151757775154,true
+4969,Hesse,0.015081386067093805,true
+4911,Saarland,0.014905350568625011,true
+4747,Saxony-Anhalt,0.014407595021230488,true
+3846,Berlin,0.011672974605361798,true
+3778,Brandenburg,0.011466588158881142,true
+3238,Thuringia,0.00982763696624064,true
+1748,Mecklenburg-Vorpommern,0.005305345712473329,true
+1672,Bremen,0.005074678507583185,true
+7,Germany,0.000021245663608302804,false
+1,Baden-Württemberg,0.0000030350948011861152,false
diff --git a/...d-of-visualizatio-da706-es-by-location--germany-should-match-snapshot-1-firefox-linux.txt b/...d-of-visualizatio-da706-es-by-location--germany-should-match-snapshot-1-firefox-linux.txt
@@ -0,0 +1,20 @@
+count,division,proportion,isShownOnMap
+109799,,0.33325037407543423,false
+50650,Baden-Wuerttemberg,0.15372755168007673,true
+50102,North Rhine Westphalia,0.15206431972902673,true
+28546,Bavaria,0.08663981619465884,true
+25459,Saxony,0.0772704785433973,true
+10343,Rheinland-Pfalz,0.03139198552866799,true
+9832,Schleswig-Holstein,0.029841052085261884,true
+9633,Hamburg,0.029237068219825846,true
+6198,Niedersachsen,0.01881151757775154,true
+4969,Hesse,0.015081386067093805,true
+4911,Saarland,0.014905350568625011,true
+4747,Saxony-Anhalt,0.014407595021230488,true
+3846,Berlin,0.011672974605361798,true
+3778,Brandenburg,0.011466588158881142,true
+3238,Thuringia,0.00982763696624064,true
+1748,Mecklenburg-Vorpommern,0.005305345712473329,true
+1672,Bremen,0.005074678507583185,true
+7,Germany,0.000021245663608302804,false
+1,Baden-Württemberg,0.0000030350948011861152,false
diff --git a/components/tests/visualizationStories.ts b/components/tests/visualizationStories.ts
@@ -127,6 +127,7 @@ export const visualizationStories = [
         id: 'visualization-sequences-by-location--germany',
         title: 'Sequences by location',
         loadingIsDoneIndicator: 'Leaflet',
+        testDownloadWithFilename: 'sequences_by_location.csv',
     },
     {
         id: 'visualization-sequences-by-location--germany-on-table-tab',
@@ -137,5 +138,6 @@ export const visualizationStories = [
         id: 'visualization-sequences-by-location--germany-table-only',
         title: 'Sequences by location',
         loadingIsDoneIndicator: 'division',
+        testDownloadWithFilename: 'sequences_by_location.csv',
     },
 ];