Elgato_dark/zeppelin
1
0
Fork 0
mirror of https://github.com/apache/zeppelin synced 2026-05-24 09:38:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

feat: Add conda install

Browse source
This commit is contained in:
1ambda 2017-01-08 10:33:32 +09:00
parent 4f58fa2bcd
commit e539c42a95
2 changed files with 27 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
python/src/main/java/org/apache/zeppelin/python/PythonCondaInterpreter.java
View file

@ -44,6 +44,7 @@ public class PythonCondaInterpreter extends Interpreter {
private Pattern createPattern = Pattern.compile("create\\s*(.*)");
private Pattern activatePattern = Pattern.compile("activate\\s*(.*)");
private Pattern deactivatePattern = Pattern.compile("deactivate");
private Pattern installPattern = Pattern.compile("install\\s*(.*)");
private Pattern helpPattern = Pattern.compile("help");
private Pattern infoPattern = Pattern.compile("info");
@ -66,6 +67,7 @@ public class PythonCondaInterpreter extends Interpreter {
InterpreterOutput out = context.out;
Matcher activateMatcher = activatePattern.matcher(st);
Matcher createMatcher = createPattern.matcher(st);
Matcher installMatcher = installPattern.matcher(st);
try {
if (st == null || listEnvPattern.matcher(st).matches()) {
@ -87,6 +89,9 @@ public class PythonCondaInterpreter extends Interpreter {
changePythonEnvironment(null);
restartPythonProcess();
return new InterpreterResult(Code.SUCCESS, "Deactivated");
} else if (installMatcher.matches()) {
String result = runCondaInstall(getRestArgsFromMatcher(installMatcher));
return new InterpreterResult(Code.SUCCESS, Type.HTML, result);
} else if (helpPattern.matcher(st).matches()) {
runCondaHelp(out);
return new InterpreterResult(Code.SUCCESS);
@ -213,6 +218,24 @@ public class PythonCondaInterpreter extends Interpreter {
return wrapCondaBasicOutputStyle("Environment Created", sb.toString());
}
private String runCondaInstall(List<String> restArgs)
throws IOException, InterruptedException {
restArgs.add(0, "conda");
restArgs.add(1, "install");
restArgs.add(2, "--yes");
StringBuilder sb = new StringBuilder();
int exit = runCommand(sb, restArgs);
if (exit != 0) {
throw new RuntimeException("Failed to execute `" +
StringUtils.join(restArgs, " ") +
"` exited with " + exit);
}
return wrapCondaBasicOutputStyle("Package Installation", sb.toString());
}
private String wrapCondaBasicOutputStyle(String title, String content) {
StringBuilder sb = new StringBuilder();
if (null != title && !title.isEmpty()) {

4
python/src/main/resources/output_templates/conda_usage.html
View file

@ -36,3 +36,7 @@ limitations under the License.
Get installed package list inside the current environment
<pre>%python.conda list</pre>
</div>
<div>
Install
<pre>%python.conda install [PACKAGE NAME]</pre>
</div>