private static void AddCorePythonTypes(PythonGraph pythonGraph, CsPyMapperDictionary typeDictionary)
{
var coreModule = pythonGraph.GetOrAddModule("@python.core");
typeDictionary[SpecialType.System_String] = coreModule.GetOrAddClass("str");
typeDictionary[SpecialType.System_Int32] = coreModule.GetOrAddClass("int");
typeDictionary[SpecialType.System_Double] = coreModule.GetOrAddClass("float");
typeDictionary[SpecialType.System_Boolean] = coreModule.GetOrAddClass("bool");
}
public static void MapAttributesToGraph(PythonGraph pythonGraph, AttributeWalker projectAttributes)
{
var typeDictionary = new CsPyMapperDictionary();
// Add core Python types
AddCorePythonTypes(pythonGraph, typeDictionary);
// Add custom Python types
foreach (var entry in projectAttributes.ClassAttributes)
{
var target = entry.Key;
var attribute = entry.Value;
var pythonClass = pythonGraph.GetOrAddModule(attribute.ModuleName).GetOrAddClass(attribute.ClassName);
typeDictionary[target] = pythonClass;
}
// Add fields
foreach (var entry in projectAttributes.FieldAttributes)
{
var target = entry.Key;
var attribute = entry.Value;
var pythonClass = typeDictionary[target.ContainingType];
var fieldName = attribute.Name ?? target.Name;
var fieldType = typeDictionary.GetPythonType(target.Type);
var pythonField = PythonField.Create(fieldName, fieldType);
pythonClass.Children.Add(fieldName, pythonField);
}
// Add methods
foreach (var entry in projectAttributes.MethodAttributes)
{
var target = entry.Key;
var attribute = entry.Value;
MapPythonFunction(target, attribute.Function, typeDictionary);
}
// Add operators
foreach (var entry in projectAttributes.OperatorAttributes)
{
var target = entry.Key;
var attribute = entry.Value;
MapPythonFunction(target, $"@operator.{attribute.Operator}", typeDictionary);
}
// Add properties
foreach (var entry in projectAttributes.PropertyAttributes)
{
var target = entry.Key;
var attribute = entry.Value;
var pythonClass = typeDictionary[target.ContainingType];
var propertyName = attribute.Name ?? target.Name;
var propertyType = typeDictionary.GetPythonType(target.Type);
var pythonProperty = PythonProperty.Create(propertyName, propertyType);
if (target.GetMethod != null)
{
var getterName = attribute.GetterFunction ?? propertyName;
var parameters = new[] { new PythonParameter("self", new PythonType(pythonClass)) };
pythonProperty.GetterFunction = PythonFunction.Create(getterName, propertyType, parameters);
}
if (target.SetMethod != null)
{
var setterName = attribute.SetterFunction ?? propertyName;
var parameters = new[] { new PythonParameter("self", new PythonType(pythonClass)),
new PythonParameter("value", propertyType) };
pythonProperty.SetterFunction = PythonFunction.Create(setterName, PythonTypes.None, parameters);
}
pythonClass.Children.Add(propertyName, pythonProperty);
}
}