loadbalancer fixes

scripts/loadbalancer/loadbalancer_lib/__init__.py

@@ -0,0 +1,9 @@
+from pathlib import Path
+
+
+if Path('/data/ofm/config').exists():
+    OFM_CONFIG_DIR = Path('/data/ofm/config')
+else:
+    OFM_CONFIG_DIR = Path(__file__).parent.parent.parent.parent / 'config'
+
+assert OFM_CONFIG_DIR.exists()

scripts/loadbalancer/loadbalancer_lib/curl.py

@@ -1,4 +1,5 @@
 from io import BytesIO
+from pathlib import Path
 
 import pycurl
 
@@ -11,7 +12,11 @@ def pycurl_status(url, domain, host_ip):
 
     c = pycurl.Curl()
     c.setopt(c.URL, url)
-    c.setopt(c.CAINFO, '/etc/ssl/certs/ca-certificates.crt')
+
+    # linux needs CA certs specified manually
+    if Path('/etc/ssl/certs/ca-certificates.crt').exists():
+        c.setopt(c.CAINFO, '/etc/ssl/certs/ca-certificates.crt')
+
     c.setopt(c.RESOLVE, [f'{domain}:443:{host_ip}'])
     c.setopt(c.NOBODY, True)
     c.setopt(c.TIMEOUT, 5)
@@ -31,7 +36,11 @@ def pycurl_get(url, domain, host_ip):
     buffer = BytesIO()
     c = pycurl.Curl()
     c.setopt(c.URL, url)
-    c.setopt(c.CAINFO, '/etc/ssl/certs/ca-certificates.crt')
+
+    # linux needs CA certs specified manually
+    if Path('/etc/ssl/certs/ca-certificates.crt').exists():
+        c.setopt(c.CAINFO, '/etc/ssl/certs/ca-certificates.crt')
+
     c.setopt(c.RESOLVE, [f'{domain}:443:{host_ip}'])
     c.setopt(c.WRITEDATA, buffer)
     c.setopt(c.TIMEOUT, 5)

scripts/loadbalancer/loadbalancer_lib/telegram_.py

@@ -2,6 +2,8 @@ import requests
 
 
 def telegram_send_message(message, bot_token, chat_id):
+    print(message)
+
     url = f'https://api.telegram.org/bot{bot_token}/sendMessage'
 
     payload = {'chat_id': chat_id, 'text': message}
@@ -9,6 +11,6 @@ def telegram_send_message(message, bot_token, chat_id):
     response = requests.post(url, data=payload)
 
     if response.status_code == 200:
-        print('Message sent successfully!')
+        print('  Message sent successfully!')
     else:
-        print('Failed to send message:', response.text)
+        print('  Failed to send message:', response.text)