15
15
import tempfile
16
16
17
17
LOG = logging .getLogger ('envkernel' )
18
- LOG .setLevel (logging .DEBUG )
18
+ LOG .setLevel (logging .INFO )
19
19
logging .lastResort .setLevel (logging .DEBUG )
20
20
21
21
@@ -127,9 +127,14 @@ def setup(self):
127
127
parser .add_argument ('--env' , action = 'append' , default = [],
128
128
help = "Environment to add, format NAME=VAL. Can be given multiple times. "
129
129
"These are statically embedded in the kernel.json file" )
130
+ parser .add_argument ('--verbose' , '-v' , action = 'store_true' ,
131
+ help = "Print more debugging information" )
130
132
args , unknown_args = parser .parse_known_args (self .argv )
131
- LOG .debug ('setup: args: %s' , args )
132
- LOG .debug ('setup: unknown_args: %s' , unknown_args )
133
+ if args .verbose :
134
+ LOG .setLevel (logging .DEBUG )
135
+
136
+ LOG .debug ('setup: envkernel setup args: %s' , args )
137
+ LOG .debug ('setup: kernel-specific args: %s' , unknown_args )
133
138
self .setup_args = args
134
139
self .name = args .name
135
140
self .user = args .user
@@ -243,6 +248,13 @@ def install_kernel(self, kernel, name, user=False, replace=None, prefix=None, lo
243
248
LOG .info (" Note: Kernel not detected with current search path." )
244
249
LOG .info (" Command line: %s" , kernel ['argv' ])
245
250
251
+ def run (self ):
252
+ """Hook that gets run before kernel invoked"""
253
+ # User does not directly see this (except interleaved in
254
+ # normal jupyter logging output), so we can set it to debug
255
+ # by default.
256
+ LOG .setLevel (logging .DEBUG )
257
+
246
258
247
259
248
260
class lmod (envkernel ):
@@ -266,6 +278,7 @@ def run(self):
266
278
267
279
before '--': the modules to load
268
280
after '--': the Python command to run after loading"""
281
+ super ().run ()
269
282
argv , rest = split_doubledash (self .argv , 1 )
270
283
parser = argparse .ArgumentParser ()
271
284
parser .add_argument ('--purge' , action = 'store_true' , default = False , help = "Purge existing modules first" )
@@ -330,6 +343,7 @@ def run(self):
330
343
331
344
before '--': the modules to load
332
345
after '--': the Python command to run after loading"""
346
+ super ().run ()
333
347
argv , rest = split_doubledash (self .argv , 1 )
334
348
parser = argparse .ArgumentParser ()
335
349
#parser.add_argument('--purge', action='store_true', default=False, help="Purge existing modules first")
@@ -399,6 +413,7 @@ def setup(self):
399
413
replace = self .replace , prefix = self .prefix )
400
414
401
415
def run (self ):
416
+ super ().run ()
402
417
argv , rest = split_doubledash (self .argv , 1 )
403
418
parser = argparse .ArgumentParser ()
404
419
parser .add_argument ('image' , help = 'Docker image name' )
@@ -534,6 +549,7 @@ def setup(self):
534
549
replace = self .replace , prefix = self .prefix )
535
550
536
551
def run (self ):
552
+ super ().run ()
537
553
argv , rest = split_doubledash (self .argv , 1 )
538
554
parser = argparse .ArgumentParser ()
539
555
parser .add_argument ('image' , help = 'image name' )
0 commit comments