Yaml configuration

README.md

@@ -9,15 +9,14 @@
 A simple Python script to verify if a service is up. Whenever the service falls, a message will be sent to a user/group/channel with Telegram
 
 ## How to use?
-- Just set two env variables:
-- `QDBotToken`, your bot token
-- `QDBotIDs`, the ID(s) the bot will use to communicate any downtime. It's possible to set multiple IDs, semicolon separated without any space
-
-### Example in bash
-```bash
-export QDBotToken="123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11" # Your bot token
-export QDBotIDs="10000000" # Single ID
-export QDBotIDs="10000000;10000001;10000002" # Multiple IDs
+**Make a copy** of the file `config/settings.yaml.dist` in the same directory and rename it to `settings.yaml`:
+- Set `QDBotToken`, your bot token
+- Set `QDBotIDs`, the ID(s) the bot will use to communicate any downtime. It's possible to set multiple IDs, semicolon separated without any space
+
+### Example
+```yaml
+QDBotToken: "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11" # Your bot token
+QDBotIDs: ["10000000", "10000001", "10000002"] # Single ID or Multiple IDs
 ```
 
 ### Run it every 5 minutes using crontab

config/settings.yaml.dist

@@ -0,0 +1,2 @@
+QDBotToken: ""
+QDBotIDs: []

requirements.txt

@@ -1,2 +1,3 @@
 httpx == 0.23.0
 httpx[http2] == 0.23.0
+pyyaml == 6.0.1

src/main.py

@@ -8,10 +8,14 @@
 from urllib import parse
 from time import sleep
 from json import load
+import yaml
+
+with open('config/settings.yaml', 'r', encoding='utf-8') as yaml_config:
+ config_map = yaml.load(yaml_config, Loader=yaml.SafeLoader)
 
 
 def get_users() -> list[str]:
- return getenv('QDBotIDs').split(';')
+ return config_map["QDBotIDs"]
 
 
 async def check_ok(url: str) -> bool:
@@ -31,7 +35,7 @@ def check_ping(host: str) -> bool:
 
 async def make_request_to_telegram(service_name: str, method_used: str, chat_id: str) -> list:
  message = f'⚠️ The service {service_name} contacted via {method_used} results offline!'
- url = f'https://api.telegram.org/bot{getenv("QDBotToken")}/sendMessage?chat_id={chat_id}&text={message}'
+ url = f'https://api.telegram.org/bot{config_map["QDBotToken"]}/sendMessage?chat_id={chat_id}&text={message}'
 
  async with AsyncClient(http2=True) as client:
  res = await client.post(url)