#!/usr/bin/env python

# -*- coding: utf-8 -*-

endpoint = "http://localhost/omeka/api"

apikey = "XXXXXXXXXXXXX"

from omekaclient import OmekaClient

from sys import stdin

import argparse

import yaml

import json

try:

import markdown

except ImportError:

print "Error: Python module for Markdown missing"

# Define and parse command-line arguments

parser = argparse.ArgumentParser()

parser.add_argument('inputfile', type=file, nargs="?", default=stdin, help='Name of input YAML file (optional)')

parser.add_argument('-p', '--public', action='store_true', help='Make item public')

parser.add_argument('-f', '--featured', action='store_true', help='Make item featured')

parser.add_argument('-c', '--collection', type=int, default=None, help='Add item to collection n')

parser.add_argument('-t', '--type', type=int, default=1, help='Specify item type using Omeka id; default is 1, for "Document"')

parser.add_argument('-u', '--upload', default=None, help='Name of file to upload and attach to item')

parser.add_argument('-m', '--mdmark', default="markdown>", help='Change string prefix that triggers markdown conversion; default is "markdown>"')

args = vars(parser.parse_args())

# A dictionary of all the element ids in the Omeka database

# TODO: Figure out how to generate this dict using Omeka API elements resource

elements = {1:"Text",2:"Interviewer",3:"Interviewee",4:"Location",5:"Transcription",6:"Local URL",7:"Original Format",10:"Physical Dimensions",11:"Duration",12:"Compression",13:"Producer",14:"Director",15:"Bit Rate/Frequency",16:"Time Summary",17:"Email Body",18:"Subject Line",19:"From",20:"To",21:"CC",22:"BCC",23:"Number of Attachments",24:"Standards",25:"Objectives",26:"Materials",27:"Lesson Plan Text",28:"URL",29:"Event Type",30:"Participants",31:"Birth Date",32:"Birthplace",33:"Death Date",34:"Occupation",35:"Biographical Text",36:"Bibliography",37:"Contributor",38:"Coverage",39:"Creator",40:"Date",41:"Description",42:"Format",43:"Identifier",44:"Language",45:"Publisher",46:"Relation",47:"Rights",48:"Source",49:"Subject",50:"Title",51:"Type",58:"Additional Creator",59:"Transcriber",60:"Producer",61:"Render Device",62:"Render Details",63:"Capture Date",64:"Capture Device",65:"Capture Details",66:"Change History",67:"Watermark",69:"Encryption",70:"Compression",71:"Post Processing",72:"Width",73:"Height",74:"Bit Depth",75:"Channels",76:"Exif String",77:"Exif Array",78:"IPTC String",79:"IPTC Array",80:"Bitrate",81:"Duration",82:"Sample Rate",83:"Codec",84:"Width",85:"Height"}

# Reversing the keys and values of elements dictionary for easier lookup

d = {}

for key, val in elements.items():

d[val] = key

# Get YAML from input file

yamlinput = args['inputfile'].read()

# Set general item settings using command line arguments or defaults

# Defaults are to make items private, not featured, no collection documents

jsonobj = {"collection": {"id": args["collection"]}, "item_type": {"id": args["type"]}, "featured": args["featured"], "public": args["public"]}

# This is where the input YAML is converted into json and added to jsonobj

data = yaml.load(yamlinput)

element_texts = []

for key in data:

element_text = {"html": True, "text": "none", "element_set": {"id": 0}}

element = {"id": d[key]}

element_text["element"] = element

text = str(data[key])

if text.startswith(args["mdmark"]):

element_text["text"] = markdown.markdown(text[len(args["mdmark"]):])

else:

element_text["text"] = text

element_texts.append(element_text)

jsonobj["element_texts"] = element_texts

# Dump json object into a string and post, printing API response

jsonstr = json.dumps(jsonobj)

response, content = OmekaClient(endpoint, apikey).post("items", jsonstr)

location = response["location"]

print response.reason, location

# Optionally, upload and attach file to new item

if args["upload"] is not None:

print "Uploading file ..."

uploadjson = {"item": {"id": int(location.split("/")[-1])}}

uploadmeta = json.dumps(uploadjson)

uploadfile = open(args["upload"], "r").read()

response, content = OmekaClient(endpoint, apikey).post_file(uploadmeta, args["upload"], uploadfile)

print response.reason, response["location"]