koffiezetter.py

import httplib2, pygame, datetime, time, os
from random import randrange
from pysqlite2 import dbapi2 as sqlite
from kwidgets import KProgressBar
from local_settings import coffee_set_url

KOFFIEBRUIN = (102, 41, 18)
zetkoffie = 0

# references:
# 1000 count units = 380 ml
# 1 cup = 220 ml


class Koffiezetter:
	bezig = 0
	bezig_malen = 0
	zettijd = 0
	wachttijd = 0
	knipper = 0

	def __init__(self, scherm1, myhal1):
		self.aantal_koppen = 0
		self.scherm = scherm1
		self.myhal = myhal1
		self.bezig = ''
		self.http = httplib2.Http(timeout=2)
		self.tempv = TemperatureView(self)
		self.kp1 = KProgressBar(self.scherm,26,86,18,127,1000)
		self.kp2 = KProgressBar(self.scherm,340,19,18,127,1000)
		self.kk = KoffieKorrel(self.scherm, self.myhal)
		self.pijl = Pijltjes(self.scherm, self.myhal)
		self.programma = []
		self.coffee_set_url = coffee_set_url
	def start(self,aantal):
		if len(self.bezig) != 0:
			print "Already started!"
			return
		self.aantal_koppen = aantal
		print("Start " + str(self.aantal_koppen))
		# AutoBaristaScripts defined; Z = add hot water; M = grind; S = sleep
		if self.aantal_koppen == 1: # 2274 is the first cup of 175ml instead of 240ml
			self.programma = ['Z6','M115','Z50','S60','Z115'] # 50 + 170 = 220
			return
		if self.aantal_koppen == 2:
			self.programma = ['Z6','M180','Z50','S60','Z270'] # 50 + 390 = 440
			return
		if self.aantal_koppen == 3:
			self.programma = ['Z6','M270','Z50','S60','Z425'] # 50 + 610 = 660
			return
		if self.aantal_koppen == 4:
			self.programma = ['Z6','M326','Z50','S60','Z580'] # 60 + 830 = 880
			return
		if self.aantal_koppen == 5: #ontkalken / descaling
			self.programma = ['Z500'] #['Z100','S50','Z100','S200','Z100','S50','Z100','S10','Z10','S10','Z10','S340','Z400']
			return

	def updateUi(self): # update all UI elements
		self.tempv.update()
		self.kp1.update(self.myhal.getMaalteller())
		self.kk.update()
		if self.zettijd > 0:
			self.kp2.update(self.zettijd)
			self.pijl.update(self.zettijd)
		else:
			self.kp2.update(self.wachttijd)
			self.pijl.update(self.wachttijd)

		# Currently, water amount is still time-based. For debugging, show flow meter count on screen.
		font = pygame.font.Font(None, 36)
		text = font.render(str(self.myhal.getPompteller()), 0, (10, 10, 10))
		self.scherm.blit(text, (200,15))
	
	def update(self): # every 1/4 second, this routine is called
		if len(self.programma) > 0 and self.bezig != self.programma[0]:
			self.bezig = self.programma[0]
			# first character of program defines action; check what action we are doing now.
			if self.bezig[0] == 'M':
				n = int(self.bezig[1:])
				print 'maal: ', n
				self.myhal.setMaalteller(n)
				self.kp1.setMaxval(n)
				if self.aantal_koppen != 5:
					self.myhal.doGrind()
			if self.bezig[0] == 'Z':
				n = int(self.bezig[1:])
				print 'zet: ',n
				self.myhal.setPompteller(n)
				self.kp2.setMaxval(n)
			if self.bezig[0] == 'S':
				n = int(self.bezig[1:])
				print 'wacht: ',n
				self.wachttijd = n
				self.kp2.setMaxval(n)

		if len(self.bezig) == 0: # no actions left? return.
			return

		if self.knipper < 4 or self.myhal.getDorst() == 1: # fancy light flashing :)
			self.myhal.setLight(1)
		else:
			self.myhal.setLight(0)
		if self.knipper > 8:
			self.knipper = 0
		self.knipper += 1

		maalteller = self.myhal.getMaalteller()

		if self.bezig[0] == 'M' and maalteller < 1:
			self.myhal.stopGrind()
			self.programma.pop(0) # next script item
			self.myhal.setMaalteller(0)

		if self.bezig[0] == 'Z':
			# font = pygame.font.Font(None, 36)
			# text = font.render("Tijd: " + str(self.zettijd), 1, (10, 10, 10))
			# self.scherm.blit(text, (40,40))
			return self.boilcheck() # boiling & pumping control gets a separate subroutine
		
		if self.bezig[0] == 'S':
			self.wachttijd -= 1
			if self.wachttijd < 1:
				self.programma.pop(0) # next script item

	def boilcheck(self):
		self.zettijd = self.myhal.getPompteller()
		print self.bezig, self.zettijd
		temperatuur = self.myhal.getTemperature()
		# some trial & error control engineering to get water of the right temperature
		# (temperatures are somewhat in degrees celcius, but no accurate calibration is done)
		if(self.zettijd > 0):
			if(temperatuur > 88 or self.zettijd < 30) and self.myhal.getDorst() == 0:
				self.myhal.doPump()
				if self.zettijd > 25:
					self.myhal.doBoil()
			else:
				self.myhal.stopPump()
				if(temperatuur < 93 and self.zettijd > 25):
					self.myhal.doBoil()
				else:
					self.myhal.stopBoil()
		else:
			self.myhal.stopAll()
			self.myhal.setLight(0)
			self.programma.pop(0)  # next script item
			self.bezig = ''
			if len(self.programma) == 0 and self.aantal_koppen != 5: # finished & not descaling? store # of cups
				try:
					con = sqlite.connect("/home/pi/code/koffiepy/koffie.db", detect_types=sqlite.PARSE_COLNAMES)
					cur = con.cursor()
					cur.execute("insert into koffie(aantal,datum) values(?, ?)", (self.aantal_koppen, datetime.datetime.now()))
					con.commit()
					cur.close()
					con.close()
					os.system("sync")
				except Exception, e:
					print "Error writing to database: " + str(e)
		
				while self.aantal_koppen > 0:
					self.myhal.doCount() # physical counter
					self.aantal_koppen -= 1
				try:
					#resp, content = self.http.request("2.php?&aantal=" + str(self.aantal_koppen) + "&datum=" + datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S"))
					resp, content = self.http.request(self.coffee_set_url) # ask internet server to pull db change from machine

				except Exception, e:
					print e
		return self.zettijd
	def handle_bot_message(self, message):
		if 'cup of coffee' in message.text.lower():
			self.start(1)
		elif 'two cups of coffee' in message.text:
			self.start(2)
		elif message.text.lower() == 'coffee':
			self.start(1)

