from MessageKeeper import MessageKeeper
from HorizScroll import HorizScroll
from random import random, seed
-from Idler import TrainIdler,GrayIdler,StringIdler,ClockIdler,FortuneIdler,FileIdler,PipeIdler
+from Idler import GreetingIdler,TrainIdler,GrayIdler,StringIdler,ClockIdler,FortuneIdler,FileIdler,PipeIdler
import socket
from posix import geteuid
"""
-GREETING = 'UCC SNACKS'
PIN_LENGTH = 4
DOOR = 1
STATE_GET_SELECTION = 6
STATE_GRANDFATHER_CLOCK = 7
+TEXT_SPEED = 0.8
+IDLE_SPEED = 0.05
+
class DispenseDatabaseException(Exception): pass
class DispenseDatabase:
def scroll_options(username, mk, welcome = False):
if welcome:
- msg = [(center('WELCOME'), False, 0.8),
- (center(username), False, 0.8)]
+ msg = [(center('WELCOME'), False, TEXT_SPEED),
+ (center(username), False, TEXT_SPEED)]
else:
msg = []
choices = ' '*10+'CHOICES: '
logging.info('getting pin for uid %d: .pin not found in home directory'%uid)
return None
if s.st_mode & 077:
- logging.info('getting pin for uid %d: .pin has wrong permissions'%uid)
- return None
+ logging.info('getting pin for uid %d: .pin has wrong permissions. Fixing.'%uid)
+ os.chmod(pinfile, 0600)
try:
f = file(pinfile)
except IOError:
]
disabled = [
]
- idler = choose_idler()
+
+def reset_idler(v, vstatus, t = None):
+ global idlers, idler
+ idler = GreetingIdler(v, t)
+ vstatus.time_of_next_idlestep = time()+idler.next()
+ vstatus.time_of_next_idler = None
+ vstatus.change_state(STATE_IDLE, 1)
def choose_idler():
global idlers, idler
iiindex = 0
+ average_affinity = 10 # guessing here...
- if idler:
+ if idler and idler.__class__ != GreetingIdler:
iiindex = idlers.index(idler)
iilen = len(idlers)
- move = int(random()*len(idlers)) + 1
+ move = int(random()*len(idlers)*average_affinity) + 1
while move >= 0:
- idler = idlers[( (iiindex + 1) % iilen)]
- move = move - idler.affinity()
+ iiindex += 1
+ iiindex %= iilen
+ idler = idlers[iiindex]
+ move -= idler.affinity()
idler.reset()
-def idle_step():
+def idle_step(vstatus):
global idler
if idler.finished():
choose_idler()
- idler.next()
+ vstatus.time_of_next_idler = time() + 30
+ nextidle = idler.next()
+ if nextidle is None:
+ nextidle = IDLE_SPEED
+ vstatus.time_of_next_idlestep = time()+nextidle
class VendState:
def __init__(self,v):
self.cur_selection = ''
self.time_to_autologout = None
- self.time_to_idle = None
-
self.last_timeout_refresh = None
+ def change_state(self,newstate,newcounter=None):
+ if self.state != newstate:
+ #print "Changing state from: ",
+ #print self.state,
+ #print " to ",
+ #print newstate
+ self.state = newstate
+
+ if newcounter is not None and self.counter != newcounter:
+ #print "Changing counter from: ",
+ #print self.counter,
+ #print " to ",
+ #print newcounter
+ self.counter = newcounter
+
def handle_tick_event(event, params, v, vstatus):
vstatus.cur_pin = ''
vstatus.cur_selection = ''
- idle_in(vstatus,2)
- vstatus.state = STATE_IDLE
-
- vstatus.mk.set_message(GREETING)
+ reset_idler(v, vstatus)
### State fully logged out ... reset variables
if vstatus.time_to_autologout and not vstatus.mk.done():
vstatus.time_to_autologout = time() + 15
vstatus.last_timeout_refresh = None
- ### State logged out ... after normal logout??
- # perhaps when logged in?
- if vstatus.time_to_idle is not None and vstatus.cur_user != '':
- vstatus.time_to_idle = None
-
-
## FIXME - this may need to be elsewhere.....
# need to check
vstatus.mk.update_display()
vstatus.cur_user = ''
vstatus.cur_selection = ''
- idle_in(vstatus,2)
- vstatus.state = STATE_IDLE
+ reset_idler(v, vstatus)
- vstatus.mk.set_messages(
- [(center('BYE!'), False, 1.5),
- (GREETING, False, None)])
+ vstatus.mk.set_messages([(center('BYE!'), False, 1.5)])
return
vstatus.cur_selection += chr(key + ord('0'))
vstatus.mk.set_message('SELECT: '+vstatus.cur_selection)
if key == 11:
if vstatus.cur_pin == '':
vstatus.cur_user = ''
- vstatus.mk.set_message(GREETING)
-
- idle_in(vstatus,5)
- vstatus.state = STATE_IDLE
+ reset_idler(v, vstatus)
return
vstatus.cur_pin = ''
if vstatus.username:
v.beep(0, False)
vstatus.cur_selection = ''
- vstatus.state = STATE_GET_SELECTION
+ vstatus.change_state(STATE_GET_SELECTION)
scroll_options(vstatus.username, vstatus.mk, True)
return
else:
v.beep(40, False)
vstatus.mk.set_messages(
[(center('BAD PIN'), False, 1.0),
- (center('SORRY'), False, 0.5),
- (GREETING, False, None)])
+ (center('SORRY'), False, 0.5)])
vstatus.cur_user = ''
vstatus.cur_pin = ''
- idle_in(vstatus,5)
- vstatus.state = STATE_IDLE
+ reset_idler(v, vstatus, 2)
return
if len(vstatus.cur_user) < 5:
if key == 11:
vstatus.cur_user = ''
- vstatus.mk.set_message(GREETING)
- idle_in(vstatus,5)
- vstatus.state = STATE_IDLE
+ reset_idler(v, vstatus)
return
vstatus.cur_user += chr(key + ord('0'))
if not has_good_pin(uid):
logging.info('user '+vstatus.cur_user+' has a bad PIN')
vstatus.mk.set_messages(
- [(' '*10+'INVALID PIN SETUP'+' '*10, False, 3),
- (GREETING, False, None)])
+ [(' '*10+'INVALID PIN SETUP'+' '*10, False, 3)])
vstatus.cur_user = ''
vstatus.cur_pin = ''
- idle_in(vstatus,5)
- vstatus.state = STATE_IDLE
+ reset_idler(v, vstatus, 5)
return
vstatus.cur_pin = ''
vstatus.mk.set_message('PIN: ')
logging.info('need pin for user %s'%vstatus.cur_user)
- vstatus.state = STATE_GETTING_PIN
+ vstatus.change_state(STATE_GETTING_PIN)
return
if key == 11:
vstatus.cur_user = ''
- vstatus.mk.set_message(GREETING)
- idle_in(vstatus,5)
- choose_idler()
+ reset_idler(v, vstatus)
return
- vstatus.state = STATE_GETTING_UID
+ vstatus.change_state(STATE_GETTING_UID)
run_handler(event, key, v, vstatus)
def handle_idle_tick(state, event, params, v, vstatus):
- ### State logged out ... initiate idler in 5 (first start?)
- if vstatus.time_to_idle == None and vstatus.cur_user == '':
- vstatus.time_to_idle = time() + 5
- choose_idler()
-
### State idling
+ idle_step(vstatus)
- if vstatus.time_to_idle is not None and time() > vstatus.time_to_idle:
- idle_step()
-
- if vstatus.time_to_idle is not None and time() > vstatus.time_to_idle + 30:
- vstatus.time_to_idle = time()
+ if vstatus.time_of_next_idler and time() > vstatus.time_of_next_idler:
+ vstatus.time_of_next_idler = time() + 30
choose_idler()
###
vstatus.mk.update_display()
-
- vstatus.state = STATE_GRANDFATHER_CLOCK
+
+ vstatus.change_state(STATE_GRANDFATHER_CLOCK)
run_handler(event, params, v, vstatus)
+ sleep(0.05)
def beep_on(when, before=0):
start = int(when - before)
hourfromnow = localtime(time() + 3600)
- #onthehour = mktime([now[0],now[1],now[2],now[3],27,0,now[6],now[7],now[8]])
+ #onthehour = mktime([now[0],now[1],now[2],now[3],03,0,now[6],now[7],now[8]])
onthehour = mktime([hourfromnow[0],hourfromnow[1],hourfromnow[2],hourfromnow[3], \
0,0,hourfromnow[6],hourfromnow[7],hourfromnow[8]])
or beep_on(quarterhour,0) \
or beep_on(threequarterhour,0) \
or beep_on(fivetothehour,0):
- vstatus.state = STATE_GRANDFATHER_CLOCK
- vstatus.counter = 2
+ vstatus.change_state(STATE_GRANDFATHER_CLOCK,2)
run_handler(event, params, v, vstatus)
else:
- vstatus.state = STATE_IDLE
+ vstatus.change_state(STATE_IDLE)
def handle_grandfather_tick(state, event, params, v, vstatus):
go_idle = 1
hourfromnow = localtime(time() + 3600)
- #onthehour = mktime([now[0],now[1],now[2],now[3],27,0,now[6],now[7],now[8]])
+# onthehour = mktime([now[0],now[1],now[2],now[3],03,0,now[6],now[7],now[8]])
onthehour = mktime([hourfromnow[0],hourfromnow[1],hourfromnow[2],hourfromnow[3], \
0,0,hourfromnow[6],hourfromnow[7],hourfromnow[8]])
next_hour=((hourfromnow[3] + 11) % 12) + 1
if onthehour - time() < next_hour and onthehour - time() > 0:
v.beep(0, False)
- msg.append((" DING!", False, None))
+
+ t = int(time())
+ if (t % 2) == 0:
+ msg.append(("DING!", False, None))
+ else:
+ msg.append((" DING!", False, None))
elif int(onthehour - time()) == 0:
v.beep(255, False)
- msg.append((" DING!", False, None))
- msg.append((" IT'S "+ str(next_hour) + "O'CLOCK AND ALL IS WELL .....", False, None))
+ msg.append((" BONG!", False, None))
+ msg.append((" IT'S "+ str(next_hour) + "O'CLOCK AND ALL IS WELL .....", False, TEXT_SPEED*4))
elif beep_on(halfhour,0):
go_idle = 0
v.beep(0, False)
- msg.append((" HALFHOUR ", False, None))
+ msg.append((" HALFHOUR ", False, 50))
elif beep_on(quarterhour,0):
go_idle = 0
v.beep(0, False)
- msg.append((" QTR HOUR ", False, None))
+ msg.append((" QTR HOUR ", False, 50))
elif beep_on(threequarterhour,0):
go_idle = 0
v.beep(0, False)
- msg.append((" 3 QTR HOUR ", False, None))
+ msg.append((" 3 QTR HR ", False, 50))
elif beep_on(fivetothehour,0):
go_idle = 0
v.beep(0, False)
- msg.append(("Quick run to your lectures! Hurry! Hurry!", False, None))
+ msg.append(("Quick run to your lectures! Hurry! Hurry!", False, TEXT_SPEED*4))
else:
go_idle = 1
## check for X seconds to the hour
-
- vstatus.mk.set_messages(msg)
+
+ if len(msg):
+ vstatus.mk.set_messages(msg)
+ sleep(1)
+
vstatus.mk.update_display()
- sleep(1)
## if no longer case, return to idle
## change idler to be clock
if go_idle and vstatus.mk.done():
- vstatus.counter = 1
- vstatus.state = STATE_IDLE
+ vstatus.change_state(STATE_IDLE,1)
def handle_door_idle(state, event, params, v, vstatus):
# don't care right now.
pass
def handle_door_event(state, event, params, v, vstatus):
- vstatus.time_to_idle = None
-
if params == 1: #door open
- vstatus.state = STATE_DOOR_OPENING
+ vstatus.change_state(STATE_DOOR_OPENING)
logging.warning("Entering open door mode")
v.display("-FEED ME-")
#door_open_mode(v);
vstatus.cur_user = ''
vstatus.cur_pin = ''
elif params == 0: #door closed
- vstatus.state = STATE_DOOR_CLOSING
- idle_in(vstatus, 5)
+ vstatus.change_state(STATE_DOOR_CLOSING)
+ reset_idler(v, vstatus, 3)
logging.warning('Leaving open door mode')
v.display("-YUM YUM!-")
-def idle_in(vstatus,seconds):
- vstatus.time_to_idle = time() + seconds
-
def return_to_idle(state,event,params,v,vstatus):
- if vstatus.time_to_idle is not None and time() > vstatus.time_to_idle:
- vstatus.mk.set_message(GREETING)
- vstatus.state = STATE_IDLE
- return
- if not vstatus.time_to_idle:
- vstatus.mk.set_message(GREETING)
- vstatus.state = STATE_IDLE
- return
+ reset_idler(v, vstatus)
def create_state_table(vstatus):
vstatus.state_table[(STATE_IDLE,TICK,1)] = handle_idle_tick
if USE_DB: db = DispenseDatabase(v, cf.DBServer, cf.DBName, cf.DBUser, cf.DBPassword)
- vstatus.mk.set_message(GREETING)
setup_idlers(v)
- choose_idler()
- vstatus.mk.set_message("Booted")
-
+ reset_idler(v, vstatus)
# This main loop was hideous and the work of the devil.
# This has now been fixed (mostly) - mtearle
logging.error('Database error: '+str(e))
- e = v.next_event(0)
+ e = v.next_event(vstatus.time_of_next_idlestep-time())
(event, params) = e
- vstatus.counter=1
run_handler(event, params, v, vstatus)
# logging.debug('Got event: ' + repr(e))