Support theme

cid/export.py

@@ -69,6 +69,16 @@ def choose_analysis(qs):
     return choices[choice]['AnalysisId']
 
 
+def get_theme(analysis):
+    theme_arn = analysis.get('ThemeArn')
+    if not theme_arn:
+        return None
+    if theme_arn.startswith('arn:aws:quicksight::aws:theme/'):
+        return theme_arn.split('/')[-1]
+    logger.warning(f'Theme {theme_arn} is not standard and is not supported yet. Theme will be ignored.')
+    return None
+
+
 def export_analysis(qs, athena):
     """ Export analysis to yaml resource File
     """
@@ -94,7 +104,7 @@ def export_analysis(qs, athena):
     resources = {}
     resources['dashboards'] = {}
     resources['datasets'] = {}
-
+    theme_id = get_theme(analysis)
 
     cur_helper = CUR(session=athena.session)
     resources_datasets = []
@@ -270,6 +280,8 @@ def export_analysis(qs, athena):
     dashboard_resource['name'] = analysis['Name']
     dashboard_resource['dashboardId'] = dashboard_id
     dashboard_resource['category'] = get_parameters().get('category', 'Custom')
+    if theme_id:
+         dashboard_resource['theme'] = theme_id
 
     dashboard_export_method = None
     if get_parameters().get('template-id'):

cid/helpers/quicksight/__init__.py

@@ -1317,6 +1317,13 @@ def _build_params_for_create_update_dash(self, definition: dict, permissions: bo
             'Name': definition.get('name'),
             'ValidationStrategy': {'Mode': 'LENIENT'},
         }
+        theme = definition.get('theme')
+        if theme:
+            if not theme.startswith('arn:'):
+                theme_arn = 'arn:aws:quicksight::aws:theme/' + theme
+            else:
+                raise NotImplementedError('Only standard themes are supported now.')
+            create_parameters['ThemeArn'] = theme_arn
 
         if definition.get('sourceTemplate'):
             dataset_references = [