class App:

	def __init__(self):
		self._app_data = AppData()
		self._client = predictionio.Client(APP_KEY, 1, API_URL)

	def run(self):
		state = "[Main Menu]"

		prompt = "\n"\
			"%s\n"\
			"%s\n"\
			"Please input selection:\n"\
			" 0: Quit application.\n"\
			" 1: Get Recommendations from PredictionIO.\n"\
			" 2: Display user's data." % (state, '-'*len(state))

		while True:
			print prompt
			choice = raw_input().lower()
			if choice == '0':
				print "\nGood Bye!\n"
				break
			elif choice == '1':
				self.recommend_task(state)
			elif choice == '2':
				self.display_user_task(state)
			else:
				print '[Error] \'%s\' is not a valid selection.' % choice

		self._client.close()

	def recommend_task(self, prev_state):
		state = prev_state + " / [Get Recommendations]"
		prompt = "\n"\
			"%s\n"\
			"%s\n"\
			"Please enter user id:" % (state, '-'*len(state))

		while True:
			print prompt
			choice = raw_input().lower()
			u = self._app_data.get_user(choice)
			if u:
				n = 10
				print "[Info] Getting top %s item recommendations for user %s..." % (n, u.uid)
				try:
					self._client.identify(u.uid)
					rec = self._client.get_itemrec_topn(ENGINE_NAME, n)
					u.rec = rec['pio_iids']
					self.display_items(u.rec)
				except predictionio.ItemRecNotFoundError:
					print "[Info] Recommendation not found"

				print "[Info] Go back to previous menu..."
				break
			else:
				print "[Error] invalid user id %s. Go back to previous menu..." % choice
				break

	def display_user_task(self, prev_state):
		state = prev_state + " / [Display User]"
		prompt = "\n"\
			"%s\n"\
			"%s\n"\
			"Please enter user id:" % (state, '-'*len(state))

		while True:
			print prompt
			choice = raw_input().lower()
			u = self._app_data.get_user(choice)
			if u:
				print "[Info] User %s:" % u.uid
				n = 10
				topn_rate_actions = self._app_data.get_top_rate_actions(u.uid, n)
				print "\n[Info] Top %s movies rated by this user:"******"\n[Info] Movies recommended to this user:"******"\n[Info] Go back to previous menu..."
				break
			else:
				print "[Error] invalid user id %s. Go back to previous menu..." % choice
				break
	
	def display_items(self, iids, all_info=False):
		"""print item info for each iid in the list
		"""
		if iids:
			for iid in iids:
				item = self._app_data.get_item(iid)
				if item:
					if all_info:
						print "[Info] %s" % item
					else:
						print "[Info] %s" % item.name
				else:
					print "[Error] Invalid item id %s" % iid
		else:
			print "[Info] Empty."

	def display_rate_actions(self, actions):
		"""print iid and rating
		"""
		if actions:
			for a in actions:
				item = self._app_data.get_item(a.iid)
				if item:
					print "[Info] %s, rating = %s" % (item.name, a.rating)
				else:
					print "[Error] Invalid item id %s" % a.iid
		else:
			print "[Info] Empty."

	def wait_for_ack(self):

		prompt = "\nPress enter to continue..."
		print prompt
		choice = raw_input().lower()
