azils3 commited on
Commit
59edad6
·
verified ·
1 Parent(s): 7b77351

Update dtb/settings.py

Browse files
Files changed (1) hide show
  1. dtb/settings.py +12 -7
dtb/settings.py CHANGED
@@ -153,13 +153,18 @@ CELERY_TASK_DEFAULT_QUEUE = 'default'
153
 
154
 
155
  # -----> TELEGRAM
156
- TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN")
157
- if TELEGRAM_TOKEN is None:
158
- logging.error(
159
- "Please provide TELEGRAM_TOKEN in .env file.\n"
160
- "Example of .env file: https://github.com/ohld/django-telegram-bot/blob/main/.env_example"
161
- )
162
- sys.exit(1)
 
 
 
 
 
163
 
164
  TELEGRAM_LOGS_CHAT_ID = os.getenv("TELEGRAM_LOGS_CHAT_ID", default=None)
165
 
 
153
 
154
 
155
  # -----> TELEGRAM
156
+
157
+ # Add this instead:
158
+
159
+ def get_telegram_token():
160
+ from django.core.exceptions import ImproperlyConfigured
161
+ token = os.getenv("TELEGRAM_TOKEN")
162
+ if not token:
163
+ raise ImproperlyConfigured("TELEGRAM_TOKEN must be set in environment")
164
+ return token
165
+
166
+
167
+ TELEGRAM_TOKEN = get_telegram_token()
168
 
169
  TELEGRAM_LOGS_CHAT_ID = os.getenv("TELEGRAM_LOGS_CHAT_ID", default=None)
170