feat:support mysql field comment from desc

Author: Your Name

build.sh

@@ -1,3 +1,3 @@
 python3 -m grpc_tools.protoc --proto_path=.  --proto_path=./ --python_out=./ --pyi_out=./ --grpc_python_out=./ ./protobuf_pydantic_gen/*.proto
 
-protoc --plugin=protoc-gen-custom=pydantic_protobuf_gen/main.py --custom_out=./models   -I ./protos protos/example.proto protos/options.proto
+protoc --plugin=protoc-gen-custom=protobuf_pydantic_gen/main.py --custom_out=./models -I ./  -I ./protos protos/example.proto

models/example_model.py

@@ -58,23 +58,44 @@ class Example(SQLModel, table=True):
         default="John Doe",
         alias="full_name",
         primary_key=True,
-        max_length=128)
-    age: Optional[int] = Field(description="Age of the example", default=30, alias="years")
-    emails: Optional[List[str]] = Field(description="Emails of the example", default=[])
-    examples: Optional[List[Example2]] = Field(description="Nested message", default=[], sa_column=Column(JSON))
-    entry: Optional[Dict[str, Any]] = Field(description="Properties of the example", default={}, sa_column=Column(JSON))
-    nested: Optional[Nested] = Field(description="Nested message", sa_column=Column(JSON))
+        max_length=128,
+        sa_column_kwargs={
+            'comment': '"Name of the example"'})
+    age: Optional[int] = Field(
+        description="Age of the example",
+        default=30,
+        alias="years",
+        sa_column_kwargs={
+            'comment': '"Age of the example"'})
+    emails: Optional[List[str]] = Field(description="Emails of the example", default=[],
+                                        sa_column_kwargs={'comment': '"Emails of the example"'})
+    examples: Optional[List[Example2]] = Field(description="Nested message", default=[], sa_column=Column(
+        JSON), sa_column_kwargs={'comment': '"Nested message"'})
+    entry: Optional[Dict[str, Any]] = Field(description="Properties of the example", default={}, sa_column=Column(
+        JSON), sa_column_kwargs={'comment': '"Properties of the example"'})
+    nested: Optional[Nested] = Field(
+        description="Nested message",
+        sa_column=Column(JSON),
+        sa_column_kwargs={
+            'comment': '"Nested message"'})
     created_at: datetime.datetime = Field(
-        description="Creation date of the example",
-        default=datetime.datetime.now(),
-        schema_extra={
-            'required': True})
+        description="Creation date of the example", default=datetime.datetime.now(), schema_extra={
+            'required': True}, sa_column_kwargs={
+            'comment': '"Creation date of the example"'})
     type: Optional[ExampleType] = Field(
         description="Type of the example",
         default=ExampleType.TYPE1,
         sa_column=Column(
-            Enum[ExampleType]))
-    score: Optional[float] = Field(description="Score of the example", default=0.0, le=100.0, sa_type=Integer)
+            Enum[ExampleType]),
+        sa_column_kwargs={
+            'comment': '"Type of the example"'})
+    score: Optional[float] = Field(
+        description="Score of the example",
+        default=0.0,
+        le=100.0,
+        sa_type=Integer,
+        sa_column_kwargs={
+            'comment': '"Score of the example"'})
 
     def to_protobuf(self) -> _message.Message:
         _proto = pool.FindMessageTypeByName("pydantic_example.Example")

protobuf_pydantic_gen/main.py

@@ -406,6 +406,8 @@ def generate_code(request: plugin_pb2.CodeGeneratorRequest,
                     sqlmodel_imports.add("JSON")
                     sqlmodel_imports.add("Column")
                     ext["sa_column"] = "Column(JSON)"
+                if ext and ext.get("description") and msg_ext.get("as_table", False):
+                    ext["sa_column_kwargs"] = {"comment": ext["description"]}
 
                 attr = ",".join(f"{key}={value}" for key,
                                 value in ext.items())

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "protobuf-pydantic-gen"
-version = "0.1.4"
+version = "0.1.5"
 description = "A tool for converting between Pydantic models and Protobuf messages, specifically enabling the generation of Pydantic BaseModel classes from .proto files."
 authors = ["vforfreedom96 <[email protected]>"]
 readme = "README.md"