def do_save_to_csv(self, params): ''' Saves specified file to csv. [command_line] input_file output_file Author: Peter ''' # print(params, type(params)) input_file = [] if params == '': params = 'plants.py output.csv' args = params.split(' ') # print(args) if len(args) >= 1: input_file.append(args[0]) output_file = 'output.csv' if len(args) >= 2: output_file = args[1] if input_file[0].endswith('.py'): fileprocessor = model.FileProcessor() fileprocessor.process_files(input_file) modules = fileprocessor.get_modules() # print(modules) csv_writer = csv.CSV_handler() if csv_writer.write_csv_file(modules, output_file): print('File successfully saved as {}'.format(output_file))
def load_csv_for_uml(self, input_file='class_data.csv'): if os.path.isfile(input_file) is True: csvloader = csv.CSV_handler() module = csvloader.open_file(input_file) if module is not False: makediagram = UMLout.MakeUML(True, True) return makediagram.create_class_diagram(module) else: return False
def test_03_compare_output(self): # Compares plugin output with output generated by model data_for_model = ['linkedlist.py'] newModelData = model.FileProcessor() newModelData.process_files(data_for_model) model_data_module = newModelData.get_modules() testclass = csv_plugin.CSV_handler() expected = testclass.write_csv_file(model_data_module, 'testdatafile02.csv') test_module = testclass.open_file('test_data_file01.csv')
def test_02_false_input(self): testclass = csv_plugin.CSV_handler() expected = False actual = testclass.open_file('dreaming.csv') self.assertEqual( expected, actual, "Expected {} got {}".format( expected, actual))
def create_csv(self, in_filename, out_file='class_data.csv'): if type(in_filename) != list: temp_filename = in_filename in_filename = [] in_filename.append(temp_filename) modules = self.run_parser(in_filename) csvhandler = csv.CSV_handler() if csvhandler.write_csv_file(modules, out_file) is True: return True else: return False
def test_04_compare_csv_with_output_file(self): #compares a file already in csv form with a newly generated file based on the original import filecmp testclass = csv_plugin.CSV_handler() input_file = 'linkedlist.csv' output_file = 'test_compare_file.csv' module = testclass.open_file(input_file) testclass.write_csv_file(module, output_file) expected = True actual = filecmp.cmp(input_file, output_file, True) self.assertEqual(expected, actual)
def do_load_csv_for_uml(self, params): ''' Loads csv file and creates UML diagram [command line] [file.csv] Author: Peter ''' if params == '': params = 'output.csv' args = params.split(' ') print(args) if len(args) >= 1: input_file = args[0] if input_file.endswith('.csv'): csvloader = csv.CSV_handler() module = csvloader.open_file(input_file) makediagram = uml_out.MakeUML(True, True) if makediagram.create_class_diagram(module): print("{} successfully converted to UML class diagram".format( input_file))
def test_01_true_input(self): testclass = csv_plugin.CSV_handler() example_type = dict() expected = type(example_type) actual = type(testclass.open_file('linkedlist.csv')) self.assertEqual(expected, actual)