Check status code when fetching library info from PyPI

Author: patrick91

package.json

@@ -2,7 +2,7 @@
     "name": "python-dependencies-vscode",
     "displayName": "Python Dependencies",
     "description": "Utilities for managing Python dependencies",
-    "version": "0.0.16",
+    "version": "0.0.17",
     "publisher": "patrick91",
     "engines": {
         "vscode": "^1.53.0"

src/core/api.ts

@@ -8,7 +8,7 @@ const getCacheKey = (library: string) => `${CACHE_PREFIX}-${library}`;
 
 export const getInfo = async (
     library: string,
-): Promise<{ version: string; summary: string }> => {
+): Promise<{ version: string; summary: string } | null> => {
     const cache = getCache();
     const key = getCacheKey(library);
 
@@ -27,12 +27,16 @@ export const getInfo = async (
 
     const data = await fetchLibrary(library);
 
-    info = {
-        version: data.info.version as string,
-        summary: data.info.summary as string,
-    };
+    if (data) {
+        info = {
+            version: data.info.version as string,
+            summary: data.info.summary as string,
+        };
 
-    await cache.put(key, JSON.stringify({ info, date: new Date() }));
+        await cache.put(key, JSON.stringify({ info, date: new Date() }));
 
-    return info;
+        return info;
+    }
+
+    return null;
 };

src/core/listener.ts

@@ -17,6 +17,10 @@ const updateVersions = (
         deps.map(async (dep) => {
             const info = await getInfo(dep.name);
 
+            if (!info) {
+                return;
+            }
+
             dep.version.installed = installedVersions[dep.name];
             dep.version.latest = info.version;
             dep.summary = info.summary;

src/core/pypi.ts

@@ -1,7 +1,13 @@
 import fetch from "node-fetch";
 
 export const fetchLibrary = async (library: string) => {
-    const x = await fetch(`https://pypi.org/pypi/${library}/json`);
+    const response = await fetch(`https://pypi.org/pypi/${library}/json`);
 
-    return await x.json();
+    if (response.status === 200) {
+        return await response.json();
+    }
+
+    console.log(`unable to fetch library ${library}`);
+
+    return null;
 };