From 13f67c169b5640189b563c66d34317a9a46d4983 Mon Sep 17 00:00:00 2001
From: faze-geek <bhat.1@iitj.ac.in>
Date: Tue, 12 Mar 2024 18:01:11 +0530
Subject: [PATCH] Add check to restrict arguments

---
 src/pocketpy.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/pocketpy.cpp b/src/pocketpy.cpp
index d6d1a765..ce4df652 100644
--- a/src/pocketpy.cpp
+++ b/src/pocketpy.cpp
@@ -498,6 +498,7 @@ void init_builtins(VM* _vm) {
     // tp_str
     _vm->bind_constructor<-1>(_vm->_t(VM::tp_str), [](VM* vm, ArgsView args) {
         if(args.size() == 1) return VAR(Str());
+        if(args.size() > 2) vm->TypeError("str() takes at most 1 argument");
         return vm->py_str(args[1]);
     });