service filepath fix

nfsinkhole/service.py

@@ -23,6 +23,7 @@
 # POSSIBILITY OF SUCH DAMAGE.
 
 from .tcpdump import TCPDump
+from .exceptions import BinaryNotFound
 from .utils import popen_wrapper
 import logging
 import os
@@ -170,6 +171,25 @@ def create_service(self):
         service = open('nfsinkhole.service', "w")
         with service:
 
+            fp = sys.executable
+
+            # Raise error if rsyslogd is not found
+            if not os.path.exists('{0}/nfsinkhole-service.py').format(fp):
+
+                if os.path.exists('/usr/local/bin/nfsinkhole-service.py'):
+
+                    fp = "/usr/local/bin"
+
+                else:
+
+                    log.debug('Path not found: {0}/nfsinkhole-service.py'
+                              ).format(fp)
+                    raise BinaryNotFound('Service file was not detected.')
+
+            else:
+
+                fp = os.path.dirname(fp)
+
             # Run pre main process execution
             execstartpre = (
                 '-{pyfp} {fp}/nfsinkhole-service.py '
@@ -185,7 +205,7 @@ def create_service(self):
                 '--loglevel {loglevel} '
                 ''.format(
                     pyfp=sys.executable,
-                    fp=os.path.dirname(sys.executable),
+                    fp=fp,
                     interface=self.interface,
                     protocol=self.protocol,
                     dport=self.dport,
@@ -204,7 +224,7 @@ def create_service(self):
                 '-{pyfp} {fp}/nfsinkhole-service.py '
                 '--delete --interface {interface} --loglevel {loglevel}'.format(
                     pyfp=sys.executable,
-                    fp=os.path.dirname(sys.executable),
+                    fp=fp,
                     interface=self.interface,
                     loglevel=self.loglevel
                 )