Implement divide and mod for Python backend

1 files changed, 3 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonCompiler.java b/src/main/java/org/zwobble/couscous/backends/python/PythonCompiler.java
index c5ea3e3..e325302 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonCompiler.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCompiler.java
@@ -25,6 +25,9 @@ public class PythonCompiler {
             compileClass(classNode);
         }
         writeClass(TypeName.of("java.lang.Integer"), "class Integer(object):\n    def parseInt(value):\n        return int(value)");
+        writeClass(TypeName.of("_couscous"),
+            "def _div_round_to_zero(a, b): return -(-a // b) if (a < 0) ^ (b < 0) else a // b\n" +
+            "def _mod_round_to_zero(a, b): return -(-a % b) if (a < 0) ^ (b < 0) else a % b");
     }
     
     private Path pathForClass(TypeName className) {