songs2slides

commit 0c9b8b113cbd3d934c4d5107aa48be3c32fe8a82
parent a581f843a445f279e44a2debb722665153e78838
Author: ashermorgan <59518073+ashermorgan@users.noreply.github.com>
Date:   Tue, 20 Feb 2024 20:28:39 -0800

Implement create_pptx function

Diffstat:
M
requirements.txt
 | 
1
+
M
songs2slides/utils.py
 | 
50
++++++++++++++++++++++++++++++++++++++++++++++++++
M
tests/test_core.py
 | 
8
++++++++

3 files changed, 59 insertions(+), 0 deletions(-)
diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,3 @@
 python-dotenv
+python-pptx
 requests
diff --git a/songs2slides/utils.py b/songs2slides/utils.py
@@ -1,6 +1,10 @@
 from dotenv import load_dotenv
 
 from dataclasses import dataclass
+import pptx
+from pptx.dml.color import RGBColor
+from pptx.enum.text import MSO_ANCHOR, PP_ALIGN
+from pptx.util import Inches, Pt
 import os
 import requests
 
@@ -97,3 +101,49 @@ def get_slide_contents(lyrics: str, lines_per_slide: int = 4):
     if slides[-1] == '': slides = slides[:-1]
 
     return slides
+
+def create_pptx(slide_contents: list[str], filepath: str):
+    """
+    Create a PowerPoint from a list of slide contents
+
+    Parameters
+    ----------
+    slide_contents : list of str
+        The list of slide contents
+    filepath : str
+        The file to save the PowerPoint to
+    """
+
+    # Create presentation
+    prs = pptx.Presentation()
+
+    # Get blank slide template
+    blank_slide_layout = prs.slide_layouts[6]
+
+    # Get textbox size parameters
+    margin = Inches(1)
+    width = prs.slide_width - Inches(2)
+    height = prs.slide_height - Inches(2)
+
+    for slide_content in slide_contents:
+        # Create and format slide
+        slide = prs.slides.add_slide(blank_slide_layout)
+        slide.background.fill.solid()
+        slide.background.fill.fore_color.rgb = RGBColor.from_string('000000')
+
+        # Create and format textbox
+        textbox = slide.shapes.add_textbox(margin, margin, width, height)
+        textbox.text_frame.vertical_anchor = MSO_ANCHOR.MIDDLE
+        textbox.text_frame.word_wrap = True
+
+        # Format paragraph
+        paragraph = textbox.text_frame.paragraphs[0]
+        paragraph.font.color.rgb = RGBColor.from_string('ffffff')
+        paragraph.font.size = Pt(48)
+        paragraph.alignment = PP_ALIGN.CENTER
+
+        # Add slide content
+        paragraph.text = slide_content
+
+    # Save to file
+    prs.save(filepath)
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -128,3 +128,11 @@ class TestUtils(unittest.TestCase):
         result = utils.get_slide_contents(lyrics)
 
         self.assertEqual(result, expected)
+
+    def test_create_pptx(self):
+        with patch('songs2slides.utils.pptx.presentation.Presentation.save') as mocked_save:
+            # Create PowerPoint
+            utils.create_pptx(['A', 'B\nC', 'D'], 'test.pptx')
+
+            # Assert PowerPoint was saved
+            mocked_save.assert_called_with('test.pptx')