add class generator

Author: MrMaydo

src/class_generator.py

@@ -0,0 +1,41 @@
+from src.header_generator import set_package, set_imports, render_javadoc
+from src.java_method_generator import generate_fields_block, generate_getters_and_setters, generate_equals, \
+    generate_hash_code
+from src.java_model import JavaClass
+
+
+def generate_java_class(java_class: JavaClass, package: str) -> str:
+    if len(java_class.fields) == 0:
+        return _empty_java_class(java_class, package)
+
+    body = [
+        set_package(package),
+        "",
+        "",
+        set_imports(java_class.fields),
+        "",
+        render_javadoc(java_class.description, indent_lvl=0),
+        f"public class {java_class.name} {{",
+        generate_fields_block(java_class.fields),
+        "",
+        generate_getters_and_setters(java_class.fields),
+        "",
+        generate_equals(java_class.name, java_class.fields),
+        "",
+        generate_hash_code(java_class.fields),
+        "}",
+        ""
+    ]
+    return "\n".join(body)
+
+
+def _empty_java_class(java_class: JavaClass, package: str) -> str:
+    body = [
+        set_package(package),
+        "",
+        render_javadoc(java_class.description, indent_lvl=0),
+        f"public class {java_class.name} {{",
+        "}",
+        ""
+    ]
+    return "\n".join(body)

src/enum_generator.py

@@ -1,8 +1,8 @@
 import re
 from typing import List
 
-from src.java_model import EnumClass, indent
 from src.header_generator import set_package, render_javadoc
+from src.java_model import EnumClass, indent
 
 
 def to_java_constant(value: str) -> str:

src/java_method_generator.py

@@ -83,7 +83,8 @@ def _render_equals_return_statement(fields: List[Field]) -> str:
         getter_name = _build_getter_name(field.name)
         end_line = ";" if i == (len(fields) - 1) else ""
         if i == 0:
-            return_statement.append(f"{indent(2)}return Objects.equals({getter_name}(), that.{getter_name}()){end_line}")
+            return_statement.append(
+                f"{indent(2)}return Objects.equals({getter_name}(), that.{getter_name}()){end_line}")
         else:
             return_statement.append(
                 f"{indent(4)}&& Objects.equals({getter_name}(), that.{getter_name}()){end_line}")

src/java_model.py

@@ -46,7 +46,3 @@ def indent(level: int, size: int = 4) -> str:
         raise ValueError("Indent size cannot be lower than 0")
 
     return " " * size * level
-
-
-
-

tests/reference_data_java_class.py

@@ -0,0 +1,114 @@
+from src.java_model import Field, JavaClass
+
+package_example = "expected.DataTypes"
+field_name_string = Field(name="name", type="String", description="This contains name.")
+field_age_int = Field(name="age", type="int", description="This contains age.")
+field_birthdate_date = Field(name="birthDate", type="Date")
+field_additionalInfo_CustomObject = Field(name="additionalInfo", type="CustomData")
+
+person_class_fields = [field_name_string, field_age_int, field_birthdate_date, field_additionalInfo_CustomObject]
+
+class_person = JavaClass(name="Person", fields=person_class_fields, description="Contains information about a person.")
+class_heartbeat = JavaClass(name="Heartbeat", fields=[])
+
+expected_java_class_person = """\
+package expected.DataTypes;
+
+
+import java.util.Date;
+import java.util.Objects;
+
+
+/**
+ * Contains information about a person.
+ */
+public class Person {
+
+    /**
+     * This contains name.
+     */
+    private String name;
+
+    /**
+     * This contains age.
+     */
+    private int age;
+
+    private Date birthDate;
+
+    private CustomData additionalInfo;
+
+
+    public String getName() {
+        return name;
+    }
+
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+
+    public int getAge() {
+        return age;
+    }
+
+
+    public void setAge(int age) {
+        this.age = age;
+    }
+
+
+    public Date getBirthDate() {
+        return birthDate;
+    }
+
+
+    public void setBirthDate(Date birthDate) {
+        this.birthDate = birthDate;
+    }
+
+
+    public CustomData getAdditionalInfo() {
+        return additionalInfo;
+    }
+
+
+    public void setAdditionalInfo(CustomData additionalInfo) {
+        this.additionalInfo = additionalInfo;
+    }
+
+
+    @Override
+    public boolean equals(Object obj) {
+        if (this == obj)
+            return true;
+        if (!(obj instanceof Person))
+            return false;
+        Person that = (Person) obj;
+        return Objects.equals(getName(), that.getName())
+                && Objects.equals(getAge(), that.getAge())
+                && Objects.equals(getBirthDate(), that.getBirthDate())
+                && Objects.equals(getAdditionalInfo(), that.getAdditionalInfo());
+    }
+
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(
+                getName(),
+                getAge(),
+                getBirthDate(),
+                getAdditionalInfo()
+        );
+    }
+}
+"""
+
+expected_java_class_heartbeat = """\
+package expected.DataTypes;
+
+
+public class Heartbeat {
+}
+"""

tests/test_class_generator.py

@@ -0,0 +1,10 @@
+from src.class_generator import generate_java_class
+from tests.reference_data_java_class import *
+
+
+def test_generate_java_class():
+    assert generate_java_class(class_person, package_example) == expected_java_class_person
+
+
+def test_generate_java_class_no_fields():
+    assert generate_java_class(class_heartbeat, package_example) == expected_java_class_heartbeat