"""
itemrank quickstart import data
"""

import predictionio

import random
import sys

def import_itemrank(access_key):

  random.seed()
  
  client = predictionio.EventClient(access_key)

  print client.get_status()
  
  # generate 10 users, with user ids 1,2,....,10
  user_ids = [str(i) for i in range(1, 11)]
  for user_id in user_ids:
    print "Set user", user_id
    client.set_user(user_id)
  
  # generate 50 items, with item ids 1,2,....,50
  # assign type id 1 to all of them
  item_ids = [str(i) for i in range(1, 51)]
  for item_id in item_ids:
    print "Set item", item_id
    client.set_item(item_id, {
      "itypes" : ['1']
    })
  
  # each user randomly views 10 items
  for user_id in user_ids:
    for viewed_item in random.sample(item_ids, 10):
      print "User", user_id ,"views item", viewed_item
      client.record_user_action_on_item("view", user_id, viewed_item)
  
  client.close()


if __name__ == '__main__':
  if len(sys.argv) < 2:
    sys.exit("Usage: python -m examples.itemrank_quick_start <access_key>")
  import_itemrank(sys.argv[1])