From ca73f058306ab817102194baac6f3cf5bce785b9 Mon Sep 17 00:00:00 2001
From: Akash Munagala <akash.munagala@gmail.com>
Date: Mon, 13 Feb 2023 19:33:25 -0800
Subject: [PATCH] Print out frozenset in disasm too

---
 bytecode.cpp | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/bytecode.cpp b/bytecode.cpp
index 607ddf7..b96c33d 100644
--- a/bytecode.cpp
+++ b/bytecode.cpp
@@ -257,6 +257,21 @@ void print_const(PycRef<PycObject> obj, PycModule* mod, const char* parent_f_str
             fputs("}", pyc_output);
         }
         break;
+    case PycObject::TYPE_FROZENSET:
+        {
+            fputs("frozenset({", pyc_output);
+            PycSet::value_t values = obj.cast<PycSet>()->values();
+            auto it = values.cbegin();
+            if (it != values.cend()) {
+                print_const(*it, mod);
+                while (++it != values.cend()) {
+                    fputs(", ", pyc_output);
+                    print_const(*it, mod);
+                }
+            }
+            fputs("})", pyc_output);
+        }
+        break;
     case PycObject::TYPE_NONE:
         fputs("None", pyc_output);
         break;
@@ -312,6 +327,8 @@ void print_const(PycRef<PycObject> obj, PycModule* mod, const char* parent_f_str
     case PycObject::TYPE_CODE2:
         fprintf(pyc_output, "<CODE> %s", obj.cast<PycCode>()->name()->value());
         break;
+    default:
+        fprintf(pyc_output, "<TYPE: %d>\n", obj->type());
     }
 }