Преглед на файлове

Narrow down escaping MarkDown

Foppe Hemminga преди 6 години
родител
ревизия
2f6169d293
променени са 1 файла, в които са добавени 8 реда и са изтрити 1 реда
  1. 8 1
      model.py

+ 8 - 1
model.py

@@ -42,7 +42,14 @@ def create_links(news_string):
 
 
 def escape_markdown(news_string):
-    news_string = news_string.translate(str.maketrans({'_': '\_', '*': '\*', '~': '\~'}))
+    regex_underscore = re.compile(r'_.+_')
+    regex_star = re.compile(r'\*.+\*')
+    regex_tilde = re.compile(r'~.+~')
+    boolean = re.search(regex_underscore, news_string) or \
+        re.search(regex_star, news_string) or \
+        re.search(regex_tilde, news_string)
+    if boolean:
+        news_string = news_string.translate(str.maketrans({'_': '\_', '*': '\*', '~': '\~'}))
     return news_string