avrotize 2.21.1__py3-none-any.whl

avrotize/avrotopython/enum_core.jinja

@@ -0,0 +1,87 @@
+from enum import Enum
+
+{%- if ordinals | length == 0 %}
+_{{ class_name }}_members = []
+{%- endif %}
+
+class {{ class_name }}(Enum):
+    """
+    {{ docstring }}
+    """
+    {%- for symbol in symbols %}
+    {{ symbol }} = '{{symbol}}'
+    {%- endfor %}
+
+    @classmethod
+    def from_ordinal(cls, ordinal: int|str) -> '{{ class_name }}':
+        """
+        Get enum member by ordinal
+
+        Args:
+            ordinal (int| str): The ordinal of the enum member. This can be an integer or a string representation of an integer.
+
+        Returns:
+            The enum member corresponding to the ordinal.
+        """
+        {%- if ordinals | length == 0 %}
+        # pylint: disable=global-statement
+        global _{{ class_name }}_members
+        # pylint: enable=global-statement
+        {%- endif %}
+
+        if ordinal is None:
+            raise ValueError("ordinal must not be None")
+        if isinstance(ordinal, str) and ordinal.isdigit():
+            ordinal = int(ordinal)
+        {%- if ordinals | length > 0 %}
+        if isinstance(ordinal, int):
+        {%- for sym, ord in ordinals.items() -%}
+            {%- if loop.first %}
+            if ordinal == {{ ord }}:
+            {%- else %}
+            elif ordinal == {{ ord }}:
+            {%- endif %}
+                return {{ class_name }}.{{ sym }}
+        {%- endfor %}
+            raise ValueError("Ordinal not found in enum")
+        else:
+            raise ValueError("Ordinal must be an integer or a string representation of an integer")
+        {%- else %}
+        if not _{{ class_name }}_members:
+            _{{ class_name }}_members = list(cls)
+        if 0 <= int(ordinal) < len(_{{ class_name }}_members):
+            return _{{ class_name }}_members[ordinal]
+        else:
+            raise IndexError("Ordinal out of range for enum")
+        {%- endif %}
+
+    @classmethod
+    def to_ordinal(cls, member: '{{ class_name }}') -> int:
+        """
+        Get enum ordinal
+
+        Args:
+            member ({{ class_name }}): The enum member to get the ordinal of.
+
+        Returns:
+            The ordinal of the enum member.
+        """
+        {%- if ordinals | length == 0 %}
+        # pylint: disable=global-statement
+        global _{{ class_name }}_members
+        # pylint: enable=global-statement
+        
+        if not _{{ class_name }}_members:
+            _{{ class_name }}_members = list(cls)
+        return _{{ class_name }}_members.index(member)
+        {%- else %}
+        {% for sym, ord in ordinals.items() %}
+        if member == {{ class_name }}.{{ sym }}:
+            return {{ ord }}
+        {%- endfor %}
+        raise ValueError("Member not found in enum")
+        {%- endif %}
+
+{%- if ordinals | length == 0 %}
+_{{ class_name }}_members = list({{ class_name }})
+{%- endif %}

avrotize/avrotopython/pyproject_toml.jinja

@@ -0,0 +1,18 @@
+[tool.poetry]
+name = "{{ package_name }}"
+version = "0.1.0"
+description = "A package for handling Avro schema data"
+authors = ["Your Name"]
+license = "MIT"
+packages = [ { include = "{{ package_name|replace('-','_')|lower }}", from="src" } ]
+
+[tool.poetry.dependencies]
+python = "<4.0,>=3.10"
+dataclasses-json = "^0.6.7"
+
+[tool.poetry.dev-dependencies]
+pylint = "^3.2.3"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"

avrotize/avrotopython/test_class.jinja

