understanding commited on
Commit
ca33005
·
verified ·
1 Parent(s): 08bb507

Update bot/config.py

Browse files
Files changed (1) hide show
  1. bot/config.py +21 -21
bot/config.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
 
3
  def _get_int(name: str, default: int) -> int:
@@ -6,27 +7,29 @@ def _get_int(name: str, default: int) -> int:
6
  except Exception:
7
  return default
8
 
9
- class Telegram:
10
- # User account session (phone) recommended
11
- API_ID = _get_int("TELEGRAM_API_ID", 0)
12
- API_HASH = os.environ.get("TELEGRAM_API_HASH", "")
13
 
14
- # Use ONLY one of these:
15
- SESSION_STRING = os.environ.get("SESSION_STRING", "") # ✅ user session string (phone)
16
- BOT_TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN", "") # optional (bot token) if you ever use bot auth
 
17
 
18
- OWNER_ID = _get_int("OWNER_ID", 0)
 
 
19
 
20
- # optional: allowlist via env (space separated ids)
21
- ALLOWED_USER_IDS = [x for x in os.environ.get("ALLOWED_USER_IDS", "").split() if x.strip()]
22
 
23
  class Server:
24
- BIND_ADDRESS = os.environ.get("BIND_ADDRESS", "0.0.0.0")
25
- PORT = _get_int("PORT", 7860)
 
26
 
27
- # Simple logging config for uvicorn + hydrogram
28
  LOGGER_CONFIG_JSON = {
29
  "version": 1,
 
30
  "formatters": {
31
  "default": {
32
  "format": "[%(asctime)s][%(name)s][%(levelname)s] -> %(message)s",
@@ -34,15 +37,12 @@ LOGGER_CONFIG_JSON = {
34
  }
35
  },
36
  "handlers": {
37
- "stream_handler": {
38
- "class": "logging.StreamHandler",
39
- "formatter": "default",
40
- }
41
  },
42
  "loggers": {
43
- "uvicorn": {"level": "INFO", "handlers": ["stream_handler"]},
44
- "uvicorn.error": {"level": "INFO", "handlers": ["stream_handler"]},
45
- "hydrogram": {"level": "INFO", "handlers": ["stream_handler"]},
46
- "bot": {"level": "INFO", "handlers": ["stream_handler"]},
47
  },
48
  }
 
1
+ # PATH: bot/config.py
2
  import os
3
 
4
  def _get_int(name: str, default: int) -> int:
 
7
  except Exception:
8
  return default
9
 
10
+ def _get_str(name: str, default: str = "") -> str:
11
+ return str(os.environ.get(name, default))
 
 
12
 
13
+ class Telegram:
14
+ # REQUIRED
15
+ API_ID: int = _get_int("TELEGRAM_API_ID", 12345)
16
+ API_HASH: str = _get_str("TELEGRAM_API_HASH", "YOUR_API_HASH")
17
 
18
+ # Use either SESSION_STRING (preferred on HF if bot token blocked) OR BOT_TOKEN
19
+ SESSION_STRING: str = _get_str("SESSION_STRING", "")
20
+ BOT_TOKEN: str = _get_str("BOT_TOKEN", "")
21
 
22
+ OWNER_ID: int = _get_int("OWNER_ID", 0) # put your tg id
23
+ BOT_USERNAME: str = _get_str("TELEGRAM_BOT_USERNAME", "BotFather")
24
 
25
  class Server:
26
+ BIND_ADDRESS: str = _get_str("BIND_ADDRESS", "0.0.0.0")
27
+ PORT: int = _get_int("PORT", 7860)
28
+ BASE_URL: str = _get_str("BASE_URL", "http://127.0.0.1:7860")
29
 
 
30
  LOGGER_CONFIG_JSON = {
31
  "version": 1,
32
+ "disable_existing_loggers": False,
33
  "formatters": {
34
  "default": {
35
  "format": "[%(asctime)s][%(name)s][%(levelname)s] -> %(message)s",
 
37
  }
38
  },
39
  "handlers": {
40
+ "stream": {"class": "logging.StreamHandler", "formatter": "default"}
 
 
 
41
  },
42
  "loggers": {
43
+ "uvicorn": {"level": "INFO", "handlers": ["stream"]},
44
+ "uvicorn.error": {"level": "INFO", "handlers": ["stream"]},
45
+ "bot": {"level": "INFO", "handlers": ["stream"]},
46
+ "hydrogram": {"level": "INFO", "handlers": ["stream"]},
47
  },
48
  }