python脚本文件传参并通过token登录后爬取数据实例

from bs4 import BeautifulSoup
import requests
import sys

class Zabbix(object):
    def __init__(self, headers):
        self.session = requests.Session()
        self.headers = headers
    def func(self):
        headers={
            'Host':ip,
            'Upgrade-Insecure-Requests':'1',
            'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/57.0.2987.133 Safari/537.36',
            'cookie':'zbx_sessionid='+token
        }
        response=self.session.get(url='http://'+ip+port+'/zabbix/zabbix.php?action=report.status&ddreset=1',headers=headers,allow_redirects=False)
        sourse = BeautifulSoup(response.text, 'html.parser')
        text=sourse.find('span', {'class': 'green'})
        if not text :
            text=sourse.find('span', {'class': 'red'})
        tq=text.string
        print(tq)
if __name__ == '__main__':
    ip=sys.argv[1]
    if len(sys.argv)==3:
        port=''
        token=sys.argv[2]
    elif len(sys.argv)==4:
        port=':'+sys.argv[2]
        token=sys.argv[3]
    else:
        exit()
    headers = {
        'Host': ip,
        'Origin': ip,
        'Referer':ip+'/zabbix/index.php',
        'Upgrade-Insecure-Requests':'1',
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/49.0.2623.87 Safari/537.36',
    }
    status = Zabbix(headers=headers)
    try:
        status.func()
    except Exception:
        exit()