computer-playbook/roles/health-nginx/files/health-nginx.py

59 lines
2.0 KiB
Python
Raw Normal View History

2023-11-17 09:36:40 +01:00
import os
import requests
import sys
2023-11-17 10:32:15 +01:00
import re
2023-11-17 09:36:40 +01:00
2023-12-12 12:52:58 +01:00
# Set up argument parser
parser = argparse.ArgumentParser(description='Check if domains of nginx configuration files respond with the correct status code.')
parser.add_argument('config_path', help='Path to the nginx configuration directory')
args = parser.parse_args()
# Use the provided argument for the config path
config_path = args.config_path
2023-11-17 09:36:40 +01:00
# Initialize the error counter
error_counter = 0
2023-11-17 10:32:15 +01:00
# Regex pattern to match domain.tld or subdomain.domain.tld
pattern = re.compile(r"^(?:[\w-]+\.)?[\w-]+\.[\w-]+\.conf$")
2023-11-17 09:36:40 +01:00
# Iterate over each file in the configuration directory
for filename in os.listdir(config_path):
2023-11-17 10:32:15 +01:00
if filename.endswith('.conf') and pattern.match(filename):
2023-11-17 09:36:40 +01:00
# Extract the domain and subdomain from the filename
name = filename.replace('.conf', '')
parts = name.split('.')
# Prepare the URL and expected status codes
2023-12-06 12:31:39 +01:00
url = f"https://{name}"
# Default: Expect status code 200 for a domain
expected_statuses = [200]
2023-11-17 09:36:40 +01:00
# Determine expected status codes based on subdomain
2023-12-06 12:31:39 +01:00
if len(parts) == 3:
if parts[0] == 'listmonk':
expected_statuses = [401]
elif parts[0] == 'www':
expected_statuses = [200,301]
elif parts[0] == 's':
expected_statuses = [403]
2023-11-17 09:36:40 +01:00
try:
# Send a HEAD request to get only the response header
response = requests.head(url, allow_redirects=True)
# Check if the status code matches the expected statuses
if response.status_code in expected_statuses:
print(f"{name}: ok")
else:
print(f"{name}: error")
error_counter += 1
except requests.RequestException as e:
# Handle exceptions for requests like connection errors
print(f"{name}: error due to {e}")
error_counter += 1
# Exit the script with the number of errors as the exit code
sys.exit(error_counter)