Add chart watermark helper

corporate-travel/main.py

@@ -1,8 +1,13 @@
 from collections import OrderedDict
 import matplotlib.pyplot as plt
 from operator import itemgetter
+from pathlib import Path
 import requests
 import re
+import sys
+
+sys.path.append(str(Path(__file__).resolve().parents[1]))
+from watermark import add_watermark
 # from https://www.businesstravelnews.com/Corporate-Travel-100/2018
 
 s = """Deloitte
@@ -119,4 +124,5 @@
 plt.xlabel('$M')
 plt.title('2017 U.S.-Booked Air Volume')
 plt.gca().get_xaxis().get_major_formatter().set_scientific(False)
+add_watermark()
 plt.show()

food/main.py

@@ -1,4 +1,9 @@
 import matplotlib.pyplot as plt
+from pathlib import Path
+import sys
+
+sys.path.append(str(Path(__file__).resolve().parents[1]))
+from watermark import add_watermark
 
 fruit = {
     'banana': [.45, 7, '🍌'],
@@ -51,4 +56,5 @@
 plt.legend([plt.plot([], [], 'o')[0], plt.plot([], [], 'oC2')[0]],
            ['Fruit', 'Vegetable'],
            labelcolor="markerfacecolor")
+add_watermark()
 plt.show()

pay/main.py

@@ -1,4 +1,9 @@
 import matplotlib.pyplot as plt
+from pathlib import Path
+import sys
+
+sys.path.append(str(Path(__file__).resolve().parents[1]))
+from watermark import add_watermark
 
 map = {
     'Amazon': [4836, 164, 3.8],
@@ -79,4 +84,5 @@
 plt.legend([plt.plot([], [], 'o')[0] for i in range(3)],
            ['West coast', 'Rest of USA', 'Rest of World'],
            labelcolor=['C0', 'C1', 'C2'])
+add_watermark()
 plt.show()

store-sales/main.py

@@ -1,6 +1,11 @@
 from collections import OrderedDict
 import matplotlib.pyplot as plt
 from operator import itemgetter
+from pathlib import Path
+import sys
+
+sys.path.append(str(Path(__file__).resolve().parents[1]))
+from watermark import add_watermark
 # from https://stores.org/stores-top-retailers-2018/
 
 s = """Walmart	$374.80	Bentonville, Ark.	5,328
@@ -114,4 +119,5 @@
 plt.xlabel('$')
 plt.title('Annual Sales per Store')
 plt.gca().get_xaxis().get_major_formatter().set_scientific(False)
+add_watermark()
 plt.show()

watermark.py

@@ -0,0 +1,14 @@
+import matplotlib.pyplot as plt
+
+
+def add_watermark(text="github.com/PatMyron/data"):
+    plt.gcf().text(
+        0.99,
+        0.01,
+        text,
+        ha="right",
+        va="bottom",
+        fontsize=8,
+        color="0.35",
+        alpha=0.5,
+    )