A way to hard-code a python object in a script using pickle dumps and loads

byte-embed.py

# This works when run as a script
# but not when copied to a notebook cell and run it.

import pickle
from collections import namedtuple

# We want to embed this object into a script
Foo = namedtuple("Foo", "a b")
a = Foo(1, 2)

obj = pickle.dumps(a)

with open("byte_embed_test_template.py") as f:
  x = f.read()

x = x.replace("VALUEHERE", str(obj))

with open("byte_embed_test.py", "wt") as f:
  f.write(x)

print("*** Contents of byte_embed_test_template.py *********")
print("*****************************************************")
with open("byte_embed_test_template.py") as f:
  print(f.read())
print("*****************************************************")



print("*** Contents of byte_embed_test.py ******************")
print("*****************************************************")
with open("byte_embed_test.py") as f:
  print(f.read())
print("*****************************************************")

import byte_embed_test
print("*** Should print Foo(a=1, a=2) below ***")
byte_embed_test.func()

print("*** Should print 1 and 2 below ***")
print(byte_embed_test.y.a)
print(byte_embed_test.y.b)

byte_embed_test_template.py

import pickle
from collections import namedtuple
Foo = namedtuple("Foo", "a b")

x = 1

y = pickle.loads(VALUEHERE)

def func():
  print(y)