Support JSON base64 bytes and enums as strings

betterproto/templates/template.py

@@ -1,8 +1,6 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # sources: {{ ', '.join(description.files) }}
 # plugin: python-betterproto
-{% if description.enums %}import enum
-{% endif %}
 from dataclasses import dataclass
 {% if description.typing_imports %}
 from typing import {% for i in description.typing_imports %}{{ i }}{% if not loop.last %}, {% endif %}{% endfor %}
@@ -20,7 +18,7 @@ import grpclib
 
 
 {% if description.enums %}{% for enum in description.enums %}
-class {{ enum.name }}(enum.IntEnum):
+class {{ enum.name }}(betterproto.Enum):
     {% if enum.comment %}
 {{ enum.comment }}