From 895c50eebb481fb01a1e93a5851861e1f6caf06a Mon Sep 17 00:00:00 2001 From: alka Date: Thu, 19 Feb 2026 02:45:26 +0700 Subject: [PATCH] Update: Config configuration --- src/calculation_time_constrains/class.py | 3 ++- src/calculation_time_constrains/service.py | 7 ++++--- src/config.py | 1 + src/utils.py | 4 ++-- 4 files changed, 9 insertions(+), 6 deletions(-) diff --git a/src/calculation_time_constrains/class.py b/src/calculation_time_constrains/class.py index 683d401..fc73344 100644 --- a/src/calculation_time_constrains/class.py +++ b/src/calculation_time_constrains/class.py @@ -3,9 +3,10 @@ import numpy as np import pandas as pd from datetime import timedelta from collections import defaultdict +from src.config import BASE_URL class OptimumCostModel: - def __init__(self, token, last_oh_date, next_oh_date,interval =30, base_url: str = "http://192.168.1.82:8000"): + def __init__(self, token, last_oh_date, next_oh_date,interval =30, base_url: str = f"{BASE_URL}"): self.api_base_url = base_url self.token = token self.last_oh_date = last_oh_date diff --git a/src/calculation_time_constrains/service.py b/src/calculation_time_constrains/service.py index c5f6390..b4b2e7a 100644 --- a/src/calculation_time_constrains/service.py +++ b/src/calculation_time_constrains/service.py @@ -41,6 +41,7 @@ from datetime import datetime, date import asyncio import json # from src.utils import save_to_pastebin +from src.config import BASE_URL, REALIBILITY_SERVICE_API client = httpx.AsyncClient(timeout=300.0) @@ -50,7 +51,7 @@ class OptimumCostModelWithSpareparts: def __init__(self, token: str, last_oh_date: date, next_oh_date: date, sparepart_manager, time_window_months: Optional[int] = None, - base_url: str = "http://192.168.1.82:8000"): + base_url: str = f"{BASE_URL}"): """ Initialize the Optimum Cost Model with sparepart management """ @@ -1152,7 +1153,7 @@ async def get_calculation_by_assetnum( async def get_number_of_failures(location_tag, start_date, end_date, token, max_interval=24): url_prediction = ( - f"http://192.168.1.82:8000/reliability/main/number-of-failures/" + f"{REALIBILITY_SERVICE_API}/main/number-of-failures/" f"{location_tag}/{start_date.strftime('%Y-%m-%d')}/{end_date.strftime('%Y-%m-%d')}" ) @@ -1219,7 +1220,7 @@ async def get_equipment_foh( token: str ): url_mdt = ( - f"http://192.168.1.82:8000/reliability/asset/mdt/{location_tag}" + f"{REALIBILITY_SERVICE_API}/asset/mdt/{location_tag}" ) try: diff --git a/src/config.py b/src/config.py index 2093e2d..663ead8 100644 --- a/src/config.py +++ b/src/config.py @@ -85,6 +85,7 @@ AUTH_SERVICE_API = config("AUTH_SERVICE_API") REALIBILITY_SERVICE_API = config("REALIBILITY_SERVICE_API") RBD_SERVICE_API = config("RBD_SERVICE_API") TEMPORAL_URL = config("TEMPORAL_URL") +BASE_URL = config("BASE_URL") API_KEY = config("API_KEY") diff --git a/src/utils.py b/src/utils.py index ed4a615..8926174 100644 --- a/src/utils.py +++ b/src/utils.py @@ -5,7 +5,7 @@ from typing import Optional import pytz from dateutil.relativedelta import relativedelta -from src.config import TIMEZONE +from src.config import TIMEZONE, REALIBILITY_SERVICE_API def parse_relative_expression(date_str: str) -> Optional[datetime]: @@ -95,7 +95,7 @@ import requests def get_latest_numOfFail(location_tag, token) -> float: today = datetime.today().strftime("%Y-%m-%d") - url_today = f"http://192.168.1.82:8000/reliability/main/number-of-failures/{location_tag}/2016-01-01/{today}" + url_today = f"{REALIBILITY_SERVICE_API}/main/number-of-failures/{location_tag}/2016-01-01/{today}" try: response = requests.get(