Added a little helper script to debug the output of the p4 python interface.
[gitweb.git] / contrib / fast-import / p4-fast-export.py
index 8455c1f41c4b0e65846e408904b0d8207d91d288..c44292473a82fd349b69176038eafdf8c177f4f6 100755 (executable)
 #       - support p4 submit (hah!)
 #
 import os, string, sys, time
-import marshal, popen2
+import marshal, popen2, getopt
 
-if len(sys.argv) != 2:
+branch = "refs/heads/p4"
+prefix = previousDepotPath = os.popen("git-repo-config --get p4.depotpath").read()
+if len(prefix) != 0:
+    prefix = prefix[:-1]
+
+try:
+    opts, args = getopt.getopt(sys.argv[1:], "", [ "branch=" ])
+except getopt.GetoptError:
+    print "fixme, syntax error"
+    sys.exit(1)
+
+for o, a in opts:
+    if o == "--branch":
+        branch = "refs/heads/" + a
+
+if len(args) == 0 and len(prefix) != 0:
+    print "[using previously specified depot path %s]" % prefix
+elif len(args) != 1:
     print "usage: %s //depot/path[@revRange]" % sys.argv[0]
     print "\n    example:"
-    print "    %s //depot/my/project/ -- to import everything"
+    print "    %s //depot/my/project/ -- to import the current head"
+    print "    %s //depot/my/project/@all -- to import everything"
     print "    %s //depot/my/project/@1,6 -- to import only from revision 1 to 6"
     print ""
     print "    (a ... is not needed in the path p4 specification, it's added implicitly)"
     print ""
     sys.exit(1)
+else:
+    if len(prefix) != 0 and prefix != args[0]:
+        print "previous import used depot path %s and now %s was specified. this doesn't work!" % (prefix, args[0])
+        sys.exit(1)
+    prefix = args[0]
 
-branch = "refs/heads/p4"
-prefix = sys.argv[1]
 changeRange = ""
+revision = ""
 users = {}
 initialParent = ""
 
-try:
+if prefix.find("@") != -1:
     atIdx = prefix.index("@")
     changeRange = prefix[atIdx:]
+    if changeRange == "@all":
+        changeRange = ""
+    elif changeRange.find(",") == -1:
+        revision = changeRange
+        changeRange = ""
     prefix = prefix[0:atIdx]
-except ValueError:
-    changeRange = ""
+elif prefix.find("#") != -1:
+    hashIdx = prefix.index("#")
+    revision = prefix[hashIdx:]
+    prefix = prefix[0:hashIdx]
+elif len(previousDepotPath) == 0:
+    revision = "#head"
 
 if prefix.endswith("..."):
     prefix = prefix[:-3]
@@ -78,11 +109,11 @@ def commit(details):
 
     gitStream.write("data <<EOT\n")
     gitStream.write(details["desc"])
-    gitStream.write("\n[ imported from %s; change %s ]\n" % (prefix, change))
+    gitStream.write("\n[ imported from %s; change %s ]\n" % (prefix, details["change"]))
     gitStream.write("EOT\n\n")
 
     if len(initialParent) > 0:
-        gitStream.write("merge %s\n" % initialParent)
+        gitStream.write("from %s\n" % initialParent)
         initialParent = ""
 
     fnum = 0
@@ -116,7 +147,7 @@ def commit(details):
 
     gitStream.write("\n")
 
-    gitStream.write("tag p4/%s\n" % change)
+    gitStream.write("tag p4/%s\n" % details["change"])
     gitStream.write("from %s\n" % branch);
     gitStream.write("tagger %s\n" % committer);
     gitStream.write("data 0\n\n")
@@ -139,43 +170,88 @@ def getUserMap():
         output = sout.read()
         tagIdx = output.index(" tags/p4/")
         caretIdx = output.index("^")
-        revision = int(output[tagIdx + 9 : caretIdx]) + 1
-        changeRange = "@%s,#head" % revision
+        rev = int(output[tagIdx + 9 : caretIdx]) + 1
+        changeRange = "@%s,#head" % rev
         initialParent = os.popen("git-rev-parse %s" % branch).read()[:-1]
     except:
         pass
 
-output = os.popen("p4 changes %s...%s" % (prefix, changeRange)).readlines()
+sys.stderr.write("\n")
 
-changes = []
-for line in output:
-    changeNum = line.split(" ")[1]
-    changes.append(changeNum)
+tz = - time.timezone / 36
+tzsign = ("%s" % tz)[0]
+if tzsign != '+' and tzsign != '-':
+    tz = "+" + ("%s" % tz)
 
-changes.reverse()
+if len(revision) > 0:
+    print "Doing initial import of %s from revision %s" % (prefix, revision)
 
-if len(changes) == 0:
-    print "no changes to import!"
-    sys.exit(1)
+    details = { "user" : "git perforce import user", "time" : int(time.time()) }
+    details["desc"] = "Initial import of %s from the state at revision %s" % (prefix, revision)
+    details["change"] = revision
+    newestRevision = 0
 
-sys.stderr.write("\n")
+    fileCnt = 0
+    for info in p4CmdList("files %s...%s" % (prefix, revision)):
+        change = int(info["change"])
+        if change > newestRevision:
+            newestRevision = change
 
-tz = - time.timezone / 36
+        if info["action"] == "delete":
+            continue
+
+        for prop in [ "depotFile", "rev", "action", "type" ]:
+            details["%s%s" % (prop, fileCnt)] = info[prop]
+
+        fileCnt = fileCnt + 1
+
+    details["change"] = newestRevision
+
+    gitOutput, gitStream, gitError = popen2.popen3("git-fast-import")
+    try:
+        commit(details)
+    except:
+        print gitError.read()
 
-gitOutput, gitStream, gitError = popen2.popen3("git-fast-import")
+    gitStream.close()
+    gitOutput.close()
+    gitError.close()
+else:
+    output = os.popen("p4 changes %s...%s" % (prefix, changeRange)).readlines()
 
-cnt = 1
-for change in changes:
-    description = p4Cmd("describe %s" % change)
+    changes = []
+    for line in output:
+        changeNum = line.split(" ")[1]
+        changes.append(changeNum)
 
-    sys.stdout.write("\rimporting revision %s (%s%%)" % (change, cnt * 100 / len(changes)))
-    sys.stdout.flush()
-    cnt = cnt + 1
+    changes.reverse()
 
-    commit(description)
+    if len(changes) == 0:
+        print "no changes to import!"
+        sys.exit(1)
 
-gitStream.close()
-gitOutput.close()
-gitError.close()
+    gitOutput, gitStream, gitError = popen2.popen3("git-fast-import")
+
+    cnt = 1
+    for change in changes:
+        description = p4Cmd("describe %s" % change)
+
+        sys.stdout.write("\rimporting revision %s (%s%%)" % (change, cnt * 100 / len(changes)))
+        sys.stdout.flush()
+        cnt = cnt + 1
+
+        try:
+            commit(description)
+        except:
+            print gitError.read()
+            sys.exit(1)
+
+    gitStream.close()
+    gitOutput.close()
+    gitError.close()
 
 print ""
+
+os.popen("git-repo-config p4.depotpath %s" % prefix).read()
+
+sys.exit(0)