From 6da7452b226e5be630e14a899ca7520ed2a5d983 Mon Sep 17 00:00:00 2001
From: Raphael Boidol <boidolr@users.noreply.github.com>
Date: Mon, 1 Jan 2024 14:02:30 +0100
Subject: [PATCH] Add `svg` export to CLI

---
 test/test_wordcloud_cli.py | 17 +++++++++++++++++
 wordcloud/wordcloud_cli.py | 10 +++++++---
 2 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/test/test_wordcloud_cli.py b/test/test_wordcloud_cli.py
index 5b50bce9a..c8fb3f6be 100644
--- a/test/test_wordcloud_cli.py
+++ b/test/test_wordcloud_cli.py
@@ -153,6 +153,23 @@ def test_cli_writes_to_imagefile(tmpdir, tmp_text_file):
 
     # expecting image to be written to imagefile
     assert tmp_image_file.size() > 0
+    # assert png header
+    assert tmp_image_file.read('rb').startswith(b'\x89PNG')
+
+
+def test_cli_writes_to_svg_imagefile(tmpdir, tmp_text_file):
+    # ensure writing works with all python versions
+    tmp_image_file = tmpdir.join('word_cloud.svg')
+
+    tmp_text_file.write(b'some text')
+
+    args, text, image_file = cli.parse_args(['--text', str(tmp_text_file), '--imagefile', str(tmp_image_file)])
+    cli.main(args, text, image_file)
+
+    # expecting image to be written to imagefile
+    assert tmp_image_file.size() > 0
+    # assert svg header
+    assert tmp_image_file.read().startswith('<svg xmlns=')
 
 
 # capsysbinary should be used here, but it's not supported in python 2.
diff --git a/wordcloud/wordcloud_cli.py b/wordcloud/wordcloud_cli.py
index 4ddd3db4f..cdf298b39 100644
--- a/wordcloud/wordcloud_cli.py
+++ b/wordcloud/wordcloud_cli.py
@@ -88,10 +88,14 @@ def __call__(self, parser, namespace, values, option_string=None):
 def main(args, text, imagefile):
     wordcloud = wc.WordCloud(**args)
     wordcloud.generate(text)
-    image = wordcloud.to_image()
 
     with imagefile:
-        image.save(imagefile, format='png', optimize=True)
+        if imagefile.name.endswith('.svg'):
+            image = wordcloud.to_svg()
+            imagefile.write(image.encode('utf-8'))
+        else:
+            image = wordcloud.to_image()
+            image.save(imagefile, format='png', optimize=True)
 
 
 def make_parser():
@@ -110,7 +114,7 @@ def make_parser():
     parser.add_argument(
         '--imagefile', metavar='file', type=FileType('wb'),
         default='-',
-        help='file the completed PNG image should be written to'
+        help='file the completed image should be written to'
              ' (default: stdout)')
     parser.add_argument(
         '--fontfile', metavar='path', dest='font_path',