30 lines
860 B
Python
30 lines
860 B
Python
from grocy.request import Request
|
|
from grocy.conf import Configuration
|
|
import logging
|
|
|
|
SCHEMA_URL_TEMPLATE = '{domain}/api/openapi/specification'
|
|
SCHEMA_MODEL_MAP = {
|
|
'products': 'Product',
|
|
'product_details': 'ProductDetailsResponse',
|
|
'stock': 'StockEntry',
|
|
'product_groups': 'ProductGroup',
|
|
'locations': 'Location',
|
|
'quantity_units': 'QuantityUnit',
|
|
'recipe_fulfillment': 'RecipeFulfillmentResponse'
|
|
}
|
|
|
|
|
|
def get_schema(name):
|
|
logger = logging.getLogger('schema')
|
|
try:
|
|
cfg = Configuration()
|
|
cfg.load()
|
|
url = SCHEMA_URL_TEMPLATE.format(domain=cfg.domain)
|
|
request = Request('get', url)
|
|
response = request.send()
|
|
schema_name = SCHEMA_MODEL_MAP[name]
|
|
return response['components']['schemas'][schema_name]
|
|
except Exception as e:
|
|
logger.error(e)
|
|
raise e
|