ingest_csv.py

import csv
import uuid
from app import app, db, Identifier, IdentifierType, Object, ObjectType, mint_ark

with app.app_context():

    with open("eric_ingest.csv") as f:
        reader = csv.DictReader(f)

        for row in reader:

            # 1. Get or create the object type
            obj_type = ObjectType.query.filter_by(name=row['object_type']).first()
            if not obj_type:
                print(f"ERROR: ObjectType '{row['object_type']}' does not exist in DB!")
                continue

            # 2. Create object with proper UUID
            obj = Object(
                uuid=uuid.uuid4(),
                type_id=obj_type.id,
                primary_id=row['luna']
            )
            db.session.add(obj)
            db.session.commit()

            # 3. Insert identifiers for this object
            ids_to_add = {
                "luna": row["luna"],
                "arch": row["arch"],
                "file": row["file"],
                "cantaloupe": row["cantaloupe"]
            }

            for shortcode, value in ids_to_add.items():
                id_type = IdentifierType.query.filter_by(shortcode=shortcode).first()
                if not id_type:
                    print(f"ERROR: Missing identifier type: {shortcode}")
                    continue

                db.session.add(
                    Identifier(
                        id=value,
                        object_id=obj.id,
                        type_id=id_type.id
                    )
                )

            # 4. Create ARK
            ark_type = IdentifierType.query.filter_by(shortcode="ark").first()
            ark_id = mint_ark()

            db.session.add(
                Identifier(
                    id=ark_id,
                    object_id=obj.id,
                    type_id=ark_type.id
                )
            )

            db.session.commit()

        print("CSV ingestion complete!")