From bd26db1f239eea0ad7f6980e32d1a01fe86aaea7 Mon Sep 17 00:00:00 2001
From: m0duspwnens <josh.patterson@securityonionsolutions.com>
Date: Mon, 11 Nov 2019 10:50:35 -0500
Subject: [PATCH] install docker python3 library and set env var PYTHONPATH -
 https://github.com/Security-Onion-Solutions/securityonion-saltstack/issues/105

---
 so-setup-network.sh | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/so-setup-network.sh b/so-setup-network.sh
index b7c858d32..8c9efa955 100644
--- a/so-setup-network.sh
+++ b/so-setup-network.sh
@@ -355,7 +355,9 @@ docker_install() {
     yum -y install yum-utils device-mapper-persistent-data lvm2 openssl
     yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
     yum -y update
-    yum -y install docker-ce docker-python python-docker
+    yum -y install docker-ce
+    pip3 install docker
+    set_environment_var "PYTHONPATH=/usr/local/lib/python3.6/site-packages/"
     if [ $INSTALLTYPE != 'EVALMODE'  ]; then
       docker_registry
     fi
@@ -1023,6 +1025,15 @@ sensor_pillar() {
 
 }
 
+set_environment_var() {
+
+  echo "Setting environment variable: $1"
+
+  export "$1"
+  echo "$1" >> /etc/environment
+
+}
+
 set_hostname() {
 
   hostnamectl set-hostname --static $HOSTNAME