diff --git a/palm/palmc/codegen.py b/palm/palmc/codegen.py index b07dc18..8b8c0df 100644 --- a/palm/palmc/codegen.py +++ b/palm/palmc/codegen.py @@ -67,7 +67,9 @@ def gen_module(messages, imports, tlenums, with_slots, packages, curr_package): out('from palm.palm import ProtoBase, is_string, RepeatedSequence, ProtoValueError\n\n_PB_type = type\n_PB_finalizers = []\n\n') for i in imports: - out('import %s\n' % convert_proto_name(i)) + converted_name = convert_proto_name(i) + out('import %s\n' % converted_name) + out('from %s import *\n' % converted_name) for ename, espec in tlenums: write_enum(ename, espec) diff --git a/test/test.py b/test/test.py index 6c51437..009ec8a 100644 --- a/test/test.py +++ b/test/test.py @@ -528,6 +528,13 @@ def test_fast_append_on_rendered_repeated_fails(self): pass else: assert 0, "SHOULD HAVE RAISED" + + def test_field_from_imported_proto(self): + pb1 = test_palm.Test(req_a=1, req_b=2, req_c=3) + pb1.ext = test_palm.Bar(message="hi") + pb2 = test_palm.Test(pb1.dumps()) + assert pb2.ext.message == "hi" + class TestNesting(object): def test_nested_messages(self): """See https://github.com/bumptech/palm/issues/24"""