Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 11:30:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add zero div error

Browse Source
This commit is contained in:
blueloveTH 2022-11-10 13:49:26 +08:00
parent 4842a108c3
commit ad23d21297
2 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/pocketpy.h
View File

@ -124,7 +124,9 @@ void __initializeBuiltinFunctions(VM* _vm) {
_vm->bindMethodMulti({"int", "float"}, "__truediv__", [](VM* vm, PyVarList args) { _vm->bindMethodMulti({"int", "float"}, "__truediv__", [](VM* vm, PyVarList args) {
if(!vm->isIntOrFloat(args[0], args[1])) if(!vm->isIntOrFloat(args[0], args[1]))
vm->typeError("unsupported operand type(s) for " "/" ); vm->typeError("unsupported operand type(s) for " "/" );
return vm->PyFloat(vm->numToFloat(args[0]) / vm->numToFloat(args[1])); _Float rhs = vm->numToFloat(args[1]);
if (rhs == 0) vm->zeroDivisionError();
return vm->PyFloat(vm->numToFloat(args[0]) / rhs);
}); });
_vm->bindMethodMulti({"int", "float"}, "__pow__", [](VM* vm, PyVarList args) { _vm->bindMethodMulti({"int", "float"}, "__pow__", [](VM* vm, PyVarList args) {
@ -141,7 +143,9 @@ void __initializeBuiltinFunctions(VM* _vm) {
_vm->bindMethod("int", "__floordiv__", [](VM* vm, PyVarList args) { _vm->bindMethod("int", "__floordiv__", [](VM* vm, PyVarList args) {
if(!args[0]->isType(vm->_tp_int) || !args[1]->isType(vm->_tp_int)) if(!args[0]->isType(vm->_tp_int) || !args[1]->isType(vm->_tp_int))
vm->typeError("unsupported operand type(s) for " "//" ); vm->typeError("unsupported operand type(s) for " "//" );
return vm->PyInt(vm->PyInt_AS_C(args[0]) / vm->PyInt_AS_C(args[1])); _Int rhs = vm->PyInt_AS_C(args[1]);
if(rhs == 0) vm->zeroDivisionError();
return vm->PyInt(vm->PyInt_AS_C(args[0]) / rhs);
}); });
_vm->bindMethod("int", "__mod__", [](VM* vm, PyVarList args) { _vm->bindMethod("int", "__mod__", [](VM* vm, PyVarList args) {

8
src/vm.h
View File

@ -655,11 +655,15 @@ private:
public: public:
void typeError(const _Str& msg){ void typeError(const _Str& msg){
typeError(msg); _error("TypeError", msg);
} }
void systemError(const _Str& msg){ void systemError(const _Str& msg){
systemError(msg); _error("SystemError", msg);
}
void zeroDivisionError(){
_error("ZeroDivisionError", "division by zero");
} }
void indexError(const _Str& msg){ void indexError(const _Str& msg){