Пример #2
0
class App:

  def __init__(self):
    self._app_data = AppData()
    self._client = predictionio.Client(APP_KEY, 1, API_URL)

  def run(self):
    state = "[Main Menu]"

    prompt = "\n"\
      "%s\n"\
      "%s\n"\
      "Please input selection:\n"\
      " 0: Quit application.\n"\
      " 1: Get personalized recommendation.\n"\
      " 2: Display user data.\n"\
      " 3: Display movie data.\n"\
      " 4: Recommend with multiple movies.\n" % (state, '-'*len(state))

    while True:
      print prompt
      choice = raw_input().lower()
      if choice == '0':
        print "\nGood Bye!\n"
        break
      elif choice == '1':
        self.recommend_task(state)
      elif choice == '2':
        self.display_user_task(state)
      elif choice == '3':
        self.get_similar_movies_task(state)
      elif choice == '4':
        self.recommend_with_multiple_movies_task(state)
      else:
        print '[Error] \'%s\' is not a valid selection.' % choice

    self._client.close()

  def recommend_task(self, prev_state):
    state = prev_state + " / [Get Recommendations]"
    prompt = "\n"\
      "%s\n"\
      "%s\n"\
      "Please enter user id:" % (state, '-'*len(state))

    while True:
      print prompt
      choice = raw_input().lower()
      u = self._app_data.get_user(choice)
      if u:
        n = 10
        print "[Info] Getting top %s item recommendations for user %s..." % (n, u.uid)
        try:
          self._client.identify(u.uid)
          rec = self._client.get_itemrec_topn(ENGINE_NAME, n)
          u.rec = rec['pio_iids']
          self.display_items(u.rec)
        except predictionio.ItemRecNotFoundError:
          print "[Info] Recommendation not found"

        print "[Info] Go back to previous menu..."
        break
      else:
        print "[Error] invalid user id %s. Go back to previous menu..." % choice
        break

  def get_similar_movies_task(self, prev_state):
    state = prev_state + " / [Get Similar Movies]"
    prompt = "\n"\
      "%s\n"\
      "%s\n"\
      "Please enter movie id (eg. 1):" % (state, '-'*len(state))

    while True:
      print prompt
      choice = raw_input().lower()
      i = self._app_data.get_item(choice)

      if i:
        n = 10
        self.display_items((i.iid,), all_info=False)
        print "\n[Info] People who liked this may also liked..."
        try:
          rec = self._client.get_itemsim_topn(SIM_ENGINE_NAME, i.iid, n,
            { 'pio_itypes' : i.genres })
          self.display_items(rec['pio_iids'], all_info=False)
        except predictionio.ItemSimNotFoundError:
          print "[Info] Similar movies not found"

        print "[Info] Go back to previous menu..."
        break
      else:
        print "[Error] invalid item id %s. Go back to previous menu..." % choice
        break

  def recommend_with_multiple_movies_task(self, prev_state):
    state = prev_state + " / [Recommend with Multiple Movies]"
    prompt = "\n"\
      "%s\n"\
      "%s\n"\
      "Please enter comma separated movie ids (eg. 1,2,3):" % (state, '-'*len(state))

    while True:
      print prompt
      choice = raw_input().lower()
      viewed_iids = choice.split(",")
      viewed_items = map(lambda x : self._app_data.get_item(x), viewed_iids)

      viewed_genres = Set()
      for i in viewed_items:
        if i:
          for g in i.genres:
            viewed_genres.add(g)

      if None not in viewed_items:
        n = 10
        self.display_items(viewed_iids, all_info=False)
        print "\n[Info] Top %s similar movies..." % n
        try:
          rec = self._client.get_itemsim_topn(SIM_ENGINE_NAME, choice, n,
            { 'pio_itypes' : list(viewed_genres) })
          self.display_items(rec['pio_iids'], all_info=False)
        except predictionio.ItemSimNotFoundError:
          print "[Info] Similar movies not found"

        print "[Info] Go back to previous menu..."
        break
      else:
        print "[Error] invalid item id %s. Go back to previous menu..." % choice
        break

  def display_user_task(self, prev_state):
    state = prev_state + " / [Display User]"
    prompt = "\n"\
      "%s\n"\
      "%s\n"\
      "Please enter user id:" % (state, '-'*len(state))

    while True:
      print prompt
      choice = raw_input().lower()
      u = self._app_data.get_user(choice)
      if u:
        print "[Info] User %s:" % u.uid
        n = 10
        topn_rate_actions = self._app_data.get_top_rate_actions(u.uid, n)
        print "\n[Info] Top %s movies rated by this user:"******"\n[Info] Getting New Recommendation..."
        n = 10
        try:
          self._client.identify(u.uid)
          rec = self._client.get_itemrec_topn(ENGINE_NAME, n)
          u.rec = rec['pio_iids']
        except predictionio.ItemRecNotFoundError:
          print "[Info] Recommendation not found"

        print "\n[Info] Movies recommended to this user:"******"\n[Info] Go back to previous menu..."
        break
      else:
        print "[Error] invalid user id %s. Go back to previous menu..." % choice
        break

  def display_items(self, iids, all_info=False):
    """print item info for each iid in the list
    """
    if iids:
      for iid in iids:
        item = self._app_data.get_item(iid)
        if item:
          if all_info:
            print "[Info] %s" % item
          else:
            print "[Info] (%s) %s %s %s" % (item.iid, item.name,
              item.release_date.strftime("%d-%b-%Y"), item.genres)
        else:
          print "[Error] Invalid item id %s" % iid
    else:
      print "[Info] Empty."

  def display_rate_actions(self, actions):
    """print iid and rating
    """
    if actions:
      for a in actions:
        item = self._app_data.get_item(a.iid)
        if item:
          print "[Info] (%s) %s %s %s, rating = %s" % (item.iid, item.name,
            item.release_date.strftime("%d-%b-%Y"), item.genres, a.rating)
        else:
          print "[Error] Invalid item id %s" % a.iid
    else:
      print "[Info] Empty."

  def wait_for_ack(self):

    prompt = "\nPress enter to continue..."
    print prompt
    choice = raw_input().lower()