clone dojo if directory is not found (e.g., database restores)

pwncollege · Nov 18, 2024 · efc9593 · efc9593
1 parent 8fec2a8
commit efc9593
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/dojo_plugin/utils/dojo.py b/dojo_plugin/utils/dojo.py
@@ -445,9 +445,13 @@ def dojo_git_command(dojo, *args):
 
 
 def dojo_update(dojo):
-    dojo_git_command(dojo, "fetch", "--depth=1", "origin")
-    dojo_git_command(dojo, "reset", "--hard", "origin")
-    dojo_git_command(dojo, "submodule", "update", "--init", "--recursive")
+    if dojo.path.exists():
+        dojo_git_command(dojo, "fetch", "--depth=1", "origin")
+        dojo_git_command(dojo, "reset", "--hard", "origin")
+        dojo_git_command(dojo, "submodule", "update", "--init", "--recursive")
+    else:
+        tmpdir = dojo_clone(dojo.repository, dojo.private_key)
+        os.rename(tmpdir.name, str(dojo.path))
     return dojo_from_dir(dojo.path, dojo=dojo)