history.py

#!/usr/bin/env python
#!-*- coding:utf8 -*-

import requests
import time
import json
import sys
import os.path
import yaml
import arrow

def login(name, password):
    url = "http://owl.fastweb.com.cn/api/v1/auth/login"
    payload = {
        'name': name,
        'password': password
    }
    r = requests.post(url, payload)
    out = json.loads(r.text)
    return out["data"]["sig"]

def hostgroup2hostnames(user, sig, hostgroup):
    url = 'http://owl.fastweb.com.cn/api/v1/hostgroup/hosts'
    payload = {
        'cName': user,
        'cSig': sig,
        'hostgroups': hostgroup
    }

    r = requests.post(url, payload)
    out = json.loads(r.text)
    res = []
    for i in out["data"]["hosts"]:
        res.append(i["hostname"])
    return res

def history(user, sig, startTs, endTs, platform, hostnameFilters):
    url = "http://owl.fastweb.com.cn/api/v2/portal/eventcases/get"
    if startTs >= endTs:
        print " start timestamp bigger than end timestamp"
        exit(1)
    
    payload = {
        "status": "PROBLEM,OK",
        "process": "ALL",
        "includeEvents": True,
        "limit": 20000,
        "startTime": startTs,
        "endTime":   endTs,
        "cName": user,
        "cSig": sig
    } 

    r = requests.post(url, payload)
    out = json.loads(r.text)

    #print json.dumps(out["data"]["eventCases"])
    result = []
    for i in out["data"]["eventCases"]:
        if i["hostname"] in hostnameFilters:
            item = {
                "hostname": i["hostname"],
                "ip": i["ip"],
                "content": i["content"],
                "metric": i["metric"],
                "severity": i["severity"],
                "status": i["status"],
                "process": i["process"],
                "timeStart": i["timeStart"]
            }
            result.append(item)

    final = {
        "platform": platform,
        "count": len(result),
        "timestamp": [startTs, endTs],
        "result": result
    }
    print json.dumps(final)

def readConf():
    if os.path.isfile("secret.yaml"):
        pass
    else:
        return {}

    with open("secret.yaml", 'r') as stream:
        try:
            return yaml.load(stream)
        except yaml.YAMLError as exc:
            print(exc)


if __name__ == "__main__":
    if len(sys.argv) != 4:
        print "usage:    ./history.py [time Start in ISO8601] [time End in ISO8601] [platformName]"
        print "example:  ./history.py 2017-02-28T06:01:00+0800 2017-03-28T06:01:00+0800 c01.i07"
        sys.exit(1) 
    #ts = int(time.time()) # input
    #platform = "c06.i06"  # input
    sTime = arrow.get(sys.argv[1])
    eTime = arrow.get(sys.argv[2])
    startTs = sTime.timestamp
    endTs = eTime.timestamp

    platform = sys.argv[3]     # input
    user = ""      # data
    password = ""    # data
    conf = readConf()
    if len(conf) == 2:
        user = conf["user"]
        password = conf["pass"]
    sig = login(user, password) 
    hostnameFilters = hostgroup2hostnames(user, sig, platform)
    history(user, sig, startTs, endTs, platform, hostnameFilters)