68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
import requests
|
|
from requests.auth import HTTPDigestAuth
|
|
import argparse
|
|
import sys
|
|
|
|
OK = 0
|
|
WARNING = 1
|
|
CRITICAL = 2
|
|
UNKNOWN = 3
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description="TVHeadend DVB Tuner Check")
|
|
parser.add_argument("--host", default="localhost", help="TVHeadend host")
|
|
parser.add_argument("--port", default="9981", help="TVHeadend port")
|
|
parser.add_argument("--user", required=True, help="API username")
|
|
parser.add_argument("--password", required=True, help="API password")
|
|
parser.add_argument("--timeout", type=int, default=10, help="Request timeout")
|
|
parser.add_argument("--debug", action="store_true", help="Enable debug mode")
|
|
return parser.parse_args()
|
|
|
|
def check_api_status(args):
|
|
base_url = f"http://{args.host}:{args.port}"
|
|
auth = HTTPDigestAuth(args.user, args.password)
|
|
|
|
try:
|
|
# First check basic API accessibility
|
|
status_url = f"{base_url}/api/status"
|
|
if args.debug:
|
|
print(f"DEBUG: Checking API status at {status_url}")
|
|
|
|
response = requests.get(status_url, auth=auth, timeout=args.timeout)
|
|
if response.status_code == 401:
|
|
raise Exception("Authentication failed - check credentials")
|
|
response.raise_for_status()
|
|
|
|
if args.debug:
|
|
print(f"DEBUG: API status response: {response.json()}")
|
|
|
|
return True
|
|
except Exception as e:
|
|
if args.debug:
|
|
print(f"DEBUG: API check failed: {str(e)}")
|
|
raise Exception(f"Cannot access TVHeadend API: {str(e)}")
|
|
|
|
def main():
|
|
args = parse_args()
|
|
|
|
try:
|
|
# Verify API is accessible
|
|
check_api_status(args)
|
|
|
|
# If we got here, API is accessible but we couldn't find specific endpoints
|
|
msg = "TVHeadend API is accessible but required endpoints not found"
|
|
if args.debug:
|
|
print(f"DEBUG: {msg}")
|
|
print("DEBUG: Try checking available endpoints with:")
|
|
print(f"DEBUG: curl -u '{args.user}:{args.password}' http://{args.host}:{args.port}/api/help")
|
|
|
|
print(msg)
|
|
sys.exit(UNKNOWN)
|
|
|
|
except Exception as e:
|
|
print(str(e))
|
|
sys.exit(CRITICAL)
|
|
|
|
if __name__ == "__main__":
|
|
main() |