Mercurial > dotfiles

annotate lib/hg/hggit_sync.py @ 295:e40cdce29e5a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Hacky hg-git sync script.
author Ludovic Chabant <ludovic@chabant.com>
date Thu, 02 Apr 2015 08:00:57 -0700
parents
children e7fe049b7f8b
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
295
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
1 import os
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
2 import os.path
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
3 import sys
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
4 import subprocess
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
5
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
6
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
7 class CommitInfo(object):
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
8 def __init__(self):
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
9 self.nodeid = None
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
10 self.timestamp = None
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
11 self.description = None
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
12
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
13
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
14 def parse_commits(text):
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
15 commits = []
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
16 cur_commit = None
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
17 for line in text.split('\n'):
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
18 if line == '':
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
19 if cur_commit:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
20 commits.append(cur_commit)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
21 cur_commit = None
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
22 continue
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
23 if cur_commit is None:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
24 cur_commit = CommitInfo()
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
25 if cur_commit.nodeid is None:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
26 id_and_date = line.split(' ', 1)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
27 cur_commit.nodeid = id_and_date[0]
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
28 cur_commit.timestamp = int(id_and_date[1].split('.')[0])
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
29 else:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
30 cur_commit.description = line
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
31 return commits
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
32
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
33
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
34 def build_commit_map(commits1, commits2):
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
35 commits1 = sorted(commits1, key=lambda c: c.timestamp)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
36 commits2 = sorted(commits2, key=lambda c: c.timestamp)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
37 commit_map = dict(map(lambda c: (c.timestamp, (c, None)), commits1))
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
38 for c in commits2:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
39 entry = commit_map.get(c.timestamp, (None, None))
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
40 entry = (entry[0], c)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
41 commit_map[c.timestamp] = entry
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
42 return commit_map
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
43
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
44
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
45 def main():
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
46 hg_repo = os.getcwd()
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
47 if not os.path.exists(os.path.join(hg_repo, '.hg')):
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
48 print "You must run this in the root of a Mercurial repository."
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
49 return 1
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
50
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
51 git_repo = os.path.join(hg_repo, '.hg', 'git')
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
52 if not os.path.exists(git_repo):
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
53 print ("This Mercurial repository doesn't seem to have any Git mirror "
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
54 "to sync with.")
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
55 return 1
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
56
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
57 hg_output = subprocess.check_output([
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
58 'hg', 'log',
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
59 '--template', "{node} {date}\n{firstline(desc)}\n\n"])
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
60 hg_commits = parse_commits(hg_output)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
61
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
62 os.chdir(git_repo)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
63 git_output = subprocess.check_output([
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
64 'git', 'log', '--format=%H %ct%n%s%n%n'])
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
65 git_commits = parse_commits(git_output)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
66 os.chdir(hg_repo)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
67
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
68 commit_map = build_commit_map(git_commits, hg_commits)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
69 for key, val in commit_map.iteritems():
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
70 if val[0] is None:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
71 print ("Mercurial commit '%s' (%s) has no Git mirror yet: %s" %
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
72 (val[1].nodeid, val[1].timestamp, val[1].description))
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
73 if val[1] is None:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
74 print ("Git commit '%s' (%s) is new: %s" %
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
75 (val[0].nodeid, val[0].timestamp, val[0].description))
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
76
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
77 map_file = os.path.join(hg_repo, '.hg', 'git-mapfile')
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
78 if len(sys.argv) > 1:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
79 map_file = sys.argv[1]
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
80 print "Saving map file: %s" % map_file
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
81 with open(map_file, 'w') as fp:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
82 for key, val in commit_map.iteritems():
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
83 if val[0] is None or val[1] is None:
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
84 continue
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
85 fp.write(val[0].nodeid)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
86 fp.write(' ')
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
87 fp.write(val[1].nodeid)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
88 fp.write('\n')
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
89
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
90
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
91 if __name__ == '__main__':
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
92 res = main()
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
93 sys.exit(res)
e40cdce29e5a Hacky hg-git sync script.
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
94