from decouple import config
import yaml
import requests
import json
import pandas as pd
from datetime import datetime

def make_request(url, headers):
    try:
        response = requests.get(url, headers=headers)
        response.raise_for_status()
    except requests.exceptions.HTTPError as errh:
        return "An Http Error occurred:" + repr(errh)
    except requests.exceptions.ConnectionError as errc:
        return "An Error Connecting to the API occurred:" + repr(errc)
    except requests.exceptions.Timeout as errt:
        return "A Timeout Error occurred:" + repr(errt)
    except requests.exceptions.RequestException as err:
        return "An Unknown Error occurred" + repr(err)
    
    return response

def GatherReportingInfo(DTAPIToken, DTENV,friendlyName):
    reportingFolder = config('REPORTING_FOLDER')
    
    env = DTENV
    DTAPIToken = DTAPIToken

    
    headers = {
    'Content-Type': 'application/json',
    'Authorization': 'Api-Token ' + DTAPIToken
    }

    #r = requests.get(DTAPIURL,headers=headers)
    

    df = pd.DataFrame()
    
    now=datetime.now()
    strnow = now.strftime("%Y%m%d_%H%M")

        ##GetRestartRequiredServices
    DTAPIURL= env +  "/api/v1/entity/infrastructure/processes?relativeTime=5mins&actualMonitoringState=OFF&expectedMonitoringState=ON"
    #response = requests.get(DTAPIURL, headers=headers)
    response = make_request(DTAPIURL,headers)
    
    for pgi in response.json():
        if(pgi['monitoringState']['restartRequired'] == True):
            row={'RestartRequired':pgi['monitoringState']['restartRequired'],'ProcessToRestart':pgi['displayName']}

            try:
                for mgmtZone in pgi['managementZones']:
                    row.update({'ManagementZone':mgmtZone['name']})
            except:
                    row.update({'ManagementZone':'NONE'})

            for tag in pgi['tags']:
                try:
                    row.update({ tag['key']:tag['value']})
                except:
                    row.update({ tag['key']:tag['key']})
            df = df.append(row,ignore_index=True)
    del response
    
    df.to_csv(reportingFolder + friendlyName + '_Restart_' + strnow +'.csv')
    

    

with open('./environment.yaml') as file:
    doc = yaml.safe_load(file)

for item, doc in doc.items():
    token = dict(doc[2])
    url = dict(doc[1])
    print("Crawling through: " + item)
    print("Check if token exists in environment...")
    if(config(token.get('env-token-name')) != ""):
        print("Gather data, hold on a minute")
        DTTOKEN = config(token.get('env-token-name'))
        DTURL = url.get('env-url')

        GatherReportingInfo(DTTOKEN,DTURL,item)
        
    else: 
        print("token not found, skipping " + item)