Python script for importing CSV to ugc.kulturarvsdata.se

ugc-csv-import.py

import csv

from ksamsok import KSamsok
from sochugc import UGC

api_key = input('Please enter your API key: ')
file_name = input('The name of input CSV: ')
username = input('Which username would you like to use: ')

soch = KSamsok()
ugc = UGC(key=api_key)

with open(file_name, newline='') as csvfile:
   spamreader = csv.reader(csvfile, delimiter=',')
   for i, row in enumerate(spamreader):
       # validation and formatting of SOCH URI
       if not soch.formatUri(row[0], 'rawurl'):
           continue

       # if there is a forth comment column
       comment = None
       if len(row) == 4:
           comment = row[3]

       # SOCH URI, relation, target URI, username, and optional comment
       print(i, soch.formatUri(row[0], 'rawurl'), row[1], row[2], username, comment)
       ugc.create_item_relation(soch.formatUri(row[0], 'rawurl'), row[1], row[2], username, comment)
print('DONE')