class KoffieKorrel: # drawing of random coffee dust
	def __init__(self, scherm1, myhal1):
		self.scherm = scherm1
		self.myhal = myhal1
		self.korrel = pygame.Surface((3,3),pygame.SRCALPHA,32)
		pygame.draw.line(self.korrel,KOFFIEBRUIN,(0,1),(2,1))
		pygame.draw.line(self.korrel,KOFFIEBRUIN,(1,0),(1,2))
		self.korrel.convert_alpha()
	def update(self):
		maalteller = self.myhal.getMaalteller()
		if(maalteller > 0):
			self.scherm.blit(self.korrel,(100+randrange(-5,5),100+randrange(-15,18)))
			self.scherm.blit(self.korrel,(100+randrange(-5,5),100+randrange(-15,18)))
			self.scherm.blit(self.korrel,(100+randrange(-5,5),100+randrange(-15,18)))
	

class Pijltjes: # moving arrows while pumping (needs improvement)
	def __init__(self, scherm1, myhal1):
		self.scherm = scherm1
		self.myhal = myhal1
		self.pijl = pygame.Surface((9,9),pygame.SRCALPHA,32)
		pygame.draw.line(self.pijl,(0,0,0),(0,5),(5,0),3)
		pygame.draw.line(self.pijl,(0,0,0),(0,5),(5,9),3)
		self.pijl.convert_alpha()
		self.cnt = 0
	def update(self,zettijd):
		if zettijd == 0:
			return
		if self.myhal.getDorst() == 0:
			self.cnt = (self.cnt + 1) % 3
		self.scherm.blit(self.pijl,(300-2*self.cnt,199))

class TemperatureView: # rectangle changes from blue to red
	def __init__(self, koffiezetter):
		self.koffiezetter = koffiezetter
	def update(self):
		font = pygame.font.Font(None, 36)
		font2 = pygame.font.Font(None, 72)
		tmp = self.koffiezetter.myhal.getTemperature()
		red = tmp*2.55
		if red > 254:
			red = 254
		blue = 255-red

		pygame.draw.rect(self.koffiezetter.scherm, (red,0,blue), (217,100,18,50),0)
		if self.koffiezetter.myhal.getDorst() == 1:
			self.koffiezetter.myhal.setLight(1)
			text = font2.render("!", 0, (255, 10, 10))
			self.koffiezetter.scherm.blit(text, (430,130))
		else:
			self.koffiezetter.myhal.setLight(0)
		aantal_koppen = self.koffiezetter.myhal.getAantal()
		if len(self.koffiezetter.bezig) != 0:
			aantal_koppen = self.koffiezetter.aantal_koppen
		text = font2.render(str(aantal_koppen),0, (255,255,255))
		self.koffiezetter.scherm.blit(text, (105,156))