56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
import json
|
|
import logging
|
|
import os
|
|
from typing import Dict, Tuple, List
|
|
|
|
import requests
|
|
|
|
logger = logging.getLogger("seeding-api")
|
|
|
|
API_URL = os.getenv("API_URL", "http://localhost:5000/") + "api/"
|
|
|
|
api_override = os.getenv("API_FORCE", "")
|
|
|
|
if api_override != "":
|
|
print(f"API_URL was set to {API_URL} but was overwritten with {api_override}")
|
|
API_URL = api_override
|
|
|
|
API_URL_GET_HEU = API_URL + "getHeuballen"
|
|
API_URL_GET_LOGISTIC_PLAN = API_URL + "getLogisticPlan"
|
|
API_URL_GET_ROBOT_STATE = API_URL + "getRobotState"
|
|
|
|
|
|
class Seeding:
|
|
"""Class used for communicating with seeding api
|
|
"""
|
|
|
|
@staticmethod
|
|
def get_heu() -> Tuple[Dict, int]:
|
|
"""Makes the /api/getHeuballen call to the api.
|
|
|
|
:return: Json Object and status code as returned by the api.
|
|
:rtype: Tuple[Dict, int]
|
|
"""
|
|
res = requests.get(API_URL_GET_HEU)
|
|
result = json.loads(res.content)
|
|
logger.debug(f"Seeding.get_heu = {result}, status code = {res.status_code}")
|
|
return result
|
|
|
|
@staticmethod
|
|
def get_logistic_plan() -> Tuple[Dict, int]:
|
|
"""Makes the /api/getLogisticPlan call to the api.
|
|
|
|
:return: Json Object and status code as returned by the api.
|
|
:rtype: List
|
|
"""
|
|
res = requests.get(API_URL_GET_LOGISTIC_PLAN)
|
|
result = json.loads(res.content)
|
|
logger.debug(f"Seeding.get_logistic_plan = {result}, status code = {res.status_code}")
|
|
return result
|
|
|
|
@staticmethod
|
|
def get_robot_state() -> Tuple[Dict, int]:
|
|
res = requests.get(API_URL_GET_ROBOT_STATE)
|
|
result = json.loads(res.content)
|
|
logger.debug(f"Seeding.get_robot_state {result}, status code = {res.status_code}")
|
|
return result, res.status_code
|