Export Formats

import requests
import json

class ICSRExporter:
    def __init__(self, base_url, token):
        self.base_url = base_url
        self.headers = {"Authorization": f"Bearer {token}"}
    
    def verify_signature(self):
        """Ensure signature is configured"""
        response = requests.get(
            f"{self.base_url}/firmas/me",
            headers=self.headers
        )
        firma = response.json()
        
        if not firma.get('firma_img_url'):
            raise ValueError("Signature not configured. Please upload signature image.")
        
        return firma
    
    def check_data_completeness(self, icsr_id):
        """Verify ICSR data is complete"""
        response = requests.get(
            f"{self.base_url}/icsr/{icsr_id}/export/context",
            headers=self.headers
        )
        context = response.json()
        
        required = ['paciente_iniciales', 'producto_sospechoso', 'fecha_inicio_evento']
        missing = [f for f in required if not context.get(f)]
        
        if missing:
            raise ValueError(f"Missing required fields: {missing}")
        
        return context
    
    def preview(self, icsr_id):
        """Generate quick preview"""
        response = requests.get(
            f"{self.base_url}/icsr/{icsr_id}/preview/pdf",
            headers=self.headers,
            params={"source": "light"}
        )
        
        if response.ok:
            with open(f"preview_{icsr_id}.pdf", "wb") as f:
                f.write(response.content)
            return True
        return False
    
    def export_docx(self, icsr_id):
        """Export editable DOCX"""
        response = requests.get(
            f"{self.base_url}/icsr/{icsr_id}/export/modelo-b.docx",
            headers=self.headers
        )
        
        if response.ok:
            filename = f"ICSR_{icsr_id}_Formato.docx"
            with open(filename, "wb") as f:
                f.write(response.content)
            print(f"DOCX exported: {filename}")
            return filename
        else:
            raise Exception(f"Export failed: {response.status_code}")
    
    def export_official_pdf(self, icsr_id):
        """Export official DIGEMID PDF"""
        response = requests.get(
            f"{self.base_url}/icsr/{icsr_id}/export/digemid",
            headers=self.headers,
            params={"formato": "pdf"}
        )
        
        if response.ok:
            # Check if we got PDF or DOCX fallback
            content_type = response.headers.get('Content-Type', '')
            
            if 'pdf' in content_type:
                filename = f"ICSR_{icsr_id}_DIGEMID_Oficial.pdf"
                print("PDF generated successfully")
            else:
                filename = f"ICSR_{icsr_id}_DIGEMID_Oficial.docx"
                print("PDF conversion failed, saved as DOCX")
            
            with open(filename, "wb") as f:
                f.write(response.content)
            
            return filename
        else:
            raise Exception(f"Export failed: {response.status_code}")
    
    def full_export_workflow(self, icsr_id):
        """Complete export workflow with validation"""
        print(f"Starting export for ICSR {icsr_id}...")
        
        # 1. Verify signature
        print("1. Verifying signature configuration...")
        firma = self.verify_signature()
        print(f"   ✓ Signature configured for {firma['nombre']}")
        
        # 2. Check data completeness
        print("2. Checking data completeness...")
        context = self.check_data_completeness(icsr_id)
        print(f"   ✓ All required fields present")
        
        # 3. Generate preview
        print("3. Generating preview...")
        if self.preview(icsr_id):
            print(f"   ✓ Preview saved as preview_{icsr_id}.pdf")
        
        # 4. Export DOCX
        print("4. Exporting DOCX...")
        docx_file = self.export_docx(icsr_id)
        print(f"   ✓ {docx_file}")
        
        # 5. Export official PDF
        print("5. Exporting official DIGEMID PDF...")
        pdf_file = self.export_official_pdf(icsr_id)
        print(f"   ✓ {pdf_file}")
        
        print("\n✓ Export workflow completed successfully!")
        return {
            'preview': f"preview_{icsr_id}.pdf",
            'docx': docx_file,
            'pdf': pdf_file
        }

# Usage
exporter = ICSRExporter(
    base_url="https://api.vigia.com/api/v1",
    token="your_token_here"
)

try:
    files = exporter.full_export_workflow(icsr_id=123)
    print(f"\nGenerated files: {json.dumps(files, indent=2)}")
except Exception as e:
    print(f"Error: {e}")