Hermes Agent commited on
Commit
b8eee28
·
1 Parent(s): fe53333

fix: support PAYMENT_ENABLED env var and pass file_unique_id in private_receive_handler

Browse files
Files changed (2) hide show
  1. Thunder/bot/plugins/stream.py +6 -1
  2. Thunder/vars.py +1 -1
Thunder/bot/plugins/stream.py CHANGED
@@ -323,7 +323,12 @@ async def link_handler(bot: Client, msg: Message, **kwargs):
323
  )
324
  async def private_receive_handler(bot: Client, msg: Message, **kwargs):
325
  async def _actual_private_receive_handler(client: Client, message: Message, **handler_kwargs):
326
- shortener_val = await validate_request_common(client, message)
 
 
 
 
 
327
  if shortener_val is None:
328
  return
329
  if not message.from_user:
 
323
  )
324
  async def private_receive_handler(bot: Client, msg: Message, **kwargs):
325
  async def _actual_private_receive_handler(client: Client, message: Message, **handler_kwargs):
326
+ media = (
327
+ message.document or message.video or message.photo or
328
+ message.audio or message.voice or message.animation or message.video_note
329
+ )
330
+ file_unique_id = getattr(media, "file_unique_id", str(message.id)) if media else str(message.id)
331
+ shortener_val = await validate_request_common(client, message, file_identifier=file_unique_id)
332
  if shortener_val is None:
333
  return
334
  if not message.from_user:
Thunder/vars.py CHANGED
@@ -112,7 +112,7 @@ class Var:
112
  MAX_QUEUE_SIZE: int = int(os.getenv("MAX_QUEUE_SIZE", "100"))
113
 
114
  # Subscription & Payment Gateway (Dormant by default)
115
- PAYMENT_ENABLED: bool = str_to_bool(os.getenv("PAYMENT", os.getenv("Payment", "False")))
116
  SUB_PRICE: int = int(os.getenv("SUB_PRICE", "10"))
117
  SUB_DURATION_DAYS: int = int(os.getenv("SUB_DURATION_DAYS", "30"))
118
  UPI_ID: str = os.getenv("UPI_ID", "jishnupg2005-2@okaxis").strip()
 
112
  MAX_QUEUE_SIZE: int = int(os.getenv("MAX_QUEUE_SIZE", "100"))
113
 
114
  # Subscription & Payment Gateway (Dormant by default)
115
+ PAYMENT_ENABLED: bool = str_to_bool(os.getenv("PAYMENT_ENABLED", os.getenv("PAYMENT", os.getenv("Payment", os.getenv("payment", "False")))))
116
  SUB_PRICE: int = int(os.getenv("SUB_PRICE", "10"))
117
  SUB_DURATION_DAYS: int = int(os.getenv("SUB_DURATION_DAYS", "30"))
118
  UPI_ID: str = os.getenv("UPI_ID", "jishnupg2005-2@okaxis").strip()