import os
import dotenv
from square.environment import SquareEnvironment

dotenv.load_dotenv()

LEGO_CAT_ID = os.getenv("SQUARE_LEGO_CAT_ID")

if os.getenv("SQUARE_ENV") == "production":
    SQUARE_APP_ID = os.getenv("PROD_SQUARE_APP_ID")
    SQUARE_TOKEN = os.getenv("PROD_SQUARE_TOKEN")
    environment = SquareEnvironment.PRODUCTION
else:
    SQUARE_APP_ID = os.getenv("SANDBOX_SQUARE_APP_ID")
    SQUARE_TOKEN = os.getenv("SANDBOX_SQUARE_TOKEN")
    environment = SquareEnvironment.SANDBOX