From f4e61929d0eb490dbf7e8ddc03a20d42d44b6f5c Mon Sep 17 00:00:00 2001
From: BonfaceKilz
Date: Tue, 21 Jul 2020 12:36:06 +0300
Subject: Add work-around for failed imports in unittest

* wqflask/utility/tools.py: Unittests will use `from wqflask.wqflask import app`
and the gn2 script will use `from wqflask import app`
---
 wqflask/utility/tools.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'wqflask')

diff --git a/wqflask/utility/tools.py b/wqflask/utility/tools.py
index 77db5d53..37f9d8fe 100644
--- a/wqflask/utility/tools.py
+++ b/wqflask/utility/tools.py
@@ -5,7 +5,10 @@ import os
 import sys
 import json
 
-from wqflask import app
+try:
+    from wqflask import app
+except ImportError:
+    from wqflask.wqflask import app
 
 # Use the standard logger here to avoid a circular dependency
 import logging
-- 
cgit v1.2.3