Show More
@@ -31,21 +31,25 b' class ExecutePreprocessor(Preprocessor):' | |||||
31 | super(ExecutePreprocessor, self).__init__(**kwargs) |
|
31 | super(ExecutePreprocessor, self).__init__(**kwargs) | |
32 | self.extra_arguments = [] |
|
32 | self.extra_arguments = [] | |
33 |
|
33 | |||
34 | def preprocess(self, nb, resources): |
|
34 | def _create_client(self): | |
35 | self.km = KernelManager() |
|
35 | self.km = KernelManager() | |
36 | self.km.start_kernel(extra_arguments=self.extra_arguments, stderr=open(os.devnull, 'w')) |
|
36 | self.km.start_kernel(extra_arguments=self.extra_arguments, stderr=open(os.devnull, 'w')) | |
37 | self.kc = self.km.client() |
|
37 | self.kc = self.km.client() | |
38 | self.kc.start_channels() |
|
38 | self.kc.start_channels() | |
39 | self.iopub = self.kc.iopub_channel |
|
39 | self.iopub = self.kc.iopub_channel | |
40 | self.shell = self.kc.shell_channel |
|
40 | self.shell = self.kc.shell_channel | |
41 |
|
||||
42 | self.shell.execute("pass") |
|
41 | self.shell.execute("pass") | |
43 | self.shell.get_msg() |
|
42 | self.shell.get_msg() | |
44 |
|
43 | |||
|
44 | def _shutdown_client(self): | |||
|
45 | self.kc.stop_channels() | |||
|
46 | self.km.shutdown_kernel() | |||
|
47 | del self.km | |||
45 |
|
48 | |||
46 | create_client() |
|
49 | def preprocess(self, nb, resources): | |
|
50 | self._create_client() | |||
47 | nb, resources = super(ExecutePreprocessor, self).preprocess(nb, resources) |
|
51 | nb, resources = super(ExecutePreprocessor, self).preprocess(nb, resources) | |
48 | shutdown_client() |
|
52 | self._shutdown_client() | |
49 | return nb, resources |
|
53 | return nb, resources | |
50 |
|
54 | |||
51 | def preprocess_cell(self, cell, resources, cell_index): |
|
55 | def preprocess_cell(self, cell, resources, cell_index): | |
@@ -109,8 +113,3 b' class ExecutePreprocessor(Preprocessor):' | |||||
109 | outs.append(out) |
|
113 | outs.append(out) | |
110 | return outs |
|
114 | return outs | |
111 |
|
115 | |||
112 |
|
||||
113 | def __del__(self): |
|
|||
114 | self.kc.stop_channels() |
|
|||
115 | self.km.shutdown_kernel() |
|
|||
116 | del self.km |
|
General Comments 0
You need to be logged in to leave comments.
Login now