Catch key not found error

Author: Smit-create

src/libasr/codegen/llvm_utils.cpp

@@ -1636,8 +1636,24 @@ namespace LCompilers {
         }
         llvm_utils->start_new_block(mergeBB);
         llvm::Value* pos = LLVM::CreateLoad(*builder, pos_ptr);
+        // Check if the actual key is present or not
+        llvm::Value* is_key_matching = llvm_utils->is_equal_by_value(key,
+                llvm_utils->list_api->read_item(key_list, pos, false, module,
+                    LLVM::is_llvm_struct(key_asr_type)), module, key_asr_type);
+
+        llvm_utils->create_if_else(is_key_matching, [&]() {
+        }, [&]() {
+            std::string message = "The dict does not contain the specified key";
+            llvm::Value *fmt_ptr = builder->CreateGlobalStringPtr("KeyError: %s\n");
+            llvm::Value *fmt_ptr2 = builder->CreateGlobalStringPtr(message);
+            print_error(context, module, *builder, {fmt_ptr, fmt_ptr2});
+            int exit_code_int = 1;
+            llvm::Value *exit_code = llvm::ConstantInt::get(context,
+                    llvm::APInt(32, exit_code_int));
+            exit(context, module, *builder, exit_code);
+        });
         llvm::Value* item = llvm_utils->list_api->read_item(value_list, pos,
-                                                        false, module, true);
+                                                        false, module, true);;
         return item;
     }
 

tests/errors/test_dict14.py

@@ -0,0 +1,8 @@
+from lpython import i32
+
+def key_not_present():
+    x: dict[i32, i32] = {}
+    x = {1: 1, 2: 2}
+    print(x[10])
+
+key_not_present()

tests/reference/runtime-test_dict14-421fe53.json

@@ -0,0 +1,13 @@
+{
+    "basename": "runtime-test_dict14-421fe53",
+    "cmd": "lpython {infile}",
+    "infile": "tests/errors/test_dict14.py",
+    "infile_hash": "c81e4a1e050c87d04f537b49fcfb0b479a5ce3ed7735f90d2c347dba",
+    "outfile": null,
+    "outfile_hash": null,
+    "stdout": null,
+    "stdout_hash": null,
+    "stderr": "runtime-test_dict14-421fe53.stderr",
+    "stderr_hash": "cb46ef04db0862506d688ebe8830a50afaaead9b0d29b0c007dd149a",
+    "returncode": 1
+}

tests/tests.toml

@@ -808,6 +808,10 @@ asr = true
 filename = "errors/test_dict13.py"
 asr = true
 
+[[test]]
+filename = "errors/test_dict14.py"
+run = true
+
 [[test]]
 filename = "errors/test_zero_division.py"
 asr = true