Add Flask CSRF protect

ruchernchong · Jul 6, 2024 · 56ed8be · 56ed8be
1 parent fba3f77
commit 56ed8be
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -12,6 +12,7 @@ dill==0.3.8
 distlib==0.3.8
 filelock==3.14.0
 Flask==3.0.3
+Flask-WTF==1.2.1
 functions-framework==3.7.0
 grpcio==1.60.1
 gunicorn==22.0.0
@@ -44,3 +45,4 @@ sniffio==1.3.1
 virtualenv==20.26.2
 watchdog==4.0.1
 Werkzeug==3.0.3
+WTForms==3.1.2
diff --git a/server.py b/server.py
@@ -5,6 +5,7 @@
 
 from dotenv import load_dotenv
 from flask import Flask, request
+from flask_wtf.csrf import CSRFProtect
 from telegram import Bot
 
 from main import webhook
@@ -17,6 +18,7 @@
 NGROK_STATIC_DOMAIN = os.getenv("NGROK_STATIC_DOMAIN")
 
 app = Flask(__name__)
+csrf = CSRFProtect(app)
 
 
 def start_ngrok():