OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
3 # Use of this source code is governed by a BSD-style license that can be | |
4 # found in the LICENSE file. | |
5 | |
6 '''The 'grit sdiff' tool. | |
7 ''' | |
8 | |
9 import os | |
10 import getopt | |
11 import tempfile | |
12 | |
13 from grit.node import structure | |
14 from grit.tool import interface | |
15 | |
16 from grit import constants | |
17 from grit import util | |
18 | |
19 # Builds the description for the tool (used as the __doc__ | |
20 # for the DiffStructures class). | |
21 _class_doc = """\ | |
22 Allows you to view the differences in the structure of two files, | |
23 disregarding their translateable content. Translateable portions of | |
24 each file are changed to the string "TTTTTT" before invoking the diff program | |
25 specified by the P4DIFF environment variable. | |
26 | |
27 Usage: grit sdiff [-t TYPE] [-s SECTION] [-e ENCODING] LEFT RIGHT | |
28 | |
29 LEFT and RIGHT are the files you want to diff. SECTION is required | |
30 for structure types like 'dialog' to identify the part of the file to look at. | |
31 ENCODING indicates the encoding of the left and right files (default 'cp1252'). | |
32 TYPE can be one of the following, defaults to 'tr_html': | |
33 """ | |
34 for gatherer in structure._GATHERERS: | |
35 _class_doc += " - %s\n" % gatherer | |
36 | |
37 | |
38 class DiffStructures(interface.Tool): | |
39 __doc__ = _class_doc | |
40 | |
41 def __init__(self): | |
42 self.section = None | |
43 self.left_encoding = 'cp1252' | |
44 self.right_encoding = 'cp1252' | |
45 self.structure_type = 'tr_html' | |
46 | |
47 def ShortDescription(self): | |
48 return 'View differences without regard for translateable portions.' | |
49 | |
50 def Run(self, global_opts, args): | |
51 (opts, args) = getopt.getopt(args, 's:e:t:', | |
52 ['left_encoding=', 'right_encoding=']) | |
53 for key, val in opts: | |
54 if key == '-s': | |
55 self.section = val | |
56 elif key == '-e': | |
57 self.left_encoding = val | |
58 self.right_encoding = val | |
59 elif key == '-t': | |
60 self.structure_type = val | |
61 elif key == '--left_encoding': | |
62 self.left_encoding = val | |
63 elif key == '--right_encoding': | |
64 self.right_encoding == val | |
65 | |
66 if len(args) != 2: | |
67 print "Incorrect usage - 'grit help sdiff' for usage details." | |
68 return 2 | |
69 | |
70 if 'P4DIFF' not in os.environ: | |
71 print "Environment variable P4DIFF not set; defaulting to 'windiff'." | |
72 diff_program = 'windiff' | |
73 else: | |
74 diff_program = os.environ['P4DIFF'] | |
75 | |
76 left_trans = self.MakeStaticTranslation(args[0], self.left_encoding) | |
77 try: | |
78 try: | |
79 right_trans = self.MakeStaticTranslation(args[1], self.right_encoding) | |
80 | |
81 os.system('%s %s %s' % (diff_program, left_trans, right_trans)) | |
82 finally: | |
83 os.unlink(right_trans) | |
84 finally: | |
85 os.unlink(left_trans) | |
86 | |
87 def MakeStaticTranslation(self, original_filename, encoding): | |
88 """Given the name of the structure type (self.structure_type), the filename | |
89 of the file holding the original structure, and optionally the "section" key | |
90 identifying the part of the file to look at (self.section), creates a | |
91 temporary file holding a "static" translation of the original structure | |
92 (i.e. one where all translateable parts have been replaced with "TTTTTT") | |
93 and returns the temporary file name. It is the caller's responsibility to | |
94 delete the file when finished. | |
95 | |
96 Args: | |
97 original_filename: 'c:\\bingo\\bla.rc' | |
98 | |
99 Return: | |
100 'c:\\temp\\werlkjsdf334.tmp' | |
101 """ | |
102 original = structure._GATHERERS[self.structure_type](original_filename, | |
103 extkey=self.section, | |
104 encoding=encoding) | |
105 original.Parse() | |
106 translated = original.Translate(constants.CONSTANT_LANGUAGE, False) | |
107 | |
108 fname = tempfile.mktemp() | |
109 with util.WrapOutputStream(open(fname, 'w')) as writer: | |
110 writer.write("Original filename: %s\n=============\n\n" | |
111 % original_filename) | |
112 writer.write(translated) # write in UTF-8 | |
113 | |
114 return fname | |
OLD | NEW |