tcl_ota_check/tcllib/requests/http.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""Base HTTP requests."""

from collections import OrderedDict

import requests


class TimeoutException(Exception):
    """Ignore timeouts."""
    pass


class HttpRequest:
    """Provides all generic features for making HTTP GET requests"""

    def __init__(self, url, timeout=10):
        self.url = url
        self.params = OrderedDict()
        self.timeout = timeout
        self.headers = {}

    def reset_session(self):
        """Reset everything to default."""
        self.sess = requests.Session()
        self.sess.headers.update(self.headers)

    def run(self):
        """Run query."""
        try:
            req = self.sess.get(self.url, params=self.params, timeout=self.timeout)
        except requests.exceptions.Timeout as exc:
            raise TimeoutException(exc)
        return req


class HttpPostRequest(HttpRequest):
    """Provides all generic features for making HTTP POST requests"""

    def run(self):
        """Run query."""
        try:
            req = self.sess.post(self.url, data=self.params, timeout=self.timeout)
        except requests.exceptions.Timeout as exc:
            raise TimeoutException(exc)
        return req