fix: use absolute imports

Author: fstagni

Pilot/dirac-pilot.py

@@ -24,22 +24,13 @@
 import time
 from io import StringIO
 
-try:
-    from Pilot.pilotTools import (
-        Logger,
-        PilotParams,
-        RemoteLogger,
-        getCommand,
-        pythonPathCheck,
-    )
-except ModuleNotFoundError:
-    from pilotTools import (
-        Logger,
-        PilotParams,
-        RemoteLogger,
-        getCommand,
-        pythonPathCheck,
-    )
+from .pilotTools import (
+    Logger,
+    PilotParams,
+    RemoteLogger,
+    getCommand,
+    pythonPathCheck,
+)
 
 ############################
 

Pilot/pilotCommands.py

@@ -30,16 +30,7 @@ def __init__(self, pilotParams):
 from http.client import HTTPSConnection
 from shlex import quote
 
-try:
-    from Pilot.pilotTools import (
-        CommandBase,
-        getSubmitterInfo,
-        retrieveUrlTimeout,
-        safe_listdir,
-        sendMessage,
-    )
-except ModuleNotFoundError:
-    from pilotTools import (
+from .pilotTools import (
         CommandBase,
         getSubmitterInfo,
         retrieveUrlTimeout,

Pilot/pilotTools.py

@@ -22,10 +22,7 @@
 from urllib.parse import urlencode
 from urllib.request import urlopen
 
-try:
-    from Pilot.proxyTools import getVO
-except ModuleNotFoundError:
-    from proxyTools import getVO
+from .proxyTools import getVO
 
 # Utilities functions
 

Pilot/tests/Test_Pilot.py

@@ -10,12 +10,8 @@
 # imports
 import unittest
 
-try:
-    from Pilot.pilotCommands import CheckWorkerNode, ConfigureSite, NagiosProbes
-    from Pilot.pilotTools import PilotParams
-except ModuleNotFoundError:
-    from pilotCommands import CheckWorkerNode, ConfigureSite, NagiosProbes
-    from pilotTools import PilotParams
+from ..pilotCommands import CheckWorkerNode, ConfigureSite, NagiosProbes
+from ..pilotTools import PilotParams
 
 
 class PilotTestCase(unittest.TestCase):

Pilot/tests/Test_proxyTools.py

@@ -7,10 +7,7 @@
 import unittest
 from unittest.mock import patch
 
-try:
-    from Pilot.proxyTools import getVO, parseASN1
-except ModuleNotFoundError:
-    from proxyTools import getVO, parseASN1
+from ..proxyTools import getVO, parseASN1
 
 class TestProxyTools(unittest.TestCase):
     def test_getVO(self):
@@ -84,46 +81,7 @@ def __createFakeProxy(self, proxyFile):
         """
         Create a fake proxy locally.
         """
-
         basedir = os.path.dirname(__file__)
-        shutil.copy(basedir + "/certs/user/userkey.pem", basedir + "/certs/user/userkey400.pem")
-        os.chmod(basedir + "/certs/user/userkey400.pem", 0o400)
-        ret = self.createFakeProxy(
-            basedir + "/certs/user/usercert.pem",
-            basedir + "/certs/user/userkey400.pem",
-            "fakeserver.cern.ch:15000",
-            "fakevo",
-            basedir + "/certs//host/hostcert.pem",
-            basedir + "/certs/host/hostkey.pem",
-            basedir + "/certs/ca",
-            proxyFile,
-        )
-        os.remove(basedir + "/certs/user/userkey400.pem")
-        return ret
-
-    def createFakeProxy(self, usercert, userkey, serverURI, vo, hostcert, hostkey, CACertDir, proxyfile):
-        """
-        voms-proxy-fake --cert usercert.pem
-                        --key userkey.pem
-                        -rfc
-                        -fqan "/fakevo/Role=user/Capability=NULL"
-                        -uri fakeserver.cern.ch:15000
-                        -voms fakevo
-                        -hostcert hostcert.pem
-                        -hostkey hostkey.pem
-                        -certdir ca
-        """
-        opt = (
-            '--cert %s --key %s -rfc -fqan "/fakevo/Role=user/Capability=NULL" -uri %s -voms %s -hostcert %s'
-            "  -hostkey %s  -certdir %s -out %s"
-            % (usercert, userkey, serverURI, vo, hostcert, hostkey, CACertDir, proxyfile)
-        )
-        proc = subprocess.Popen(
-            shlex.split("voms-proxy-fake " + opt),
-            bufsize=1,
-            stdout=sys.stdout,
-            stderr=sys.stderr,
-            universal_newlines=True,
-        )
-        proc.communicate()
-        return proc.returncode
+        shutil.copy(basedir + "/certs/voms/proxy.pem", proxyFile)
+        return 0
+

Pilot/tests/Test_simplePilotLogger.py

@@ -8,10 +8,7 @@
 import unittest
 from unittest.mock import patch
 
-try:
-    from Pilot.pilotTools import CommandBase, Logger, PilotParams
-except ModuleNotFoundError:
-    from pilotTools import CommandBase, Logger, PilotParams
+from ..pilotTools import CommandBase, Logger, PilotParams
 
 class TestPilotParams(unittest.TestCase):
     @patch("sys.argv")