Add option to configure API URL

1 files changed, 5 insertions, 5 deletions

diff --git a/session.py b/session.py
index c30fb19..37cc312 100644
--- a/session.py
+++ b/session.py
@@ -1,28 +1,28 @@
-import logging
 import os
 import random
 import secrets
 import signal
 import string
 import subprocess
-import threading
 from typing import Dict, Union
 
 import port_for
 import psutil
 import requests
-from flask import Flask, abort, request
-from werkzeug.serving import make_server
 
+from config import settings
+from log import logging
 from vnc import run_vnc, save_password
 
-API_SERVER = "http://127.0.0.1:8000"
+API_SERVER = settings.get("api_url", "http://127.0.0.1")
 BASE_URL = API_SERVER + "/app/rwa/api/"
 REGISTER_URL = BASE_URL + "register/"
 STOP_URL = BASE_URL + "stop/"
 STATUS_URL = BASE_URL + "status/"
 MARK_JOB_AS_DONE_URL = BASE_URL + "jobs/mark_as_done/"
 
+logging.info(f"Load API config: {API_SERVER}")
+
 
 def random_digits(length: int):
     return "".join(random.choice(string.digits) for _ in range(length))