Use regex for keeping the ?

filias · Nov 20, 2023 · 2ffa18f · 2ffa18f
1 parent 8566fae
commit 2ffa18f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mango/mango/views.py b/mango/mango/views.py
@@ -1,3 +1,5 @@
+import re
+
 import requests
 from django.views.generic import TemplateView
 
@@ -14,8 +16,8 @@ def get_joke():
         return get_joke()
 
     # split the joke into question and answer
-    question, answer = joke.split("?")
-    return f"{question}?", answer
+    question, answer = re.split(r'(?<=\?)', joke)
+    return question, answer
 
 
 class IndexView(TemplateView):