问题描述
my_proto文件夹里有两个proto文件,book.proto想要引用person.proto文件中的Person,如下
book.proto
syntax = "proto2";import "person.proto"; // 导入person.proto文件message Book {optional string name = 1;optional float price = 2;optional Person author = 3;
}
person.proto
syntax = "proto2";message Person {optional string name = 1;optional float height = 2;optional int32 age = 3;optional SEX sex = 4 [ default = MALE ];repeated Phone phone = 5;message Phone {optionale string number = 1; optional Type type = 2;}enum Type {HOME = 0;MOBILE = 1;}enum SEX {MALE = 0;FEMALE = 1;}
}
使用protoc编译protoc文件后,my_proto文件夹中的文件为:person.proto, book.proto, person_pb.py, book_pb.py
person_pb.py
文件中出现以下引用
import person_pb2 as person__pb2
构建my_test.py进行测试
import string
from my_proto import person_pb2, book_pb2book = book_pb2.Book()
book.name = "python"
book.price = 56.7
person = book.personperson.name = "test"
person.sex = 0
person.height = 175.3
person.age = 26for i in range(2)tel = person.phone.add()tel.number = "".join(random.choice(string.digits) for k in range(11))tel.type = random.choice([0, 1])serizalized_book = book.SerizalizeToString()
print(f"serizalized book {serizalized_book}")new_book = book_pb.Book()
new_book.ParseFromString(serizalized_book)
print(f"deserizalized book {new_book}")
运行时报错:import person_pb2 as person__pb2
ModuleNotFoundError: No module named ‘person_pb2’
解决方案
ModuleNotFoundError错误通常都是由于引用路径的位置不正确导致的
方法一:
# 修改book_pb.py文件
# import person_pb.py as person__pb.py
from . import person_pb.py as person__pb.py
方法二:
# 在my_proto文件夹中增加__init__.py文件
import sys
import ossys.path.append(os.path.dirname(__file__))
参考文章:https://qa.1r1g.com/sf/ask/3775421401/