Add Go compilation and execution command

nok · Oct 7, 2017 · 5d24f57 · 5d24f57
1 parent 788c8b4
commit 5d24f57
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/sklearn_porter/Porter.py b/sklearn_porter/Porter.py
@@ -469,7 +469,7 @@ def _get_filename(class_name, language):
         lang = str(language)
 
         # Name:
-        if language == 'java':
+        if language in ['java', 'php']:
             name = name.capitalize()
 
         # Suffix:
@@ -513,7 +513,9 @@ def _get_commands(filename, class_name, language):
             # gcc brain.c -o brain
             'c': 'gcc {} -lm -o {}'.format(fname, cname),
             # javac Brain.java
-            'java': 'javac {}'.format(fname)
+            'java': 'javac {}'.format(fname),
+            # go build -o brain brain.go
+            'go': 'go build -o {} {}.go'.format(cname, cname)
         }
         comp_cmd = comp_vars.get(lang, None)
 
@@ -525,10 +527,12 @@ def _get_commands(filename, class_name, language):
             'java': 'java -classpath . {}'.format(cname.capitalize()),
             # node brain.js
             'js': 'node {}'.format(fname),
-            # php -f brain.php
-            'php': 'php -f {}'.format(fname),
+            # php -f Brain.php
+            'php': 'php -f {}'.format(cname.capitalize()),
             # ruby brain.rb
-            'ruby': 'ruby {}'.format(fname)
+            'ruby': 'ruby {}'.format(fname),
+            # ./brain
+            'go': os.path.join('.', cname),
         }
         exec_cmd = exec_vars.get(lang, None)