瀏覽代碼

Escape markdown in player name

Foppe Hemminga 6 年之前
父節點
當前提交
3c9096c8c4
共有 2 個文件被更改,包括 6 次插入0 次删除
  1. 1 0
      main.py
  2. 5 0
      model.py

+ 1 - 0
main.py

@@ -23,6 +23,7 @@ if __name__ == '__main__':
             is_broadcast = model.is_broadcast(cur, result[0])
             if not is_broadcast:
                 text = model.create_links(result[2])
+                text = model.escape_markdown(text)
                 print(result[1])
                 print(text)
                 view.broadcast(text)

+ 5 - 0
model.py

@@ -39,6 +39,11 @@ def create_links(news_string):
     return return_string
 
 
+def escape_markdown(news_string):
+    news_string = news_string.translate(str.maketrans({'_': '\_', '*': '\*', '~': '\~'}))
+    return news_string
+
+
 def is_broadcast(cur, this_id):
     query = f"""SELECT COUNT(*) FROM human_resource WHERE event = {this_id};"""
     res = database_select(cur, query)