private static void WriteMethods(TypeDecl decl, CodeWriter writer) { bool HasImplement(HandlerDeclareItem declare) { return(decl.Implement?.Handlers.FirstOrDefault(t => t.Name == declare.Name) == null); } var declarations = decl.Declare.Handlers; foreach (var declare in declarations) { bool isAbstract = !HasImplement(declare); if (isAbstract) { writer.AppendLine("@abstractmethod"); } var parameters = ""; foreach (var parameter in declare.Params) { parameters += ($", {parameter.Name.Underscore()}: {GetTypeHint(decl, parameter)}"); } writer.AppendLine($"def {declare.Name.Underscore()}(self{parameters}):"); writer.PushIndent(); writer.AppendLines(CommentHelper.PyComment(declare.Comment)); writer.AppendLine(isAbstract ? "pass" : "raise NotImplemented"); if (declarations.IndexOf(declare) != declarations.Count - 1) { writer.AppendNewLineWithoutIndent(); } writer.PopIndent(); } }
/// <summary> /// Generate python classes from declaration. /// </summary> public static string Build(TypeDecl decl, List <IDecl> declarations) { var writer = new CodeWriter(); string name = decl.Name; // Determine if we are inside datacentric package // based on module name. This affects the imports // and namespace use. bool insideDc = PyExtensions.GetPackage(decl) == "datacentric"; // If not generating for DataCentric package, use dc. namespace // in front of datacentric types, otherwise use no prefix string dcNamespacePrefix = insideDc ? "" : "dc."; PythonImportsBuilder.WriteImports(decl, declarations, writer); writer.AppendNewLineWithoutIndent(); writer.AppendNewLineWithoutIndent(); // Get base classes for current declaration List <string> bases = new List <string>(); if (decl.Keys.Any()) { bases.Add(dcNamespacePrefix + "Record"); } else if (decl.Inherit != null) { // Full package name and short namespace of the parent class, // or null if there is no parent bool parentClassInDifferentModule = !PyExtensions.IsPackageEquals(decl, decl.Inherit); string parentPackage = PyExtensions.GetPackage(decl.Inherit); string parentClassNamespacePrefix = parentClassInDifferentModule ? PyExtensions.GetAlias(parentPackage) + "." : ""; bases.Add(parentClassNamespacePrefix + decl.Inherit.Name); } else { bases.Add(dcNamespacePrefix + "Data"); } if (decl.Kind == TypeKind.Abstract) { bases.Add("ABC"); } // Python 3.8: // if (decl.Kind == TypeKind.Final) // writer.AppendLine("@final"); writer.AppendLine("@attr.s(slots=True, auto_attribs=True)"); writer.AppendLine($"class {name}({string.Join(", ", bases)}):"); writer.PushIndent(); writer.AppendLines(CommentHelper.PyComment(decl.Comment)); writer.AppendNewLineWithoutIndent(); if (!decl.Elements.Any()) { writer.AppendLine("pass"); } foreach (var element in decl.Elements) { // TODO: Should be replaced with callable with specific format instead of skipping string skipRepresentation = element.Vector == YesNo.Y ? ", repr=False" : ""; writer.AppendLine($"{element.Name.Underscore()}: {GetTypeHint(decl, element)} = attr.ib(default=None, kw_only=True{skipRepresentation}{GetMetaData(element)})"); writer.AppendLines(CommentHelper.PyComment(element.Comment)); if (element != decl.Elements.Last()) { writer.AppendNewLineWithoutIndent(); } } // Add to_key and create_key() methods if (decl.Keys.Any()) { var keyElements = decl.Elements.Where(e => decl.Keys.Contains(e.Name)).ToList(); writer.AppendNewLineWithoutIndent(); writer.AppendLine("def to_key(self) -> str:"); writer.PushIndent(); writer.AppendLine(CommentHelper.PyComment($"Get {decl.Name} key.")); writer.AppendLines($"return '{decl.Name}='{GetToKeyArgs(decl.Name, keyElements, true)}"); writer.PopIndent(); writer.AppendNewLineWithoutIndent(); var namedParams = keyElements.Select(e => $"{e.Name.Underscore()}: {GetTypeHint(decl, e)}").ToList(); var joinedNamedParams = string.Join(", ", namedParams); string start = "def create_key("; // Check if tokens should be separated by new line if (4 + start.Length + joinedNamedParams.Length > 120) { var indent = new string(' ', start.Length); joinedNamedParams = string.Join("," + Environment.NewLine + indent, namedParams); } writer.AppendLine("@classmethod"); writer.AppendLines($"def create_key(cls, *, {joinedNamedParams}) -> str:"); writer.PushIndent(); writer.AppendLine(CommentHelper.PyComment($"Create {decl.Name} key.")); writer.AppendLines($"return '{decl.Name}='{GetToKeyArgs(decl.Name, keyElements, false)}"); writer.PopIndent(); } if (decl.Declare != null) { writer.AppendNewLineWithoutIndent(); WriteMethods(decl, writer); } // Class end writer.PopIndent(); return(writer.ToString()); }