From bd499e01b67f2eb3cd9263f5d5d819fb118c5756 Mon Sep 17 00:00:00 2001
From: Benjamin Diaz <bdiaz@whitestack.com>
Date: Thu, 20 Sep 2018 14:12:52 -0300
Subject: [PATCH] Fixes dependency error in setup.py

Signed-off-by: Benjamin Diaz <bdiaz@whitestack.com>
---
 osm_policy_module/core/database.py                         | 2 +-
 osm_policy_module/tests/integration/test_kafka_messages.py | 1 -
 osm_policy_module/tests/integration/test_policy_agent.py   | 4 ++--
 setup.py                                                   | 2 +-
 tox.ini                                                    | 2 +-
 5 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/osm_policy_module/core/database.py b/osm_policy_module/core/database.py
index a89baed..f9a37c2 100644
--- a/osm_policy_module/core/database.py
+++ b/osm_policy_module/core/database.py
@@ -23,7 +23,7 @@
 ##
 import logging
 
-from peewee import *
+from peewee import CharField, IntegerField, ForeignKeyField, Model, TextField
 from playhouse.sqlite_ext import SqliteExtDatabase
 
 from osm_policy_module.core.config import Config
diff --git a/osm_policy_module/tests/integration/test_kafka_messages.py b/osm_policy_module/tests/integration/test_kafka_messages.py
index d17d2b2..581d53f 100644
--- a/osm_policy_module/tests/integration/test_kafka_messages.py
+++ b/osm_policy_module/tests/integration/test_kafka_messages.py
@@ -30,7 +30,6 @@ import unittest
 from kafka import KafkaProducer, KafkaConsumer
 from kafka.errors import KafkaError
 
-from osm_policy_module.core.agent import PolicyModuleAgent
 from osm_policy_module.core.config import Config
 
 log = logging.getLogger()
diff --git a/osm_policy_module/tests/integration/test_policy_agent.py b/osm_policy_module/tests/integration/test_policy_agent.py
index 4de0dbb..3a82ea7 100644
--- a/osm_policy_module/tests/integration/test_policy_agent.py
+++ b/osm_policy_module/tests/integration/test_policy_agent.py
@@ -29,12 +29,12 @@ from unittest.mock import patch, Mock
 
 from kafka import KafkaProducer
 from osm_common.dbmongo import DbMongo
-from osm_policy_module.core import database
 from peewee import SqliteDatabase
 
 from osm_policy_module.common.mon_client import MonClient
+from osm_policy_module.core import database
 from osm_policy_module.core.agent import PolicyModuleAgent
-from osm_policy_module.core.database import ScalingRecord, ScalingAlarm, BaseModel
+from osm_policy_module.core.database import ScalingRecord, ScalingAlarm
 
 log = logging.getLogger()
 log.level = logging.INFO
diff --git a/setup.py b/setup.py
index c751cbf..c3c735d 100644
--- a/setup.py
+++ b/setup.py
@@ -44,7 +44,7 @@ setup(
     maintainer_email=_maintainer_email,
     url=_url,
     license=_license,
-    packages=setuptools.find_packages(),
+    packages=[_name],
     include_package_data=True,
     install_requires=[
         "kafka==1.3.*",
diff --git a/tox.ini b/tox.ini
index ca49be9..4ef6a64 100644
--- a/tox.ini
+++ b/tox.ini
@@ -25,7 +25,7 @@
 # test suite on all supported python versions. To use it, "pip install tox"
 # and then run "tox" from this directory.
 [tox]
-envlist = py3, flake8
+envlist = py3
 toxworkdir={homedir}/.tox
 
 [testenv]
-- 
2.25.1