PEP8 compliant stubs

QuantConnectStubsGenerator/Model/Property.cs

@@ -3,11 +3,15 @@ namespace QuantConnectStubsGenerator.Model
     public class Property
     {
         public string Name { get; }
+
         public PythonType Type { get; set; }
 
         public bool Static { get; set; }
+
         public bool Abstract { get; set; }
 
+        public bool Constant { get; set; }
+
         public string Value { get; set; }
 
         public string Summary { get; set; }

QuantConnectStubsGenerator/Parser/PropertyParser.cs

@@ -57,6 +57,7 @@ public override void VisitEnumMemberDeclaration(EnumMemberDeclarationSyntax node
                     : _currentClass.Properties.Count.ToString(),
                 Static = true,
                 Abstract = _currentClass.Interface || HasModifier(node, "abstract"),
+                Constant = true,
                 DeprecationReason = GetDeprecationReason(node)
             };
 
@@ -169,6 +170,7 @@ private void VisitField(BaseFieldDeclarationSyntax node, PythonType type)
                     Type = type,
                     Static = _currentClass.Static || HasModifier(node, "static") || HasModifier(node, "const"),
                     Abstract = _currentClass.Interface || HasModifier(node, "abstract"),
+                    Constant = HasModifier(node, "const") || HasModifier(node, "readonly"),
                     DeprecationReason = GetDeprecationReason(node)
                 };
 

QuantConnectStubsGenerator/QuantConnectStubsGenerator.csproj

@@ -1,16 +1,17 @@
-<Project Sdk="Microsoft.NET.Sdk">
-    <PropertyGroup>
-        <OutputType>Exe</OutputType>
-        <TargetFramework>net6.0</TargetFramework>
-    </PropertyGroup>
-    <ItemGroup>
-        <PackageReference Include="log4net" Version="2.0.12" />
-        <PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.6.0" />
-        <PackageReference Include="QuikGraph" Version="2.3.0" />
-    </ItemGroup>
-    <ItemGroup>
-        <None Update="log4net.config">
-            <CopyToOutputDirectory>Always</CopyToOutputDirectory>
-        </None>
-    </ItemGroup>
+<Project Sdk="Microsoft.NET.Sdk">
+	<PropertyGroup>
+		<OutputType>Exe</OutputType>
+		<TargetFramework>net6.0</TargetFramework>
+	</PropertyGroup>
+	<ItemGroup>
+		<PackageReference Include="log4net" Version="2.0.12" />
+		<PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.6.0" />
+		<PackageReference Include="QuikGraph" Version="2.3.0" />
+		<PackageReference Include="QuantConnect.pythonnet" Version="2.0.*" />
+	</ItemGroup>
+	<ItemGroup>
+		<None Update="log4net.config">
+			<CopyToOutputDirectory>Always</CopyToOutputDirectory>
+		</None>
+	</ItemGroup>
 </Project>

QuantConnectStubsGenerator/Renderer/MethodRenderer.cs

@@ -1,6 +1,7 @@
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
+using Python.Runtime;
 using QuantConnectStubsGenerator.Model;
 using QuantConnectStubsGenerator.Utility;
 
@@ -14,6 +15,13 @@
 
         public override void Render(Method method)
         {
+            var snakeCasedMethod = GetSnakeCasedMethod(method);
+            if (snakeCasedMethod.Name != method.Name)
+            {
+                Render(snakeCasedMethod);
+                return;
+            }
+
             if (method.Static)
             {
                 WriteLine("@staticmethod");
@@ -82,5 +90,28 @@
 
             return str;
         }
+
+        private static Method GetSnakeCasedMethod(Method method)
+        {
+            var snakeCasedMethod = new Method(method.Name.ToSnakeCase(), method.ReturnType)
+            {
+                Static = method.Static,
+                Overload = method.Overload,
+                Summary = method.Summary,
+                File = method.File,
+                DeprecationReason = method.DeprecationReason
+            };
+
+            foreach (var parameter in method.Parameters)
+            {
+                snakeCasedMethod.Parameters.Add(new Parameter(parameter.Name.ToSnakeCase(), parameter.Type)
+                {
+                    VarArgs = parameter.VarArgs,
+                    Value = parameter.Value
+                });
+            }
+
+            return snakeCasedMethod;
+        }
     }
 }

QuantConnectStubsGenerator/Renderer/PropertyRenderer.cs

@@ -1,4 +1,5 @@
 using System.IO;
+using Python.Runtime;
 using QuantConnectStubsGenerator.Model;
 using QuantConnectStubsGenerator.Utility;
 
@@ -12,6 +13,13 @@
 
         public override void Render(Property property)
         {
+            var snakeCasedProperty = GetSnakeCasedProperty(property);
+            if (snakeCasedProperty.Name != property.Name)
+            {
+                Render(snakeCasedProperty);
+                return;
+            }
+
             if (property.Static)
             {
                 RenderAttribute(property);
@@ -22,6 +30,26 @@
             }
         }
 
+        private static Property GetSnakeCasedProperty(Property property)
+        {
+            var name = property.Name.ToSnakeCase();
+            if (property.Constant)
+            {
+                name = name.ToUpper();
+            }
+
+            return new Property(name)
+            {
+                Type = property.Type,
+                Static = property.Static,
+                Abstract = property.Abstract,
+                Constant = property.Constant,
+                Value = property.Value,
+                Summary = property.Summary,
+                DeprecationReason = property.DeprecationReason
+            };
+        }
+
         private void RenderAttribute(Property property)
         {
             // Some attributes have names in C# that are illegal in Python