From ac17562579a6bb665171cb434acb77b78c5736fa Mon Sep 17 00:00:00 2001
From: Klazomenai <19855141+Klazomenai@users.noreply.github.com>
Date: Wed, 28 Jul 2021 09:36:08 +0100
Subject: [PATCH] Adding exception for KeyError to ensure it does not fail when
 finding a directory

---
 main.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 30c21c7..c3804cb 100755
--- a/main.py
+++ b/main.py
@@ -63,10 +63,13 @@ def get_files(org, header_auth, repo, branch, search_dir):
     files = []
     if result["data"]["repository"]["filename"]:
         for file in result["data"]["repository"]["filename"]["entries"]:
-            if not bool(file["object"]["isBinary"]):
-                files.append({"path": file["path"],
-                              "text": file["object"]["text"]
-                              })
+            try:
+                if not bool(file["object"]["isBinary"]):
+                    files.append({"path": file["path"],
+                                  "text": file["object"]["text"]
+                                  })
+            except KeyError:
+                pass
     return files