commit a4e615d9bab28eefa0ae1e713ad75289ed2f65cc
author Benjamin Diaz <bdiaz@whitestack.com> Mon Oct 08 16:34:17 2018 -0300
committer Benjamin Diaz <bdiaz@whitestack.com> Sun Oct 14 18:38:20 2018 -0300
tree f100dd5c7968ab31d8b702ef62cab60ddf1cefd9
parent f35f914f45080e150cc7c6a2928de47a9ae5c848

Adds use of OSMPOL_SQL_DATABASE_URI config param to connect to DB

The OSMPOL_SQL_DATABASE_URI was not being used by POL to connect
to the database. There was a hardcoded sqlite database name.
This change should now provide capability to the user to use any
SQL database engine supported by the peewee ORM (mysql/mariadb,
postgres, sqlite).

Signed-off-by: Benjamin Diaz <bdiaz@whitestack.com>
Change-Id: Iff406862fb3dfeb4453fb3213c94db6d62385d96

docker/Dockerfile

diff --git a/docker/Dockerfile b/docker/Dockerfile
index ab5b411..5953489 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -44,7 +44,7 @@
 ENV OSMPOL_DATABASE_HOST mongo
 ENV OSMPOL_DATABASE_PORT 27017
 
-ENV OSMPOL_SQL_DATABASE_URI sqlite:///mon_sqlite.db
+ENV OSMPOL_SQL_DATABASE_URI sqlite:///policy_module.db
 
 ENV OSMPOL_LOG_LEVEL INFO
 ENV OSMPOL_KAFKA_LOG_LEVEL WARN

osm_policy_module/core/config.py

diff --git a/osm_policy_module/core/config.py b/osm_policy_module/core/config.py
index 84a1f57..94dec98 100644
--- a/osm_policy_module/core/config.py
+++ b/osm_policy_module/core/config.py
@@ -65,7 +65,7 @@
         CfgParam('OSMPOL_DATABASE_DRIVER', "mongo", six.text_type),
         CfgParam('OSMPOL_DATABASE_HOST', "mongo", six.text_type),
         CfgParam('OSMPOL_DATABASE_PORT', 27017, int),
-        CfgParam('OSMPOL_SQL_DATABASE_URI', "sqlite:///mon_sqlite.db", six.text_type),
+        CfgParam('OSMPOL_SQL_DATABASE_URI', "sqlite:///policy_module.db", six.text_type),
         CfgParam('OSMPOL_LOG_LEVEL', "INFO", six.text_type),
         CfgParam('OSMPOL_KAFKA_LOG_LEVEL', "WARN", six.text_type),
     ]

osm_policy_module/core/database.py

diff --git a/osm_policy_module/core/database.py b/osm_policy_module/core/database.py
index 8ad19f2..e522b8d 100644
--- a/osm_policy_module/core/database.py
+++ b/osm_policy_module/core/database.py
@@ -25,14 +25,14 @@
 import logging
 
 from peewee import CharField, IntegerField, ForeignKeyField, Model, TextField, AutoField, DateTimeField
-from playhouse.sqlite_ext import SqliteExtDatabase
+from playhouse.db_url import connect
 
 from osm_policy_module.core.config import Config
 
 log = logging.getLogger(__name__)
 cfg = Config.instance()
 
-db = SqliteExtDatabase('policy_module.db')
+db = connect(cfg.OSMPOL_SQL_DATABASE_URI)
 
 
 class BaseModel(Model):