add basic ability to print png out

app.py

@@ -44,5 +44,5 @@ def calculate_weeks(sex: bool, country_index: int, dob: Tuple[int, int, int]) ->
 if __name__ == '__main__':
     inputs = input_all()
     weeks = calculate_weeks(*inputs)
-    generate_calendar_svg(weeks)
+    generate_calendar_svg(weeks, 'png')
 

requirements.txt

@@ -3,4 +3,5 @@ Pillow>=5.3.0
 python-dateutil
 pycountry>=18.12.8
 requests
-svgwrite
+svgwrite
+wand

utils/calendar.py

@@ -1,8 +1,9 @@
 import svgwrite
+from wand.image import Image
 from utils.constants import CAL_SIZE, CSS_STYLES
 
 
-def generate_calendar_svg(units: int, unit_type: str = 'weeks'):
+def generate_calendar_svg(units: int, file_format: str, unit_type: str = 'weeks'):
     """
     Generate an svg calendar based on the number of weeks
     """
@@ -43,6 +44,14 @@ def group(classname):
     dwg.add(dwg.text(units_label, insert=(360, 580), font_size='5px', fill='grey'))
     dwg.save()
 
+    svg_file = open('calendar.svg', mode='wb')
+    with Image(blob=svg_file.read(), format="svg") as image:
+        png_image = image.make_blob("png")
+    svg_file.close()
+
+    with open('calendar.png', mode='wb') as png_out:
+        png_out.write(png_image)
+
 
 if __name__ == '__main__':
-    generate_calendar_svg(100)
+    generate_calendar_svg(100, 'png')