"""API Client for ows-salessheets.""" import requests class APIClient: """API Client for ows-salessheets.""" def __init__(self, base_url, session=None): """Initialize APIClient class.""" self.base_url = base_url self.session = session def post_single_sales_sheet(self, product_id): """Execute a POST against salessheets/single.""" endpoint = '{}/salessheets/single/{}'.format(self.base_url, product_id) return requests.post(endpoint, headers={'session': self.session}) def single_sales_sheet(self, product_id): """Execute a GET against salessheets/single.""" endpoint = '{}/salessheets/single/{}'.format(self.base_url, product_id) return requests.get(endpoint, headers={'session': self.session}) def simple_get(self, url): """Execute a GET against provided URL.""" return requests.get(url)