Add requests functionality to the OGServer class

These functions will be used to make HTTP GET and POST requests to the
OGServer.
multi-ogserver
Roberto Hueso Gómez 2020-09-02 14:42:10 +02:00
parent a4957f2284
commit 46b88fff0b
1 changed files with 19 additions and 0 deletions

View File

@ -1,3 +1,4 @@
import requests
import json
class OGServer:
@ -5,6 +6,7 @@ class OGServer:
self.ip = ip
self.port = port
self.api_token = api_token
self._prepare_requests()
def load_config(self, path):
with open(path, 'r') as f:
@ -13,3 +15,20 @@ class OGServer:
self.ip = cfg['ip']
self.port = cfg['port']
self.api_token = cfg['api_token']
self._prepare_requests()
def _prepare_requests(self):
self.URL = f'http://{self.ip}:{self.port}'
self.HEADERS = {'Authorization' : self.api_token}
def get(self, path, payload=None):
r = requests.get(f'{self.URL}{path}',
headers=self.HEADERS,
json=payload)
return r
def post(self, path, payload):
r = requests.post(f'{self.URL}{path}',
headers=self.HEADERS,
json=payload)
return r