subscriber.py

from read_config import read_config_subscriber
from fitconnect import FITConnectClient, Environment
from jwcrypto.jwe import InvalidJWEData
import argparse
import json
import jsonschema
import logging

# parse command line arguments
parser = argparse.ArgumentParser(
                    prog = 'subscriber',
                    description = 'This script uses a subscriber client to retrieve all submissions the subscriber has access to')

parser.add_argument('-c', '--config', help='Path to config file', default='conf/subscriber.yaml')
parser.add_argument('-d', '--data_dir', help='Path to config file', default='./subscriber-data')

args = parser.parse_args()

# configure logging
logging.basicConfig()
#logging.getLogger('fitconnect').level = logging.INFO

# read config_file
config = read_config_subscriber(args.config)

# initialize SDK
fitc = FITConnectClient(Environment[config['sdk']['environment']], config['sdk']['client_id'], config['sdk']['client_secret'])

# activate destination
fitc.activate_destination(config['destination_id'])

# get a list of available submissions
submissions = fitc.available_submissions()

for submission in submissions:
    submission_id = submission['submissionId']
    try:
        print(f"\n== Retrieving submission {submission_id} ==")
        submission = fitc.retrieve_submission(submission_id, config['private_key_decryption'])

        if submission['metadata']:
            print("=== Metadaten ===")
            print(json.dumps(submission['metadata'], indent=2, ensure_ascii=False).encode('utf-8').decode())

        if submission['data_json']:
            print("\n=== Fachdaten ===")
            print(json.dumps(submission['data_json'], indent=2, ensure_ascii=False).encode('utf-8').decode())

        for attachment_id, attachment in submission['attachments'].items():
            print(f"\n=== Anhang ({attachment_id}) ===")
            if attachment.startswith(b'%PDF'):
                with open(f'{args.data_dir}/{submission_id}--{attachment_id}-data.pdf', 'wb') as f:
                    f.write(attachment)
                    print("File written (Type: pdf)")
    except InvalidJWEData as e:
        print(f"Could not decrypt submission {submission_id}")
    except jsonschema.exceptions.ValidationError as e:
        print(f"Invalid schema in submission {submission_id}:", e)
    except json.decoder.JSONDecodeError as e:
        print(f"Unparsable json in submission {submission_id}")
    except ValueError as e:
        print("ValueError", e)