fix:__table_args__ missing

Author: Your Name

build.sh

@@ -1 +1,3 @@
-python3 -m grpc_tools.protoc --proto_path=.  --proto_path=./ --python_out=./ --pyi_out=./ --grpc_python_out=./ ./protobuf_pydantic_gen/*.proto
+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

models/example_model.py

@@ -20,7 +20,7 @@
 
 from pydantic import BaseModel, ConfigDict, Field as _Field
 
-from sqlmodel import Column, Enum, Field, Integer, JSON, SQLModel
+from sqlmodel import Column, Enum, Field, Integer, JSON, PrimaryKeyConstraint, SQLModel, UniqueConstraint
 
 from typing import Any, Dict, List, Optional, Type
 
@@ -49,7 +49,10 @@ def from_protobuf(cls: Type[PydanticModel], src: _message.Message) -> PydanticMo
 class Example(SQLModel, table=True):
     model_config = ConfigDict(protected_namespaces=())
     __tablename__ = "users"
-
+    __table_args__ = (
+        UniqueConstraint(
+            "name", "age", name='uni_name_age'), PrimaryKeyConstraint(
+            "name", name='index_name'),)
     name: Optional[str] = Field(
         description="Name of the example",
         default="John Doe",

protobuf_pydantic_gen/main.py

@@ -426,7 +426,8 @@ def generate_code(request: plugin_pb2.CodeGeneratorRequest,
             message_ext = message.options.Extensions[pydantic_pb2.database]
             # ext = MessageToDict(message_ext)
 
-            table_args = get_table_args(ext, sqlmodel_imports)
+            table_args = get_table_args(msg_ext, sqlmodel_imports)
+            # logging.info(f"table args is {table_args}")
             sqlmodel_imports_str = ", ".join(set(sqlmodel_imports))
             sqlmodel_imports_str = f"from sqlmodel import {sqlmodel_imports_str}" if sqlmodel_imports_str else ""
             imports.add(sqlmodel_imports_str)