AS2协议下使用pyas2lib发送XML文件并确保文件名正确的方法

from pyas2lib.message import Message
from pyas2lib.context import Context
from pyas2lib.partner import Partner

# 假设已经配置好了Context和Partner对象
# context = Context(...)
# sender_partner = Partner(...)
# receiver_partner = Partner(...)

# 模拟一个AS2上下文和伙伴信息
class MockContext:
    def get_sender_partner(self, as2_id):
        return MockPartner(as2_id)
    def get_receiver_partner(self, as2_id):
        return MockPartner(as2_id)

class MockPartner:
    def __init__(self, as2_id):
        self.as2_id = as2_id
        self.url = "http://mock-as2-receiver.com"
        self.signing_certificate = None # 实际应用中应配置证书
        self.encryption_certificate = None # 实际应用中应配置证书
        self.mdn_url = None
        self.mdn_options = None

context = MockContext()
sender_partner = MockPartner("SENDER_AS2_ID")
receiver_partner = MockPartner("RECEIVER_AS2_ID")

# 待发送的XML内容
xml_data = b"<data><item>example</item></data>"
filename = "my_report.xml"
content_type = "application/xml"

try:
    # 1. 创建Message实例
    msg = Message(context, sender_partner, receiver_partner)

    # 2. **关键步骤：在build()方法中传入文件名和内容类型**
    #    pyas2lib会根据这些参数自动生成Content-Disposition和Content-Type头部
    msg.build(
        data=xml_data,
        filename=filename,
        content_type=content_type,
        # 其他可选参数如 signing_certificate, encryption_certificate, compression等
        # signing_certificate=sender_partner.signing_certificate,
        # encryption_certificate=receiver_partner.encryption_certificate,
        # compression=True
    )

    # 此时，msg对象内部已经包含了正确的头部信息
    # 可以通过 msg.headers 查看，但通常不需要手动修改
    print(f"Message built successfully with filename: {filename} and content type: {content_type}")
    print("Example Content-Disposition header (internally managed): attachment; filename=my_report.xml")
    print("Example Content-Type header (internally managed): application/xml")

    # 3. 发送消息（实际发送逻辑会调用msg.send()或类似方法）
    # 通常会涉及到HTTP请求库，例如requests
    # response = requests.post(receiver_partner.url, headers=msg.headers, data=msg.payload)
    # print("Message sent (simulated).")

except Exception as e:
    print(f"An error occurred: {e}")