Ejemplo n.º 1
0
 def write(self, queue):
   with open(self.queuefile, 'w') as f:
     for order in queue or []:
       recipe = Recipe.from_json(order)
       f.write(str(recipe))
   with open(self.jsonqueuefile, 'w') as f:
     f.write(json.dumps(queue or []))
Ejemplo n.º 2
0
def recipe_from_json_object(recipe_obj):
  """Takes a dict decoded from a JSON recipe and returns a Recipe object."""
  recipe = Recipe.from_json(recipe_obj)
  if not recipe.ingredients:
    if recipe_obj['drink_name'] == "Random Sour":
      recipe = RandomSourDrink()
      recipe.user_name = recipe_obj['user_name']
    elif recipe_obj['drink_name'] == "Random Boozy":
      recipe = RandomSpirituousDrink()
      recipe.user_name = recipe_obj['user_name']
    elif recipe_obj['drink_name'] == "Random Bubbly Boozy":
      recipe = RandomBubblySpirituousDrink()
      recipe.user_name = recipe_obj['user_name']
    elif recipe_obj['drink_name'] == "Random Bubbly Sour":
      recipe = RandomBubblySourDrink()
      recipe.user_name = recipe_obj['user_name']
  recipe = water_down_recipe(recipe)
  return recipe
Ejemplo n.º 3
0
 def run(self):
   last_drink_id = None
   while True:
     try:
       queue = json.loads(self.get('next_drink'))
       if not self.controller and queue:
         json_recipe = queue[0]
         drink_id = json_recipe['id']
         if drink_id == last_drink_id:
           print "Refusing to remake order %s" % last_drink_id
           time.sleep(20)   # Sleep extra long
           continue  # Don't make the same drink twice
         last_drink_id = drink_id
         next_recipe = water_down_recipe(Recipe.from_json(json_recipe))
         raw_actions = actions_for_recipe(next_recipe)
         actions = []
         for i, action in enumerate(raw_actions):
           progress = 10 + 90 * i / len(raw_actions)
           actions.append(UpdateProgressAction(self, drink_id, progress))
           actions.append(action)
         actions.append(FinishDrinkAction(self, drink_id))
         self.controller.EnqueueGroup(actions)
       else:
         actions = self.controller.InspectQueue()
         if actions:
           logging.info("Current action: %s", actions[0])
           if (isinstance(actions[0],
                          (WaitForGlassRemoval, WaitForGlassPlaced)) and
               isinstance(self.controller.robot, FakeRobot)):
             logging.info("Placing glass")
             actions[0].force = True
       self.write(queue)
     except urllib2.URLError, e:
       logging.warning("URLError: %s", e)
     except ValueError, e:
       print e