def generate_class(gen_dict, class_name, path_to_gen):
"""generate the corresponding class file (erase the previous code)
Parameters
----------
gen_dict : dict
Dict with key = class name and value = class dict (name, package, properties, methods...)
class_name : str
name of the class to generate
path_to_gen : str
path to the file to save the class code
Returns
-------
None
"""
class_dict = gen_dict[class_name]
class_pack = class_dict["package"]
# Start of file generation
# Create or Erase the file by opening it
class_file = open_co(join(path_to_gen, class_name + ".py"), "w", "utf-8")
# List of non python type to import
import_type_list = list(find_import_type(gen_dict, class_dict, []))
# Encoding
class_file.write("# -*- coding: utf-8 -*-\n")
# Warning
class_file.write('"""Warning : this file has been generated, '
'you shouldn\'t edit it"""\n\n')
# Import
class_file.write("from os import linesep\n")
if "ndarray" in import_type_list:
class_file.write("from pyleecan.Classes.check import set_array, " +
"check_init_dict, check_var\n")
else:
class_file.write(
"from pyleecan.Classes.check import check_init_dict, " +
"check_var\n")
# Save function
class_file.write("from pyleecan.Functions.save import save\n")
# Import of the mother_class (FrozenClass by default)
# All the classes file are in the Classes folder (regardless of their main package)
if class_dict["mother"] != "":
class_file.write("from pyleecan.Classes." + class_dict["mother"] +
" import " + class_dict["mother"] + "\n\n")
else:
class_file.write("from pyleecan.Classes.frozen import FrozenClass\n\n")
# Import all the methods of the class
# The methods are in Methods.<Main package>.<class name>, one file per method
for meth in class_dict["methods"]:
class_file.write("from pyleecan.Methods." + class_pack + "." +
class_name + "." + meth + " import " +
meth.split(".")[-1] + "\n")
if len(class_dict["methods"]) > 0:
class_file.write("\n")
# For Matrix and Vector (numpy) property
if "ndarray" in import_type_list:
class_file.write("from numpy import array, array_equal\n")
import_type_list.remove("ndarray")
# Import of all needed pyleecan type for property polymorphism
class_file.write(
"from pyleecan.Classes.check import InitUnKnowClassError\n")
for pyleecan_type in import_type_list:
type_dict = gen_dict[pyleecan_type]
class_file.write("from pyleecan.Classes." + pyleecan_type +
" import " + pyleecan_type + "\n")
for daughter_name in type_dict["daughters"]:
# Import of the inherited class (for init)
class_file.write("from pyleecan.Classes." + daughter_name +
" import " + daughter_name + "\n")
if len(import_type_list) > 0:
class_file.write("\n")
# Class declaration
if class_dict["mother"] != "":
class_file.write("\n\nclass " + class_name + "(" +
class_dict["mother"] + "):\n")
else:
class_file.write("\n\nclass " + class_name + "(FrozenClass):\n")
# Class Docstring
if class_dict["desc"] != "":
class_file.write(TAB + '"""' + class_dict["desc"] + '"""\n')
class_file.write("\n")
# Declare all class Constante (VERSION should be a constante for every classes)
for cst in class_dict["constants"]:
class_file.write(TAB + cst["name"] + " = " + str(cst["value"]) + "\n")
class_file.write("\n")
# Asign all the Methods of the class
for meth in class_dict["methods"]:
class_file.write(TAB + "# cf Methods." + class_pack + "." +
class_name + "." + meth + "\n")
class_file.write(TAB + meth.split(".")[-1] + " = " +
meth.split(".")[-1] + "\n")
class_file.write(TAB + "# save method is available in all object\n")
class_file.write(TAB + "save = save\n\n")
# Add the __init__ method
if len(class_dict["properties"]) == 0 and class_dict["mother"] == "":
class_file.write(generate_init_void() + "\n")
else:
class_file.write(generate_init(gen_dict, class_dict) + "\n")
# Add the __str__ method
class_file.write(generate_str(gen_dict, class_dict) + "\n")
# Add the __eq__ method
class_file.write(generate_eq(gen_dict, class_dict) + "\n")
# Add the as_dict method
class_file.write(generate_as_dict(gen_dict, class_dict) + "\n")
# Add the _set_None method
class_file.write(generate_set_None(gen_dict, class_dict) + "\n")
# Add all the properties getter and setter
class_file.write(generate_properties(gen_dict, class_dict) + "\n")
# End of class generation
class_file.close()
def generate_class(gen_dict, class_name, path_to_gen):
"""generate the corresponding class file (erase the previous code)
Parameters
----------
gen_dict : dict
Dict with key = class name and value = class dict (name, package, properties, methods...)
class_name : str
name of the class to generate
path_to_gen : str
path to the file to save the class code
Returns
-------
None
"""
class_dict = gen_dict[class_name]
class_pack = class_dict["package"]
# Start of file generation
# Create or Erase the file by opening it
class_file = open_co(join(path_to_gen, class_name + ".py"), "w", "utf-8")
# List of non python type to import
import_type_list = list(find_import_type(gen_dict, class_dict, []))
if class_name in import_type_list: # For recursive class
import_type_list.remove(class_name)
# Encoding
class_file.write("# -*- coding: utf-8 -*-\n")
# Warning
class_file.write('"""File generated according to ' + class_dict["path"] +
"\n")
if class_dict["is_internal"]:
class_file.write(
"WARNING! Internal version of the class: DO NOT SHARE ON GITHUB !\n"
)
class_file.write(
'WARNING! All changes made in this file will be lost!\n"""\n\n')
# Import
class_file.write("from os import linesep\n")
if "ndarray" in import_type_list:
class_file.write("from pyleecan.Classes.check import set_array, " +
"check_init_dict, check_var, raise_\n")
else:
class_file.write(
"from pyleecan.Classes.check import check_init_dict, check_var, raise_\n"
)
# Save function
class_file.write("from pyleecan.Functions.save import save\n")
# Import of the mother_class (FrozenClass by default)
# All the classes file are in the Classes folder (regardless of their main package)
if class_dict["mother"] != "":
class_file.write("from pyleecan.Classes." + class_dict["mother"] +
" import " + class_dict["mother"] + "\n\n")
else:
class_file.write("from pyleecan.Classes.frozen import FrozenClass\n\n")
# Import all the methods of the class
# The methods are in Methods.<Main package>.<class name>, one file per method
if len(class_dict["methods"]) > 0:
class_file.write("# Import all class method\n")
class_file.write(
"# Try/catch to remove unnecessary dependencies in unused method\n"
)
for meth in class_dict["methods"]:
class_file.write(import_method(class_pack, class_name, meth))
if len(class_dict["methods"]) > 0:
class_file.write("\n")
# For Matrix and Vector (numpy) property
if "ndarray" in import_type_list:
class_file.write("from numpy import array, array_equal\n")
import_type_list.remove("ndarray")
# Import of all needed pyleecan type for empty init
class_file.write(
"from pyleecan.Classes.check import InitUnKnowClassError\n")
for pyleecan_type in import_type_list:
class_file.write("from pyleecan.Classes." + pyleecan_type +
" import " + pyleecan_type + "\n")
# Class declaration
if class_dict["mother"] != "":
class_file.write("\n\nclass " + class_name + "(" +
class_dict["mother"] + "):\n")
else:
class_file.write("\n\nclass " + class_name + "(FrozenClass):\n")
# Class Docstring
if class_dict["desc"] != "":
class_file.write(TAB + '"""' + class_dict["desc"] + '"""\n')
class_file.write("\n")
# Declare all class Constante (VERSION should be a constante for every classes)
for cst in class_dict["constants"]:
class_file.write(TAB + cst["name"] + " = " + str(cst["value"]) + "\n")
class_file.write("\n")
# Asign all the Methods of the class
if len(class_dict["methods"]) > 1:
class_file.write(
TAB +
"# Check ImportError to remove unnecessary dependencies in unused method\n"
)
for meth in class_dict["methods"]:
meth_name = meth.split(".")[-1]
class_file.write(TAB + "# cf Methods." + class_pack + "." +
class_name + "." + meth + "\n")
class_file.write(TAB + "if isinstance(" + meth_name +
", ImportError):\n")
class_file.write(TAB2 + meth_name + " = property(\n")
class_file.write(TAB3 + "fget=lambda x: raise_(\n")
# PEP8 formating
if len(class_name) + 2 * len(meth_name) > 39:
# 2 lines Import text
class_file.write(TAB4 + "ImportError(\n")
class_file.write(TAB5 + """"Can't use """ + class_name +
" method " + meth_name + ': "\n')
class_file.write(TAB5 + "+ str(" + meth_name + ")\n")
class_file.write(TAB4 + ")\n")
elif len(class_name) + 2 * len(meth_name) > 29:
# Import text on line different line
class_file.write(TAB4 + "ImportError(\n")
class_file.write(TAB5 + """"Can't use """ + class_name +
" method " + meth_name + ': "')
class_file.write(" + str(" + meth_name + ")\n")
class_file.write(TAB4 + ")\n")
else: # On one line
class_file.write(TAB4 + """ImportError("Can't use """ +
class_name + " method " + meth_name +
': " + str(' + meth_name + "))\n")
class_file.write(TAB3 + ")\n")
class_file.write(TAB2 + ")\n")
class_file.write(TAB + "else:\n")
class_file.write(TAB2 + meth_name + " = " + meth_name + "\n")
class_file.write(TAB + "# save method is available in all object\n")
class_file.write(TAB + "save = save\n\n")
# Add the __init__ method
if len(class_dict["properties"]) == 0 and class_dict["mother"] == "":
class_file.write(generate_init_void() + "\n")
else:
class_file.write(generate_init(gen_dict, class_dict) + "\n")
# Add the __str__ method
class_file.write(generate_str(gen_dict, class_dict) + "\n")
# Add the __eq__ method
class_file.write(generate_eq(gen_dict, class_dict) + "\n")
# Add the as_dict method
class_file.write(generate_as_dict(gen_dict, class_dict) + "\n")
# Add the _set_None method
class_file.write(generate_set_None(gen_dict, class_dict))
# Add all the properties getter and setter
if len(class_dict["properties"]) > 0:
class_file.write("\n" + generate_properties(gen_dict, class_dict) +
"\n")
# End of class generation
class_file.close()