quote to handle source

1 files changed, 11 insertions, 3 deletions

diff --git a/quote.py b/quote.py
index 06897d2..56ae0bc 100644
--- a/quote.py
+++ b/quote.py
@@ -4,19 +4,27 @@ These all can be styled in css.
 
 Usage (without the backslashes):
 \"""my quote\"""
+or
+\"""my quote "" source \"""
 """
 from marko import inline
 from marko.helpers import MarkoExtension
 
 
 class QuoteElement(inline.InlineElement):
-    pattern = '"""(.+)"""'
+    pattern = '"""([^"]*)(?:""([^"]*))?"""'
     parse_children = True
 
+    def __init__(self, match):
+        self.quote = match.group(1)
+        self.source = match.group(2) if match.group(2) is not None and not match.group(2).isspace() else None
+
+
 
 class QuoteRenderer:
-    def render_quote_element(self, element):
-        return f'<div class="quote"><p>"</p><p>{self.render_children(element)}</p><p>"</p></div>'
+    def render_quote_element(self, element: QuoteElement):
+        source = f'<p class="quote-source">-- {element.source}</p>' if element.source is not None else ''
+        return f'<div class="quote"><div class="quote-main"><p>"</p><p>{element.quote}</p><p>"</p></div>{source}</div>'
 
 
 Quote = MarkoExtension(