Advertisement
Typhoon

Zabbix - Python Custom URL Parser

Aug 19th, 2015
291
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.25 KB | None | 0 0
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. import sys
  4. from bs4 import BeautifulSoup
  5. from requests.auth import HTTPBasicAuth
  6. import requests
  7.  
  8.  
  9. def get_A():
  10.     db_url = "http://www.SERVER.sk/abc/Find&Fields=&ResultCount=1&datex=&srt1=score"
  11.     checkpage = requests.get(db_url, auth=HTTPBasicAuth('username', 'password'))
  12.     checksoup = BeautifulSoup(checkpage.content, "html.parser")
  13.     try:
  14.         query_result = checksoup.find_all('td')[3].text.strip()
  15.         return query_result
  16.     except:
  17.         return "ERROR : SERVER DB Finding"
  18.  
  19. def get_B():
  20.     try:
  21.         site_status = requests.get('http://www.SERVER.sk/abc/findsometext.php').status_code
  22.         if site_status == 401:
  23.             return site_status
  24.         else :
  25.             return "ERROR : SERVER LOGIN"
  26.     except:
  27.         return "ERROR : SERVER LOGIN"
  28.  
  29.  
  30. def main(argv):
  31.     if not argv:
  32.         print "!ERROR! Use Script with arguments like : SCRIPT.PY -a"
  33.     for arg in argv:
  34.         if arg == '-a':
  35.             print get_A()
  36.             exit(1)
  37.         elif arg == '-b':
  38.             print get_B()
  39.             exit(1)
  40.         else:
  41.             print "!ERROR! Use Script with arguments like : SCRIPT.PY -a"
  42.  
  43.  
  44. if __name__ == "__main__":
  45.     main(sys.argv[1:])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement