sysadmin/openfreemap
1
0
Fork 0
mirror of https://github.com/hyperknot/openfreemap.git synced 2026-05-21 14:02:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

load balancer in fix mode

Browse Source
This commit is contained in:
Zsolt Ero
2024-06-22 08:43:55 +02:00
parent 8b3bd33766
commit 62bf87115e
6 changed files with 30 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
scripts/loadbalancer/loadbalancer.py
View File

@@ -76,7 +76,15 @@ def check_or_fix(fix=False):
if not working_hosts:
working_hosts = set(c['http_host_list'])
update_records(c, working_hosts)
message = 'OFM loadbalancer FIX found no working hosts, reverting to full list!'
print(message)
telegram_send_message(message, c['telegram_token'], c['telegram_chat_id'])
updated = update_records(c, working_hosts)
if updated:
message = f'OFM loadbalancer FIX modified records, new records: {working_hosts}'
print(message)
telegram_send_message(message, c['telegram_token'], c['telegram_chat_id'])
def run_area(c, area):
@@ -117,14 +125,16 @@ def get_target_version(area):
return response.text.strip()
def update_records(c, working_hosts):
def update_records(c, working_hosts) -> bool:
config = dotenv_values('/data/ofm/config/cloudflare.ini')
cloudflare_api_token = config['dns_cloudflare_api_token']
domain = '.'.join(c['domain_ledns'].split('.')[-2:])
zone_id = get_zone_id(domain, cloudflare_api_token=cloudflare_api_token)
set_records_round_robin(
updated = False
updated |= set_records_round_robin(
zone_id=zone_id,
name=c['domain_ledns'],
host_ip_set=working_hosts,
@@ -134,7 +144,7 @@ def update_records(c, working_hosts):
cloudflare_api_token=cloudflare_api_token,
)
set_records_round_robin(
updated |= set_records_round_robin(
zone_id=zone_id,
name=c['domain_cf'],
host_ip_set=working_hosts,
@@ -143,6 +153,8 @@ def update_records(c, working_hosts):
cloudflare_api_token=cloudflare_api_token,
)
return updated
if __name__ == '__main__':
cli()