Mercurial > hg > mercurial
annotate mercurial-common/src/python/load-substates-command-46.py @ 1123:7f82951e9391 development/2023.11.x tip
merge default to development/2023.11.x branch
| author | Nadia Burnasheva <nadezhda.burnasheva@jetbrains.com> |
|---|---|
| date | Wed, 18 Oct 2023 13:30:24 +0200 |
| parents | e7df2ef5064b |
| children |
| rev | line source |
|---|---|
| 1007 | 1 #!/usr/bin/env python |
| 2 ## | |
| 3 ## Copyright 2000-2014 JetBrains | |
| 4 ## | |
| 5 ## Licensed under the Apache License, Version 2.0 (the "License"); | |
| 6 ## you may not use this file except in compliance with the License. | |
| 7 ## You may obtain a copy of the License at | |
| 8 ## | |
| 9 ## http://www.apache.org/licenses/LICENSE-2.0 | |
| 10 ## | |
| 11 ## Unless required by applicable law or agreed to in writing, software | |
| 12 ## distributed under the License is distributed on an "AS IS" BASIS, | |
| 13 ## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| 14 ## See the License for the specific language governing permissions and | |
| 15 ## limitations under the License. | |
| 16 ## | |
| 17 ## | |
| 18 ## http://www.gnu.org/licenses/gpl-faq.html#GPLModuleLicense | |
| 19 ## http://www.gnu.org/licenses/license-list.html#apache2 | |
| 20 ## http://en.wikipedia.org/wiki/Apache_License#GPL_compatibility | |
| 21 ## | |
| 22 ## | |
| 23 """ | |
| 24 load-substates-command | |
| 25 """ | |
| 26 | |
| 27 import base64 | |
| 28 from mercurial import util, node | |
| 29 from mercurial.utils import dateutil | |
| 30 from threading import Thread | |
| 31 | |
| 32 | |
| 33 def load_substates_command(ui, repo, outputFile, **opts): | |
| 34 """Tons of docs""" | |
| 35 | |
| 36 ui.write("Fetching commits...") | |
| 37 | |
| 38 NONE = "=====" | |
| 39 | |
| 40 def b64(x): | |
| 41 if x is None or x == "": | |
| 42 return NONE | |
| 43 return base64.b64encode( x ) | |
| 44 | |
| 45 def fetch_commits(): | |
| 46 ui.write("Iterating over commits...\n") | |
| 47 with open(outputFile + ".commits", "w", 5 * 1024 * 1024) as result: | |
| 48 result.write("format: prefix commitID commitHash num_parents parent branch num_tags tag user message date [.hgsub] [.hgsubstate]\n") | |
| 49 result.flush() | |
| 50 | |
| 51 commit_to_substates = {} | |
| 52 | |
| 53 def update_sub_states(ctx): | |
| 54 def filenode(ctx, filename, i): | |
| 55 if filename in ctx.files(): | |
| 56 try: | |
| 57 return node.hex(ctx.filenode(filename)) | |
| 58 except: | |
| 59 # file could have been deleted => so there would be no filenode for it | |
| 60 # this also means we should avoid parents as file source | |
| 61 return NONE | |
| 62 else: | |
| 63 for p in ctx.parents(): | |
| 64 if commit_to_substates.has_key(p.hex()): | |
| 65 v = commit_to_substates[p.hex()][i] | |
| 66 if v != NONE: | |
| 67 return v | |
| 68 return NONE | |
| 69 | |
| 70 best_sub = filenode(ctx, ".hgsub", 0) | |
| 71 best_state = filenode(ctx, ".hgsubstate", 1) | |
| 72 commit_to_substates[ctx.hex()] = (best_sub, best_state) | |
| 73 return best_sub, best_state | |
| 74 | |
| 75 for r in list(repo.changelog): | |
| 76 ctx = repo[r] | |
| 77 | |
| 78 result.write("$$@@@@ ") # magic | |
| 79 result.write( str( ctx.rev() ) ) # commit Num | |
| 80 result.write(" ") | |
| 81 result.write( ctx.hex() ) # commit ID | |
| 82 result.write(" ") | |
| 83 result.write( str( len( ctx.parents()) ) ) # num parents | |
| 84 | |
| 85 for p in ctx.parents(): # parents | |
| 86 result.write(" ") | |
| 87 result.write(p.hex()) | |
| 88 | |
| 89 result.write(" ") | |
| 90 result.write( b64( ctx.branch() ) ) # commit branch | |
| 91 | |
| 92 result.write(" ") | |
| 93 result.write( str( len( ctx.tags() ) ) ) # num tags | |
| 94 | |
| 95 for tag in ctx.tags(): # tags | |
| 96 result.write(" ") | |
| 97 result.write( b64 ( tag ) ) | |
| 98 | |
| 99 result.write(" ") # user | |
| 100 result.write( b64( ctx.user() ) ) | |
| 101 | |
| 102 result.write(" ") # message | |
| 103 result.write( b64( ctx.description() ) ) | |
| 104 | |
| 105 result.write(" ") # date | |
| 106 result.write( dateutil.datestr( ctx.date(), "%Y-%m-%dZ%H:%M:%ST%1%2") ) | |
| 107 | |
| 108 #resolve sub-repo mounts | |
| 109 (sub_node, state_node) = update_sub_states(ctx) | |
| 110 result.write(" " + sub_node + " " + state_node) | |
| 111 result.write("\n") | |
| 112 | |
| 113 ui.write("Commits iteration completed") | |
| 114 | |
| 115 def fetch_file_revisions(filename): | |
| 116 ui.write("Fetching revisions of " + filename + " file\n") | |
| 117 with open(outputFile + filename, "w", 5 * 1024 * 1024) as result: | |
| 118 result.write("format: prefix commitID base64(" + filename + ")\n") | |
| 119 result.flush() | |
| 120 | |
| 121 log = repo.file(filename) | |
| 122 for r in log: | |
| 123 result.write("$$@@@@ " + node.hex(log.node(r)) + " " + b64(log.read(r)) + "\n") | |
| 124 | |
| 125 ui.write("All revisions of file " + filename + " are fetched\n") | |
| 126 | |
| 127 tasks = [ | |
| 128 Thread(target=fetch_commits, args=[], name="Fetch commits graph"), | |
| 129 Thread(target=fetch_file_revisions, args=[".hgsub"], name="Fetch .hgsub"), | |
| 130 Thread(target=fetch_file_revisions, args=[".hgsubstate"], name="Fetch .hgsubstate"), | |
| 131 ] | |
| 132 | |
| 133 for task in tasks: | |
| 134 task.start() | |
| 135 | |
| 136 for task in tasks: | |
| 137 task.join() | |
| 138 | |
| 139 ui.write("\n##Completed##\n") | |
| 140 | |
| 141 | |
| 142 #so here goes command registration and options | |
| 143 cmdtable = { | |
|
1044
e7df2ef5064b
fixed extensions for recent hg versions, should fix tests
victory.bedrosova
parents:
1007
diff
changeset
|
144 b"load-substates": (load_substates_command, [ ], b" [options] OUTPUT_FILE") |
| 1007 | 145 } |
| 146 | |
| 147 load_substates_command.norepo = False | |
| 148 load_substates_command.optionalrepo = False | |
| 149 load_substates_command.inferrepo = False | |
| 150 load_substates_command.intents = None | |
| 151 | |
|
1044
e7df2ef5064b
fixed extensions for recent hg versions, should fix tests
victory.bedrosova
parents:
1007
diff
changeset
|
152 testedwith = '5.6.1' |
| 1007 | 153 buglink = "upsource-support@jetbrains.com" |
| 154 |