#!/usr/bin/env python

"""In this example, we have defined a number of taps and "flow change"
objects and have then turned the taps on and off in a test application
with the flows limited within the class. There's a complete "Mk2"
example which is also available from the same source as this. See
links from
        http://www.wellho.net/resources/Y301.html
"""

class change:
  def __init__(self):
    self.summat = 0
  def act_on(self,dowhat):
    self.temperature, self.destinedstate = dowhat.split(",")
  def gettap(self):
    return self.temperature
  def getdirn(self):
    return self.destinedstate

class tap:
  def __init__(self,room):
    self.title = room
    self.hot = 0
    self.cold = 0
    self.maxflow = 1
  def getname(self):
    return self.title
  def gethotflow(self):
    return self.hot
  def getcoldflow(self):
    return self.cold
  def getmaxflow(self):
    return self.maxflow
  def apply(self,doingwhat):
    whichtap = doingwhat.gettap()
    whichway = doingwhat.getdirn()
    maxup = tap.mainsmax - tap.mainsnow
    if maxup >= 1:
      if whichway == "on":
        if whichtap == "hot":
          if self.hot < self.maxflow:
            self.hot += 1
            tap.mainsnow += 1
        else:
          if self.cold < self.maxflow:
            self.cold += 1
            tap.mainsnow += 1
      else:
        if whichtap == "hot":
          if self.hot > 0:
            self.hot -= 1
            tap.mainsnow -= 1
        else:
          if self.cold > 0:
            self.cold -= 1
            tap.mainsnow -= 1

  def report(self):
    amount1 = self.gethotflow()
    amount2 = self.getcoldflow()
    print "The flow is "+str(amount1)+" hot plus cold "+str(amount2)

# This example supports a single water connection
# See the "mk 2" example that's enhanced to use an extra class
# and can support multiple connections each with different maximum flows
tap.mainsmax = 4
tap.mainsnow = 0

class supertap(tap):
  def __init__(self,room):
    # Some of this code is repeated from tap class. In Mk2 example
    # we use an internally called function to avoid the duplication
    self.title = room
    self.hot = 0
    self.cold = 0
    self.maxflow = 2

# ###########################################

# define a number of taps

users = []
users.append(supertap("kitchen"))
users.append(tap("utility room"))
users.append(tap("shower room"))
# Following line is a demo to show that users is a list of objects
# and you would not "normally" print out such a thing - or if you
# did, you would redefine the __str__ method to make it print neatly!
print users

# Define what Lisa and Leah would like to do with taps

leahdoes = change()
lisadoes = change()
action1 = "hot,on"
action2 = "cold,on"
# Encapsulate the action within the "change" object because
# there's going to be some complex logic to hide
leahdoes.act_on(action1)
lisadoes.act_on(action2)

# Have leah and lisa visit each tap in turn and apply their changes

for flow in users:
  # Better say which tap we're looking at to help the reader
  where = flow.getname()
  upto = flow.getmaxflow()
  print "The tap in the "+where+" has a maximum flow of "+str(upto)

  # Leah visits each tap twice and does her thing and the Lisa
  # then visits each tap once and does hers!
  flow.apply(leahdoes)
  flow.report()
  flow.apply(leahdoes)
  flow.report()
  flow.apply(lisadoes)
  flow.report()