Fix error when running add and status commands on empty repo

Signed-off-by: Jacob Stopak <jacob@initialcommit.io>

Author: initialcommit-io

git_sim/git_sim_add.py

@@ -7,6 +7,7 @@ def __init__(self, scene):
         super().__init__(scene)
         self.maxrefs = 2
         self.hide_first_tag = True
+        self.allow_no_commits = True
 
         try:
             self.selected_branches.append(self.repo.active_branch.name)
@@ -40,10 +41,14 @@ def populate_zones(self, firstColumnFileNames, secondColumnFileNames, thirdColum
                     if name == x.a_path:
                         thirdColumnFileNames.add(x.a_path)
                         secondColumnArrowMap[x.a_path] = Arrow(stroke_width=3, color=self.scene.fontColor)
-
-        for y in self.repo.index.diff("HEAD"):
-            if "git-sim_media" not in y.a_path:
-                thirdColumnFileNames.add(y.a_path)
+        try:
+            for y in self.repo.index.diff("HEAD"):
+                if "git-sim_media" not in y.a_path:
+                    thirdColumnFileNames.add(y.a_path)
+        except git.exc.BadName:
+            for (y, _stage), entry in self.repo.index.entries.items():
+                if "git-sim_media" not in y:
+                    thirdColumnFileNames.add(y)
 
         for z in self.repo.untracked_files:
             if "git-sim_media" not in z:

git_sim/git_sim_base_command.py

@@ -23,6 +23,7 @@ def __init__(self, scene):
         self.hide_first_tag = False
         self.stop = False
         self.zone_title_offset = 2.6 if platform.system() == "Windows" else 2.6
+        self.allow_no_commits = False
 
         self.logo = ImageMobject(self.scene.args.logo)
         self.logo.width = 3
@@ -43,8 +44,14 @@ def execute(self):
 
     def get_commits(self, start="HEAD"):
         if not self.numCommits:
-            print("git-sim error: No commits in current Git repository.")
-            sys.exit(1)
+            if self.allow_no_commits:
+                self.numCommits = self.defaultNumCommits
+                self.commits = ["dark"]*5
+                self.zone_title_offset = 2
+                return
+            else:
+                print("git-sim error: No commits in current Git repository.")
+                sys.exit(1)
 
         try:
             self.commits = list(self.repo.iter_commits(start)) if self.numCommits == 1 else list(self.repo.iter_commits(start + "~" + str(self.numCommits) + "..." + start))
@@ -416,9 +423,14 @@ def populate_zones(self, firstColumnFileNames, secondColumnFileNames, thirdColum
             if "git-sim_media" not in x.a_path:
                 secondColumnFileNames.add(x.a_path)
 
-        for y in self.repo.index.diff("HEAD"):
-            if "git-sim_media" not in y.a_path:
-                thirdColumnFileNames.add(y.a_path)
+        try:
+            for y in self.repo.index.diff("HEAD"):
+                if "git-sim_media" not in y.a_path:
+                    thirdColumnFileNames.add(y.a_path)
+        except git.exc.BadName:
+            for (y, _stage), entry in self.repo.index.entries.items():
+                if "git-sim_media" not in y:
+                    thirdColumnFileNames.add(y)
 
         for z in self.repo.untracked_files:
             if "git-sim_media" not in z:

git_sim/git_sim_status.py

@@ -7,6 +7,7 @@ def __init__(self, scene):
         super().__init__(scene)
         self.maxrefs = 2
         self.hide_first_tag = True
+        self.allow_no_commits = True
 
         try:
             self.selected_branches.append(self.repo.active_branch.name)