diff --git a/src/wireviz/DataClasses.py b/src/wireviz/DataClasses.py index c01de1a..4e4a2c8 100644 --- a/src/wireviz/DataClasses.py +++ b/src/wireviz/DataClasses.py @@ -42,6 +42,7 @@ class Metadata: @dataclass class Options: fontname: PlainText = 'arial' + bgcolor: Color = 'WH' color_mode: ColorMode = 'SHORT' mini_bom_mode: bool = True diff --git a/src/wireviz/Harness.py b/src/wireviz/Harness.py index 95cb530..9e50c03 100644 --- a/src/wireviz/Harness.py +++ b/src/wireviz/Harness.py @@ -95,7 +95,7 @@ class Harness: dot.body.append(f'// {APP_URL}') dot.attr('graph', rankdir='LR', ranksep='2', - bgcolor='white', + bgcolor=wv_colors.translate_color(self.options.bgcolor, "HEX"), nodesep='0.33', fontname=self.options.fontname) dot.attr('node', shape='record', diff --git a/src/wireviz/wv_html.py b/src/wireviz/wv_html.py index fe8d7b5..94c8d62 100644 --- a/src/wireviz/wv_html.py +++ b/src/wireviz/wv_html.py @@ -5,7 +5,7 @@ from pathlib import Path from typing import List, Union import re -from wireviz import __version__, APP_NAME, APP_URL +from wireviz import __version__, APP_NAME, APP_URL, wv_colors from wireviz.DataClasses import Metadata, Options from wireviz.wv_helper import flatten2d, open_file_read, open_file_write @@ -16,7 +16,8 @@ def generate_html_output(filename: Union[str, Path], bom_list: List[List[str]], file.write(' \n') file.write(f' \n') file.write(f'