From 034289ddd2e517afc1a400341b01801afcf3f455 Mon Sep 17 00:00:00 2001
From: Dusan Klinec <dusan.klinec@gmail.com>
Date: Tue, 18 Sep 2018 14:56:55 +0200
Subject: [PATCH] pb2py: fields moved to classmethod get_fields()

- garbage collectable after (de)serialization
---
 protob/pb2py | 62 +++++++++++++++++++++++++++++-----------------------
 1 file changed, 35 insertions(+), 27 deletions(-)

diff --git a/protob/pb2py b/protob/pb2py
index cf73456f8d..4bb14f3bfb 100755
--- a/protob/pb2py
+++ b/protob/pb2py
@@ -211,6 +211,39 @@ class Descriptor:
                 yield "        self.{0} = {0}".format(field.name)
         # fmt: on
 
+    def create_fields_method(self, fields):
+        # fmt: off
+        yield "    @classmethod"
+        yield "    def get_fields(cls):"
+        yield "        return {"
+        for field in fields:
+            comments = []
+            if field.required:
+                comments.append("required")
+            if field.orig.HasField("default_value"):
+                comments.append("default={}".format(field.orig.default_value))
+
+            if comments:
+                comment = "  # " + " ".join(comments)
+            else:
+                comment = ""
+
+            if field.repeated:
+                flags = "p.FLAG_REPEATED"
+            else:
+                flags = "0"
+
+            yield "            {num}: ('{name}', {type}, {flags}),{comment}".format(
+                num=field.number,
+                name=field.name,
+                type=field.proto_type,
+                flags=flags,
+                comment=comment,
+            )
+
+        yield "        }"
+        # fmt: on
+
     def process_message(self, message):
         logging.debug("Processing message {}".format(message.name))
         msg_id = self.message_types.get(message.name)
@@ -238,35 +271,10 @@ class Descriptor:
             yield "    MESSAGE_WIRE_TYPE = {}".format(msg_id)
 
         if fields:
-            yield "    FIELDS = {"
-            for field in fields:
-                comments = []
-                if field.required:
-                    comments.append("required")
-                if field.orig.HasField("default_value"):
-                    comments.append("default={}".format(field.orig.default_value))
-
-                if comments:
-                    comment = "  # " + " ".join(comments)
-                else:
-                    comment = ""
-
-                if field.repeated:
-                    flags = "p.FLAG_REPEATED"
-                else:
-                    flags = "0"
-
-                yield "        {num}: ('{name}', {type}, {flags}),{comment}".format(
-                    num=field.number,
-                    name=field.name,
-                    type=field.proto_type,
-                    flags=flags,
-                    comment=comment,
-                )
-
-            yield "    }"
             yield ""
             yield from self.create_init_method(fields)
+            yield ""
+            yield from self.create_fields_method(fields)
 
         if not fields and not msg_id:
             yield "    pass"