@@ -0,0 +1,97 @@
+"""
+Test case for {{ class_name }}
+"""
+
+import os
+import sys
+import unittest
+
+sys.path.append(os.path.realpath(os.path.join(os.path.dirname(__file__), '../src'.replace('/', os.sep))))
+
+from {{ package_name | lower }} import {{ class_name }}
+
+{%- for import_type in import_types if import_type not in ['decimal.Decimal', 'datetime.datetime', 'datetime.date', 'datetime.time', 'datetime.timedelta'] %}
+{%- set import_type_name = 'Test_'+import_type.split('.')[-1] %}
+{%- set flat_package = '_'.join(import_type.split('.')[:-1]) | lower %}
+{%- set import_package_name = flat_package if flat_package.startswith('test_') else 'test_' + flat_package %}
+
+{%- if import_type.startswith('.') %}
+from .{{ import_package_name }} import {{ import_type_name }}
+{%- else %}
+from {{ import_package_name }} import {{ import_type_name }}
+{%- endif -%}
+{%- endfor %}
+{%- for import_type in import_types if import_type in ['datetime.datetime', 'datetime.date', 'datetime.time', 'datetime.timedelta'] %}
+{%- if loop.first %}
+import datetime
+{%- endif %}
+{%- endfor %}
+
+
+class {{ test_class_name }}(unittest.TestCase):
+    """
+    Test case for {{ class_name }}
+    """
+
+    def setUp(self):
+        """
+        Set up test case
+        """
+        self.instance = {{ test_class_name }}.create_instance()
+
+    @staticmethod
+    def create_instance():
+        """
+        Create instance of {{ class_name }} for testing
+        """
+        instance = {{ class_name }}(
+            {%- for field in fields %}
+            {{ field.name }}={{ field.test_value }}{%- if not loop.last %},{%- endif %}
+            {%- endfor %}
+        )
+        return instance
+
+    {% for field in fields %}
+    def test_{{ field.name }}_property(self):
+        """
+        Test {{ field.name }} property
+        """
+        test_value = {{ field.test_value }}
+        self.instance.{{ field.name }} = test_value
+        self.assertEqual(self.instance.{{ field.name }}, test_value)
+    {% endfor %}
+
+    {%- if avro_annotation %}
+    def test_to_byte_array_avro(self):
+        """
+        Test to_byte_array method with avro media type
+        """
+        media_type = "application/vnd.apache.avro+avro"
+        bytes_data = self.instance.to_byte_array(media_type)
+        new_instance = {{ class_name }}.from_data(bytes_data, media_type)
+        bytes_data2 = new_instance.to_byte_array(media_type)
+        self.assertEqual(bytes_data, bytes_data2)
+    {%- endif %}
+
+    {%- if dataclasses_json_annotation %}
+    def test_to_byte_array_json(self):
+        """
+        Test to_byte_array method with json media type
+        """
+        media_type = "application/json"
+        bytes_data = self.instance.to_byte_array(media_type)
+        new_instance = {{ class_name }}.from_data(bytes_data, media_type)
+        bytes_data2 = new_instance.to_byte_array(media_type)
+        self.assertEqual(bytes_data, bytes_data2)
+
+    def test_to_json():
+        """
+        Test to_json method
+        """
+        json_data = self.instance.to_json()
+        media_type = "application/json"
+        new_instance = {{ class_name }}.from_data(json_data)
+        json_data2 = new_instance.to_json()
+        self.assertEqual(json_data, json_data2)
+    {%- endif %}
+

avrotize/avrotopython/test_enum.jinja

@@ -0,0 +1,23 @@
+import os
+import sys
+import unittest
+
+sys.path.append(os.path.realpath(os.path.join(os.path.dirname(__file__), '../src'.replace('/', os.sep))))
+
+from {{ package_name | lower }} import {{ class_name }}
+
+class {{ test_class_name }}(unittest.TestCase):
+
+    def setUp(self):
+        """
+        Setup test
+        """
+        self.instance = {{ test_class_name }}.create_instance()
+
+    @staticmethod
+    def create_instance():
+        """
+        Create instance of {{ class_name }}
+        """
+        return "{{ symbols[0] }}"
+