Add salary detection configuration and payloads
Updated config.py to include the salary detection endpoint, headers, and sample payloads. Enhanced salary_detect.py to utilize the new configuration for making POST requests to the endpoint with random salary data. Improved error handling during the request process.
This commit is contained in:
@@ -4,6 +4,7 @@ Configuration settings for the salary analytics package.
|
|||||||
|
|
||||||
import os
|
import os
|
||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
|
import random
|
||||||
|
|
||||||
# Load environment variables
|
# Load environment variables
|
||||||
load_dotenv()
|
load_dotenv()
|
||||||
@@ -76,4 +77,21 @@ OUTPUT_PATHS = {
|
|||||||
"inconsistent_model": os.path.join(MODEL_DIR, "inconsistent_model.joblib"),
|
"inconsistent_model": os.path.join(MODEL_DIR, "inconsistent_model.joblib"),
|
||||||
"consistent_scaler": os.path.join(MODEL_DIR, "consistent_scaler.joblib"),
|
"consistent_scaler": os.path.join(MODEL_DIR, "consistent_scaler.joblib"),
|
||||||
"inconsistent_scaler": os.path.join(MODEL_DIR, "inconsistent_scaler.joblib")
|
"inconsistent_scaler": os.path.join(MODEL_DIR, "inconsistent_scaler.joblib")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# Salary Detect Endpoint Config
|
||||||
|
SALARY_DETECT_URL = "http://www.simbrellang.net:5000/autocall/analytic-salary-detect"
|
||||||
|
SALARY_DETECT_HEADERS = {
|
||||||
|
"accept": "/",
|
||||||
|
"Content-Type": "application/json"
|
||||||
|
}
|
||||||
|
SALARY_DETECT_PAYLOADS = [
|
||||||
|
{"salaryDate": "2022-01-01", "customerId": "CN621868", "accountId": "2017821799", "salaryAmount": 200000},
|
||||||
|
{"customerId": "CUC2268333011", "accountId": "ACC8116931898", "salaryDate": "2025-07-08", "salaryAmount": 200000},
|
||||||
|
{"customerId": "CUC2163677018", "accountId": "ACC8118539484", "salaryDate": "2025-07-08", "salaryAmount": 200000},
|
||||||
|
{"customerId": "CUC1968062010", "accountId": "ACC8115473093", "salaryDate": "2025-07-08", "salaryAmount": 200000},
|
||||||
|
{"customerId": "CUC1302360013", "accountId": "ACC8117628489", "salaryDate": "2025-07-08", "salaryAmount": 200000}
|
||||||
|
]
|
||||||
|
|
||||||
|
def get_random_salary_payload():
|
||||||
|
return [random.choice(SALARY_DETECT_PAYLOADS)]
|
||||||
@@ -1,6 +1,8 @@
|
|||||||
import time
|
import time
|
||||||
import logging
|
import logging
|
||||||
import threading
|
import threading
|
||||||
|
import requests
|
||||||
|
from .config import SALARY_DETECT_URL, SALARY_DETECT_HEADERS, get_random_salary_payload
|
||||||
|
|
||||||
logging.basicConfig(level=logging.INFO)
|
logging.basicConfig(level=logging.INFO)
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
@@ -13,9 +15,14 @@ class SalaryDetect:
|
|||||||
def _run(self):
|
def _run(self):
|
||||||
while self._running:
|
while self._running:
|
||||||
logger.info(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Detecting salary...")
|
logger.info(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Detecting salary...")
|
||||||
time.sleep(1)
|
try:
|
||||||
|
payload = get_random_salary_payload()
|
||||||
|
response = requests.post(SALARY_DETECT_URL, headers=SALARY_DETECT_HEADERS, json=payload)
|
||||||
|
logger.info(f"POST {SALARY_DETECT_URL} status: {response.status_code}, response: {response.text}")
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Error during POST: {e}")
|
||||||
logger.info(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Salary detection complete")
|
logger.info(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Salary detection complete")
|
||||||
time.sleep(1)
|
time.sleep(120)
|
||||||
|
|
||||||
def start(self):
|
def start(self):
|
||||||
if not self._running:
|
if not self._running:
|
||||||
|
|||||||
Reference in New Issue
Block a user