Fixed trailing newline problem #26 and #27

Author: tndrle

pyluatex.lua

@@ -67,6 +67,14 @@ local function not_empty(str)
     return str ~= nil and str ~= ""
 end
 
+local function split_lines(str)
+    local lines = str:splitlines()
+    if lines[#lines] == "" then
+        table.remove(lines, #lines)
+    end
+    return lines
+end
+
 function pyluatex.start(executable, local_imports)
     local script = file.join(folder, "pyluatex-interpreter.py")
 
@@ -130,8 +138,8 @@ function pyluatex.execute(code, auto_print, write, repl_mode, store)
         repl_mode = repl_mode,
         ignore_errors = pyluatex.ignore_errors
     })
-    local code_lines = code:splitlines()
-    local output_lines = resp.output:splitlines()
+    local code_lines = split_lines(code)
+    local output_lines = split_lines(resp.output)
     if store then
         last_code = code_lines
         last_output = output_lines

test/test-cases/article_template.tex

@@ -11,6 +11,7 @@
 \documentclass{article}
 \pagenumbering{gobble}
 \usepackage[%s]{pyluatex}
+\usepackage{luacode}
 
 \def\TESTID{%s}
 \typeout{TestID: \TESTID}

test/test-cases/code_and_output_storage.yaml

@@ -0,0 +1,29 @@
+tests:
+- code: |
+    \begin{pythonrepl}
+    def func(x):
+      return 2 * x
+
+    func(7)
+    \end{pythonrepl}
+    \begin{luacode}
+      local output = pyluatex.get_last_output()
+      texio.write_nl("Last output line: |" .. output[#output] .. "|")
+    \end{luacode}
+  log: |
+    Last output line: |14|
+- code: |
+    \begin{python}
+    a = 11
+    print(a)
+    \end{python}
+    \begin{luacode}
+      local code = pyluatex.get_last_code()
+      texio.write_nl("Last code line: |" .. code[#code] .. "|")
+      local output = pyluatex.get_last_output()
+      texio.write_nl("Last output line: |" .. output[#output] .. "|")
+    \end{luacode}
+  log: |
+    Last code line: |print(a)|
+    Last output line: |11|
+  pdf: 11