Allow custom color of the ID. Must be given in RGB values between 0 and 1.

plotid/plotoptions.py

@@ -53,6 +53,9 @@ class PlotOptions:
         file has to be given.
     fontsize: int, optional
         Fontsize for the displayed ID. Default: 12.
+    fontcolor: tuple, optional
+        Fontcolor for the ID. Must be given as tuple containing three rgb values with
+        each value between [0, 1]. Default: (0, 0, 0).
     """
 
     def __init__(self, figs, rotation, position, **kwargs):
@@ -71,6 +74,7 @@ class PlotOptions:
         if self.font:
             self.font = os.path.abspath(self.font)
         self.fontsize = kwargs.get("fontsize", 12)
+        self.fontcolor = kwargs.get("fontcolor", (0, 0, 0))
 
     def __str__(self):
         """Representation if an object of this class is printed."""

plotid/tagplot.py

@@ -63,6 +63,9 @@ def tagplot(figs, engine, **kwargs):
         file has to be given.
     fontsize: int, optional
         Fontsize for the displayed ID. Default: 12.
+    fontcolor: tuple, optional
+        Fontcolor for the ID. Must be given as tuple containing three rgb values with
+        each value between [0, 1]. Default: (0, 0, 0).
 
     Raises
     ------

plotid/tagplot_image.py

@@ -41,7 +41,7 @@ def tagplot_image(plotid_object):
             raise TypeError("File does not exist.")
             # Check if figs is a valid file is done by pillow internally
 
-    color = (128, 128, 128)  # grey
+    color = tuple(rgb_value * 255 for rgb_value in plotid_object.fontcolor)
     font = ImageFont.load_default()
 
     if plotid_object.font:

plotid/tagplot_matplotlib.py

@@ -39,8 +39,6 @@ def tagplot_matplotlib(plotid_object):
         if not isinstance(figure, matplotlib.figure.Figure):
             raise TypeError("Figure is not a valid matplotlib-figure.")
 
-    color = "grey"
-
     # Loop to create and position the IDs
     for fig in plotid_object.figs:
         fig_id = create_id(plotid_object.id_method)
@@ -63,7 +61,7 @@ def tagplot_matplotlib(plotid_object):
                 wrap=True,
                 rotation=plotid_object.rotation,
                 fontsize=plotid_object.fontsize,
-                color=color,
+                color=plotid_object.fontcolor,
                 font=font,
             )
 

tests/test_plotoptions.py

@@ -46,13 +46,13 @@ class TestTagplot(unittest.TestCase):
             id_method="random",
             id_on_plot=False,
         )
-        print(str(plot_obj))
         self.assertEqual(
             str(plot_obj),
             "<class 'plotid.plotoptions.PlotOptions'>: {'figs': 'FIG', 'figure_ids': "
             "[], 'rotation': 270, 'position': (100, 200), 'prefix': 'xyz', 'id_method':"
             " 'random', 'qrcode': False, 'qr_position': (1, 0), 'qr_size': 100, "
-            "'id_on_plot': False, 'font': False, 'fontsize': 12}",
+            "'id_on_plot': False, 'font': False, 'fontsize': 12, 'fontcolor': "
+            "(0, 0, 0)}",
         )
 
     def test_str_plotidtransfer(self):

tests/test_tagplot_image.py

@@ -48,6 +48,7 @@ class TestTagplotImage(unittest.TestCase):
             qrcode=True,
             fontsize=10,
             font="fonts/xolonium-fonts-4.1/ttf/Xolonium-Bold.ttf",
+            fontcolor=(0, 1, 0),
         )
         options.validate_input()
         figs_and_ids = tagplot_image(options)

tests/test_tagplot_matplotlib.py

@@ -38,6 +38,7 @@ class TestTagplotMatplotlib(unittest.TestCase):
             qrcode=True,
             fontsize=10,
             font="tests/fonts/xolonium-fonts-4.1/otf/Xolonium-Regular.otf",
+            fontcolor=(0, 0, 1),
         )
         options.validate_input()
         figs_and_ids = tagplot_matplotlib(options)