Skip to content

Commit 4080500

Browse files
committed
fix: modular import bug
1 parent 1870adc commit 4080500

File tree

4 files changed

+22
-22
lines changed

4 files changed

+22
-22
lines changed

powerapi/database/__init__.py

+3-16
Original file line numberDiff line numberDiff line change
@@ -27,22 +27,9 @@
2727
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
2828
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2929

30-
import logging
3130
from powerapi.database.base_db import BaseDB, IterDB, DBError
3231
from powerapi.database.csvdb import CsvDB, CsvBadFilePathError
3332
from powerapi.database.csvdb import CsvBadCommonKeysError, HeaderAreNotTheSameError
34-
try:
35-
from powerapi.database.mongodb import MongoDB, MongoBadDBError
36-
37-
except ImportError:
38-
logging.getLogger().info("PyMongo is not installed.")
39-
40-
try:
41-
from powerapi.database.opentsdb import OpenTSDB, CantConnectToOpenTSDBException
42-
except ImportError:
43-
logging.getLogger().info("opentsdb-py is not installed.")
44-
45-
try:
46-
from powerapi.database.influxdb import InfluxDB, CantConnectToInfluxDBException
47-
except ImportError:
48-
logging.getLogger().info("influx-client is not installed.")
33+
from powerapi.database.mongodb import MongoDB, MongoBadDBError
34+
from powerapi.database.opentsdb import OpenTSDB, CantConnectToOpenTSDBException
35+
from powerapi.database.influxdb import InfluxDB, CantConnectToInfluxDBException

powerapi/database/influxdb.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,18 @@
2727
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
2828
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2929

30+
31+
import logging
32+
try:
33+
from influxdb import InfluxDBClient
34+
except ImportError:
35+
logging.getLogger().info("influx-client is not installed.")
36+
3037
from typing import List
3138

32-
from influxdb.exceptions import InfluxDBClientError, InfluxDBServerError
3339
from requests.exceptions import ConnectionError
3440
from powerapi.database import BaseDB, DBError
35-
from influxdb import InfluxDBClient
41+
3642
from powerapi.report import Report
3743
from powerapi.report_model import ReportModel
3844

powerapi/database/mongodb.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,11 @@
2727
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
2828
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2929

30-
import pymongo
31-
from pymongo.errors import ConnectionFailure
30+
import logging
31+
try:
32+
import pymongo
33+
except ImportError:
34+
logging.getLogger().info("PyMongo is not installed.")
3235

3336
from typing import List
3437
from powerapi.database.base_db import BaseDB, DBError, IterDB

powerapi/database/opentsdb.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,14 @@
2626
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
2727
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
2828
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29+
import logging
30+
try:
31+
from opentsdb import TSDBClient
32+
except ImportError:
33+
logging.getLogger().info("opentsdb-py is not installed.")
34+
2935
from typing import List
3036

31-
import logging
32-
from opentsdb import TSDBClient
3337
from powerapi.report import PowerReport, Report
3438
from powerapi.report_model import ReportModel
3539
from powerapi.database import BaseDB, DBError

0 commit comments

Comments
 (0)