make api compatible for both runtime and unit tests

This commit is contained in:
Sefik Ilkin Serengil 2024-02-04 11:03:57 +00:00
parent d080b24cad
commit 2e170474f6
3 changed files with 16 additions and 8 deletions

View File

@ -1,6 +1,12 @@
# 3rd parth dependencies
from flask import Flask
from modules.core.routes import blueprint
try:
# unit tests
from api.src.modules.core.routes import blueprint
except:
# runtime
from modules.core.routes import blueprint
def create_app():

View File

@ -1,5 +1,12 @@
from flask import Blueprint, request
from modules.core import service
try:
# unit tests
from api.src.modules.core import service
except:
# runtime
from modules.core import service
from deepface.commons.logger import Logger
logger = Logger(module="api/src/routes.py")

View File

@ -1,12 +1,7 @@
import sys
import unittest
from deepface.commons import constant
from deepface.commons.logger import Logger
# api is not available under deepface import
sys.path.insert(1, f"{constant.ROOT_DIR}/api/src")
# pylint: disable=wrong-import-order, wrong-import-position
from app import create_app
from api.src.app import create_app
logger = Logger("tests/test_api.py")