from pybman import Client
from pybman import extract

# initialize client
cl = Client()

# retrieve data of specific context
ctx = cl.get_data(ctx_id="ctx_924547")

# number of records in this context
num_records = ctx.num

# access records via index
item = ctx.records[0]

# get genre from item ...
genre = extract.genre_from_item(item)
print("GENRE:", genre)

# ... or get genres and associated items from context
genres = ctx.get_genres()

# list all genres used in this context
list(genres.keys())

# get all articles (list of item IDs) from context
articles = genres['ARTICLE']

# get data of item (article) by id
item = ctx.get_item(articles[0])

# get title from item
title = extract.title_from_item(item)
print("TITLE:", title)

# get publication date from item
date = extract.date_from_item(item)
print("PUBLISHED:", date)

# get creators (persons) from item
persons = extract.persons_from_item(item)
for person in persons:
    role = extract.role_from_creator(person)
    first_name, last_name = extract.persons_name_from_creator(person)
    id_val, id_type = extract.persons_id_from_creator(person)
    print(role+':', first_name, last_name, '('+ id_val +')')

# intialize client instance (login)
cl_auth = Client(secret="./conf/secret.json")

# retrieve context you are allowed to modify
ctx = cl_auth.get_data(ctx_id="ctx_924547")

# choose item to change
item = ctx.records[0]

# change title (string strip)
title = extract.title_from_item(item)
title = title.strip()
item['data']['metadata']['title'] = title

# get identifier from item
identifier = extract.idx_from_item(item)

# comment on the changes
comment = 'delete unnecessary white space'

# update data in repository
cl_auth.update_data(identifier, item['data'], comment)

# to change values in collection of items use inspector class
from pybman import Inspector
from pybman import DataSet

# create data set of released items
ctx_released = DataSet(data_id="ctx_924547_released", raw=ctx.get_items_released())

# create inspector instance of all released items from context
inspector = Inspector(cl_auth, ctx_released.records)

# strip title strings, i.e. remove leading and trailing white spaces
titles = inspector.clean_titles()
print("successfully cleaned", titles, "titles!")