Compensated for keyerror exception for new questions not found

Zanger67 · Jul 21, 2024 · 615471d · 615471d
1 parent 56b9f66
commit 615471d
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 4 deletions.
diff --git a/main.ipynb b/main.ipynb
@@ -565,11 +565,23 @@
     "        number      = int(re.search(\"\\d{1,4}\", leetcodeFile).group())   # Takes the first full number as the question\n",
     "        level       = questionDetailsDict[number].level                 # number and uses that as reference\n",
     "                                                                        # e.g. 'e123 v1.py' becomes 123\n",
-    "    except AttributeError as ae :\n",
-    "        print(f'Error in parsing {leetcodeFile}: Attribute Error encountered while trying to extract question number int(...).',\n",
+    "    except (AttributeError) as ae :\n",
+    "        print(f'Error in parsing {leetcodeFile}: {ae.name} encountered while trying to extract question number int(...).',\n",
     "                '\\nparseCase(...)',\n",
     "                '\\nSkipping')\n",
     "        return False\n",
+    "    except (KeyError) as ke :\n",
+    "        print(f'Error in parsing {leetcodeFile}: {ke} encountered while trying to extract question level from questionDetailsDict.',\n",
+    "                '\\nparseCase(...)',\n",
+    "                '\\nAttempting to pull from the name...')\n",
+    "        level = leetcodeFile[0].lower()\n",
+    "        \n",
+    "        if level in ['e', 'm', 'h'] :\n",
+    "            print(f'Level found: {level}')\n",
+    "        else :\n",
+    "            print(f'Level not found. Defaulting to \"Unknown\"')\n",
+    "            level = 'Unknown'\n",
+    "        return False\n",
     "\n",
     "    creationtime, modificationtime = getCtimeMtimes(join(README_PATH, path))\n",
     "\n",

diff --git a/main.py b/main.py
@@ -502,11 +502,23 @@ def parseCase(leetcodeFile:         str,  # file name
         number      = int(re.search("\d{1,4}", leetcodeFile).group())   # Takes the first full number as the question
         level       = questionDetailsDict[number].level                 # number and uses that as reference
                                                                         # e.g. 'e123 v1.py' becomes 123
-    except AttributeError as ae :
-        print(f'Error in parsing {leetcodeFile}: Attribute Error encountered while trying to extract question number int(...).',
+    except (AttributeError) as ae :
+        print(f'Error in parsing {leetcodeFile}: {ae.name} encountered while trying to extract question number int(...).',
                 '\nparseCase(...)',
                 '\nSkipping')
         return False
+    except (KeyError) as ke :
+        print(f'Error in parsing {leetcodeFile}: {ke} encountered while trying to extract question level from questionDetailsDict.',
+                '\nparseCase(...)',
+                '\nAttempting to pull from the name...')
+        level = leetcodeFile[0].lower()
+
+        if level in ['e', 'm', 'h'] :
+            print(f'Level found: {level}')
+        else :
+            print(f'Level not found. Defaulting to "Unknown"')
+            level = 'Unknown'
+        return False
 
     creationtime, modificationtime = getCtimeMtimes(join(README_PATH, path))