@@ -46,7 +46,7 @@ def load_file(self, data_file):
4646 for idx , cell in enumerate (line_data ):
4747 self .types [header [idx ]] = self .parser .parse_type (cell ), camel_case (header [idx ])
4848
49- print ( self .types ) # TODO: Pretty print the mappings to the console
49+ self ._pretty_print_types ()
5050
5151 def generate_entity_files (self , entity_name , folder_path , create_typescript = False ):
5252 if not os .path .isdir (folder_path ):
@@ -137,7 +137,6 @@ def _generate_getters_setters(self):
137137 for key in self .types :
138138 var = self .types [key ]
139139 java_type = self .parser .translate_type (var [0 ], Language .Java )
140- print ('GETTER: Type for {0}: {1}' .format (var [1 ], java_type ))
141140 code_lines .append (GETTER_TEMPLATE .format (java_type , first_upper (var [1 ]), var [1 ]))
142141 code_lines .append (SETTER_TEMPLATE .format (first_upper (java_type ), java_type , var [1 ]))
143142 return code_lines
@@ -147,7 +146,6 @@ def _generate_definitions(self):
147146 for key in self .types :
148147 var = self .types [key ]
149148 code_lines .append (DEFINITION_TEMPLATE .format (self .parser .translate_type (var [0 ], Language .Java ), var [1 ]))
150- print (code_lines )
151149 return code_lines
152150
153151 def _generate_tostring (self , entity_name ):
@@ -157,4 +155,11 @@ def _generate_tostring(self, entity_name):
157155 variables .append (first_upper (var [1 ]))
158156 variables .append (var [1 ])
159157 inner_format_templates = ', ' .join ([TOSTRING_INNER_TEMPLATE ] * len (self .types .keys ()))
160- return TOSTRING_TEMPLATE .format (entity_name , inner_format_templates .format (* variables ))
158+ return TOSTRING_TEMPLATE .format (entity_name , inner_format_templates .format (* variables ))
159+
160+ def _pretty_print_types (self ):
161+ print ('Parsed type info:' )
162+ # \tSQL -> camelCase: type
163+ for key in self .types :
164+ var = self .types [key ]
165+ print ('\t {} -> {}: {}' .format (key , var [1 ], var [0 ]))
0 commit comments