Version 2.0.4-1: configuration can be changed via cmdline.

Author: pascalbauermeister

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = umlsequence2
-version = 2.0.3-3
+version = 2.0.4-1
 author = Pascal Bauermeister
 author_email = [email protected]
 description = UML Sequence diagram generator from text input

src/umlsequence2/config.py

@@ -39,7 +39,7 @@
     'TEXT_MARGIN_Y',
 ))
 
-CONFIG = Config(
+_CONFIG = Config(
     ACTIVITY_WIDTH  = 0.20,
 
     ACTOR_ASCENT    = 0.25,
@@ -72,3 +72,10 @@
     TEXT_MARGIN_X   = 0.20,
     TEXT_MARGIN_Y   = 0.15,
 )
+
+def set_config(cfg):
+    global _CONFIG
+    _CONFIG = cfg
+
+def get_config():
+    return _CONFIG

src/umlsequence2/main.py

@@ -23,6 +23,7 @@
 from .converter import convert
 from .uml_builder import UmlBuilder
 from .parser import Parser
+from .config import Config, set_config, get_config
 
 
 def generate_svg(input_fp, output_path, percent_zoom, debug, bgcolor):
@@ -102,9 +103,24 @@ def main():
                         default=False,
                         help='emits debug messages')
 
+    # add config float values, e.g. COLUMN_WIDTH as --COLUMN-WIDTH
+    cfg = get_config()._asdict()
+    conf_keys = [k for k, v in cfg.items() if isinstance(v, float)]
+    for k in conf_keys:
+        parser.add_argument('--' + k.replace('_', '-'),
+                            metavar='FLOAT', type=float,
+                            help=f'change {k} (default {cfg[k]})')
+
     args = parser.parse_args()
     args.format = args.format.lower()
 
+    # parse back config modifiers args
+    conf_args = {k:args.__dict__[k] for k in conf_keys
+                 if args.__dict__[k] is not None}
+    if conf_args:
+        cfg.update(conf_args)
+        set_config(Config(**cfg))
+
     # treat input
     if args.INPUT_FILE is None:
         inp = sys.stdin

src/umlsequence2/svg_renderer.py

@@ -2,7 +2,7 @@
 import svgwrite
 from svgwrite import cm, mm
 
-from .config import CONFIG
+from .config import get_config
 
 
 def cm2px(cm):
@@ -61,7 +61,7 @@ def polyline(self, points, grey=False, filled=False):
         points_px = [(cm2px(x), cm2px(y)) for x, y in points]
         a = dict(points=points_px,
                  fill='white' if filled else 'none',
-                 stroke=CONFIG.COLOR_GREY if grey else 'black')
+                 stroke=get_config().COLOR_GREY if grey else 'black')
         self.add(self.dwg.polyline(**a))
         for xp, yp in points_px:
             self.set_max(xp, yp)
@@ -75,15 +75,15 @@ def polygon(self, points):
 
     def get_text_width(self, text):
         #FIXME
-        return CONFIG.TEXT_CHAR_WIDTH * len(text)
+        return get_config().TEXT_CHAR_WIDTH * len(text)
 
     def text(self, x, y, text, underline=False,
              start=False, middle=False, end=False,
              light=False):
         xp, yp = cm2px(x), cm2px(y)
         a = dict(fill='#444' if light else 'black',
                  insert=(xp, yp),
-                 **CONFIG.TEXT_FONT)
+                 **get_config().TEXT_FONT)
         l = cm2px(self.get_text_width(text))
         if start or not start and not middle and not end:
             a['text_anchor'] = 'start'
@@ -105,7 +105,7 @@ def rect(self, x, y, w, h,
         a = dict(insert=(xp, yp),
                  size=(wp, hp),
                  fill='none' if transparent else 'white',
-                 stroke=CONFIG.COLOR_GREY if grey else 'black',
+                 stroke=get_config().COLOR_GREY if grey else 'black',
                  stroke_width=1)
         self.add(self.dwg.rect(**a))
         self.set_max(xp + wp, yp + hp)
@@ -118,7 +118,7 @@ def line(self, x1, y1, x2, y2,
         x2p, y2p = cm2px(x2), cm2px(y2)
         a = dict(start=(x1p, y1p),
                  end=(x2p, y2p),
-                 stroke=CONFIG.COLOR_GREY if grey else 'black',
+                 stroke=get_config().COLOR_GREY if grey else 'black',
                  stroke_width=2 if thick else 1)
         if dashed:
             a['stroke_dasharray'] = '4'