Commit a06eae69 authored by Karagkounis Dimitris's avatar Karagkounis Dimitris
Browse files

update config to properly read variables

parent 04ca70a0
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -5,7 +5,7 @@ from dotenv import load_dotenv
load_dotenv()


class Config(BaseSettings):
class Configuration(BaseSettings):
    MONGO_URI: str = os.getenv("MONGO_URI")
    PI_EDGE_BASE_URL: str = os.getenv("PI_EDGE_BASE_URL")
    PI_EDGE_USERNAME: str = os.getenv("PI_EDGE_USERNAME")
@@ -13,4 +13,4 @@ class Config(BaseSettings):
    HTTP_PROXY: str = os.getenv("HTTP_PROXY")


# config = Config()
config = Configuration()
+3 −3
Original line number Diff line number Diff line
from pymongo import MongoClient
from edge_cloud_management_api.configs.env_config import Config
from edge_cloud_management_api.configs.env_config import config


class MongoManager:
@@ -25,8 +25,8 @@ class MongoManager:
        """
        Initializes the MongoDB connection using the URI from Config.
        """
        self.client = MongoClient(Config.MONGO_URI, maxPoolSize=50)
        mongo_db_name: str = Config.MONGO_URI.split("/")[-1]
        self.client = MongoClient(config.MONGO_URI, maxPoolSize=50)
        mongo_db_name: str = config.MONGO_URI.split("/")[-1]
        self.db = self.client[mongo_db_name]

    def __enter__(self):
+6 −6
Original line number Diff line number Diff line
import requests
from edge_cloud_management_api.managers.log_manager import logger
from requests.exceptions import Timeout, ConnectionError
from edge_cloud_management_api.configs.env_config import Config
from edge_cloud_management_api.configs.env_config import config


proxies = {
    "http": Config.HTTP_PROXY,
    "https": Config.HTTP_PROXY,
    "http": config.HTTP_PROXY,
    "https": config.HTTP_PROXY,
}


@@ -155,9 +155,9 @@ class PiEdgeAPIClientFactory:
    """

    def __init__(self):
        self.default_base_url = Config.PI_EDGE_BASE_URL
        self.default_username = Config.PI_EDGE_USERNAME
        self.default_password = Config.PI_EDGE_PASSWORD
        self.default_base_url = config.PI_EDGE_BASE_URL
        self.default_username = config.PI_EDGE_USERNAME
        self.default_password = config.PI_EDGE_PASSWORD

    def create_pi_edge_api_client(self, base_url=None, username=None, password=None):
        """
+1 −1
Original line number Diff line number Diff line
@@ -19,7 +19,7 @@ def mock_mongo_manager():
    Fixture to provide a MongoManager instance with a mocked MongoDB client.
    """

    with patch("edge_cloud_management_api.managers.db_manager.Config", new=TestConfig):
    with patch("edge_cloud_management_api.managers.db_manager.config", new=TestConfig):
        with patch(
            "edge_cloud_management_api.managers.db_manager.MongoClient",
            new=mongomock.MongoClient,