Total Complexity | 6 |
Total Lines | 39 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | # -*- coding: utf-8 -*- |
||
|
|||
2 | |||
3 | import requests |
||
4 | from collections import OrderedDict |
||
5 | |||
6 | class TimeoutException(Exception): |
||
7 | pass |
||
8 | |||
9 | class HttpRequest: |
||
10 | """Provides all generic features for making HTTP GET requests""" |
||
11 | def __init__(self, url, timeout=10): |
||
12 | self.url = url |
||
13 | self.params = OrderedDict() |
||
14 | self.timeout = timeout |
||
15 | self.headers = {} |
||
16 | |||
17 | def reset_session(self): |
||
18 | """Reset everything to default.""" |
||
19 | self.sess = requests.Session() |
||
20 | self.sess.headers.update(self.headers) |
||
21 | |||
22 | def run(self): |
||
23 | """Run query.""" |
||
24 | try: |
||
25 | req = self.sess.get(self.url, params=self.params, timeout=self.timeout) |
||
26 | except requests.exceptions.Timeout as e: |
||
27 | raise TimeoutException(e) |
||
28 | return req |
||
29 | |||
30 | class HttpPostRequest(HttpRequest): |
||
31 | """Provides all generic features for making HTTP POST requests""" |
||
32 | def run(self): |
||
33 | """Run query.""" |
||
34 | try: |
||
35 | req = self.sess.post(self.url, data=self.params, timeout=self.timeout) |
||
36 | except requests.exceptions.Timeout as e: |
||
37 | raise TimeoutException(e) |
||
38 | return req |
||
39 |
The coding style of this project requires that you add a docstring to this code element. Below, you find an example for methods:
If you would like to know more about docstrings, we recommend to read PEP-257: Docstring Conventions.