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)
