mirror of
https://github.com/mbirth/tcl_ota_check.git
synced 2024-11-10 06:16:46 +00:00
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
#!/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
|