From cd69e09af08cb7c208ff2a426a16f84303f92367 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcin=20W=C4=85troba?= <markowanga@gmail.com>
Date: Mon, 16 Aug 2021 23:31:39 +0200
Subject: [PATCH] Add port to base service
---
sziszapangma/integration/base_asr_service/asr_processor.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/sziszapangma/integration/base_asr_service/asr_processor.py b/sziszapangma/integration/base_asr_service/asr_processor.py
index 35586d7..394fb92 100644
--- a/sziszapangma/integration/base_asr_service/asr_processor.py
+++ b/sziszapangma/integration/base_asr_service/asr_processor.py
@@ -11,13 +11,15 @@ from flask_httpauth import HTTPTokenAuth
from sziszapangma.integration.base_asr_service.asr_result import AsrResult
_TEMP_DIRECTORY = "asr_processing"
+_AUTH_TOKEN = 'AUTH_TOKEN'
+_SERVICE_PORT = 'SERVICE_PORT'
class AsrProcessor(ABC):
user_token: str
def __init__(self):
- self.user_token = os.environ["AUTH_TOKEN"]
+ self.user_token = os.environ[_AUTH_TOKEN]
@abstractmethod
def process_asr(self, audio_file_path: str) -> AsrResult:
@@ -61,4 +63,5 @@ class AsrProcessor(ABC):
Path(_TEMP_DIRECTORY).mkdir(parents=True, exist_ok=True)
app.route("/process_asr", methods=["POST"])(auth.login_required(self.process_request))
app.route("/health_check", methods=["GET"])(self.health_check)
- app.run(debug=True, host="0.0.0.0")
+ port = int(os.environ[_SERVICE_PORT]) if _SERVICE_PORT in os.environ else 5000
+ app.run(debug=True, host="0.0.0.0", port=port)
--
GitLab