from pybman import Client
from pybman import DataSet

# create client instance
client = Client(secret="conf/secret.json")

# retrieve data of given contexts
import_context = client.get_data(ctx_id='ctx_906588')
productive_context = client.get_data(ctx_id='ctx_924547')

# combine records of contexts
records = import_context.records + productive_context.records

# create data set for collected records
total = DataSet(data_id="eth-mpg", raw=records)

# create data set for released records
data_set = DataSet(data_id="eth-mpg_released", raw=total.get_items_released())

# get relationship of genres
grel = data_set.get_genre_relationships()

data_dir = '../pure-eth/data/cleanse/issues/#6-genre-relationships/'

genres = list(grel.keys())
genres.sort()

with open(data_dir + '2019-07-12-grel.txt', mode="w+", encoding="utf-8") as f:
    f.write('GENRE RELATIONSHIPS\n')
    f.write('2019-07-12\n\n')
    for g in genres:
        f.write(g+'\n')
        for g2 in grel[g]:
            f.write(g2 + '\t' + str(len(grel[g][g2])))
            f.write('\n')
        f.write('\n')

with open(data_dir + '2019-07-12-grel-idx.txt', mode="w+", encoding="utf-8") as f:
    f.write('GENRE RELATIONSHIPS\n')
    f.write('2019-07-12\n\n\n')
    for g in genres:
        f.write(g+'\n\n')
        for g2 in grel[g]:
            f.write(g2 + '\t')
            for idx in grel[g][g2]:
                f.write(idx+' ')
            f.write('\n\n')
        f.write('\n\n')