clean_api.py

import json
import requests
from datetime import datetime
from termcolor import colored
import logging


server = "http://192.168.10.90:8000/"
headers =  { "Content-Type":"application/json", "Connection":"keep-alive" }
cookies = {}
logger = logging.getLogger()


def get_datetime(datetime_str: str):
    try:
        curDt = datetime.strptime(datetime_str, '%Y-%m-%d %H:%M:%S')
    except:
        try:
            curDt = datetime.strptime(datetime_str, '%Y-%m-%dT%H:%M:%S')
        except:
            return None
    return curDt


def post_connect_request(json_data):
    response = requests.post(f'{server}connectDWH/', data=json_data, headers=headers)
    if response.ok:
        global cookies
        cookies = response.cookies
        return True
    else:
        logger.error(f'Could not connect; status = {response.status_code}, info = {response.text}')
        return False


def post_disconnect_request():
    response = requests.post(f'{server}disconnectDWH/', data='', headers=headers)
    if response.ok:
        return True
    else:
        logger.error(f'disconnectDWH failed with status {colored(response.status_code, "light_red")} and info {colored(response.text, "light_red")}')
        return False


def get_request(request_name: str, json_data = {}):
    logger.debug(colored(f"GET /{request_name}/ with body {json.dumps(json_data)}", 'dark_grey'))
    response = requests.get(f'{server}{request_name}/', data=json.dumps(json_data), headers=headers, cookies=cookies)
    if response.ok:
        # logger.debug(response.json())
        return response.json()
    else:
        logger.error(f'{request_name} failed with status {colored(response.status_code, "light_red")} and info {colored(response.text, "light_red")}')
        return None


def post_request(request_name: str, json_data):
    logger.debug(colored(f"POST /{request_name}/ with body {json.dumps(json_data)}", 'dark_grey'))
    response = requests.post(f'{server}{request_name}/', data=json.dumps(json_data), headers=headers, cookies=cookies)
    if response.ok:
        # logger.debug(response.json())
        return response.json()
    else:
        logger.error(f'{request_name} failed with status {colored(response.status_code, "light_red")} and info {colored(response.text, "light_red")}')
        return None


def connect(username: str, password: str) -> bool:
    json_data = f'{{"Username": "{username}", "Password": "{password}"}}'
    return post_connect_request(json_data)