Skip to content

Commit 018423a

Browse files
committed
fix: Small fix to the legacy pilot
1 parent 71717b0 commit 018423a

File tree

2 files changed

+11
-6
lines changed

2 files changed

+11
-6
lines changed

Pilot/dirac-pilot.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@
6262
pilotUUID=pilotParams.pilotUUID,
6363
debugFlag=pilotParams.debugFlag,
6464
jwt=pilotParams.jwt,
65-
legacy_logging=pilotParams.isLegacyLogging,
65+
legacy_logging=pilotParams.isLegacyPilot,
6666
clientID=pilotParams.clientID
6767
)
6868
log.info("Remote logger activated")
@@ -126,7 +126,7 @@
126126
if remote:
127127
log.buffer.flush(force=True)
128128

129-
if not pilotParams.isLegacyLogging:
129+
if not pilotParams.isLegacyPilot:
130130
log.info("Revoking pilot token.")
131131
revokePilotToken(
132132
pilotParams.diracXServer,

Pilot/pilotTools.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,14 @@ def load_module_from_path(module_name, path_to_module):
2727
return module
2828

2929

30-
from threading import Timer
31-
32-
from proxyTools import BaseRequest, extract_diracx_payload, getVO, refreshUserToken
30+
from proxyTools import (
31+
BaseRequest,
32+
extract_diracx_payload,
33+
getVO,
34+
refreshUserToken,
35+
refreshPilotToken,
36+
TokenBasedRequest,
37+
)
3338

3439
# Utilities functions
3540

@@ -768,7 +773,7 @@ def __init__(self, pilotParams):
768773
flushInterval=interval,
769774
bufsize=bufsize,
770775
jwt=pilotParams.jwt,
771-
legacy_logging=pilotParams.isLegacyLogging,
776+
legacy_logging=pilotParams.isLegacyPilot,
772777
clientID=pilotParams.clientID,
773778
)
774779

0 commit comments

Comments
 (0)