|
1 #!/usr/bin/env python |
|
2 # coding: utf-8 |
|
3 # |
|
4 # Copyright 2015 Teemu Piippo |
|
5 # All rights reserved. |
|
6 # |
|
7 # Redistribution and use in source and binary forms, with or without |
|
8 # modification, are permitted provided that the following conditions |
|
9 # are met: |
|
10 # |
|
11 # 1. Redistributions of source code must retain the above copyright |
|
12 # notice, this list of conditions and the following disclaimer. |
|
13 # 2. Redistributions in binary form must reproduce the above copyright |
|
14 # notice, this list of conditions and the following disclaimer in the |
|
15 # documentation and/or other materials provided with the distribution. |
|
16 # 3. Neither the name of the copyright holder nor the names of its |
|
17 # contributors may be used to endorse or promote products derived from |
|
18 # this software without specific prior written permission. |
|
19 # |
|
20 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
21 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED |
|
22 # TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A |
|
23 # PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER |
|
24 # OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
|
25 # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
|
26 # PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
|
27 # PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
|
28 # LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT(INCLUDING |
|
29 # NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
|
30 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
31 # |
|
32 |
|
33 import argparse |
|
34 import outputfile |
|
35 |
|
36 def main(): |
|
37 import subprocess |
|
38 from datetime import datetime |
|
39 parser = argparse.ArgumentParser(description='Writes a header file with Hg commit information') |
|
40 parser.add_argument('--cwd', default = '.') |
|
41 parser.add_argument('output') |
|
42 args = parser.parse_args() |
|
43 f = outputfile.OutputFile(args.output) |
|
44 data = subprocess.check_output(['hg', 'log', '--cwd', args.cwd, '-r.', '--template', |
|
45 '{node|short} {branch} {date|hgdate}']).decode().replace('\n', '').split(' ') |
|
46 |
|
47 rev = data[0] |
|
48 branch = data[1] |
|
49 timestamp = int(data[2]) |
|
50 date = datetime.utcfromtimestamp(timestamp) |
|
51 datestring = date.strftime('%y%m%d-%H%M') if date.year >= 2000 else '000000-0000' |
|
52 |
|
53 if len(rev) > 7: |
|
54 rev = rev[0:7] |
|
55 |
|
56 if subprocess.check_output(['hg', 'id', '--cwd', args.cwd, '-n']).decode().replace('\n', '').endswith('+'): |
|
57 rev += '+' |
|
58 |
|
59 f.write('#define HG_NODE "%s"\n' % rev) |
|
60 f.write('#define HG_BRANCH "%s"\n' % branch) |
|
61 f.write('#define HG_DATE_VERSION "%s"\n' % datestring) |
|
62 f.write('#define HG_DATE_STRING "%s"\n' % date.strftime('%d %b %Y')) |
|
63 f.write('#define HG_DATE_TIME %d\n' % int(timestamp)) |
|
64 if f.save(): |
|
65 print('%s updated to %s' %(f.filename, rev)) |
|
66 |
|
67 if __name__ == '__main__': |
|
68 main() |