more places to use yaml.safe_load

12 years ago · 04d3834a71
parent dd085c7e9f
commit 04d3834a71
1 changed files with 1 additions and 1 deletions
--- a/2
+++ b/2
@ -160,7 +160,7 @@ def get_head_branch(module, dest, remote):
    # Check if the .git is a file. If it is a file, it means that we are in a submodule structure.
    if os.path.isfile(repo_path):
        try:
-            gitdir = yaml.load(open(repo_path)).get('gitdir')
+            gitdir = yaml.safe_load(open(repo_path)).get('gitdir')
            # There is a posibility the .git file to have an absolute path.
            if os.path.isabs(gitdir):
                repo_path = gitdir