def test_use_machine_as_model(self): states = ['A', 'B', 'C', 'D'] m = Machine(states=states, initial='A') m.add_transition('move', 'A', 'B') m.add_transition('move_to_C', 'B', 'C') m.move() self.assertEquals(m.state, 'B')
def test_send_event_data(self): states = ['A', 'B', 'C', 'D'] s = Stuff() # First pass positional and keyword args directly to the callback m = Machine(model=s, states=states, initial='A', send_event=False) m.add_transition( trigger='advance', source='A', dest='B', before='set_message') s.advance(message='Hallo. My name is Inigo Montoya.') self.assertTrue(s.message.startswith('Hallo.')) # Now wrap arguments in an EventData instance m.send_event = True m.add_transition( trigger='advance', source='B', dest='C', before='extract_message') s.advance(message='You killed my father. Prepare to die.') self.assertTrue(s.message.startswith('You'))
class QueryDateRangeNarrower(object): states = ['init', 'whole', 'left', 'right', 'finished', 'asleep'] OLDEST = 1 NEWEST = 2 def __init__(self, datasource, criteria, kind): self.datasource = datasource self.criteria = criteria self.kind = kind self.maxcount = 50 self.hits = -1 self.querycount = 0 self.machine = Machine(model=self, states=QueryDateRangeNarrower.states, initial='asleep') self.machine.add_transition('start', '*', 'init', after='work') self.machine.add_transition('check', '*', 'finished', conditions=['is_ready']) self.machine.add_transition('step', 'init', 'finished', conditions='is_century_out_of_bounds') self.machine.add_transition('step', 'init', 'whole', after=['runquery', 'check']) self.machine.add_transition('step', 'whole', 'init', conditions='no_hits', after='range_next_century') if self.kind == self.OLDEST: self.date_from = Arrow.fromdatetime( datetime.datetime(1800, 0o1, 0o1)) self.date_to = Arrow.fromdatetime(datetime.datetime(1899, 12, 31)) self.factor = +1 self.machine.add_transition( 'step', 'whole', 'left', unless='is_ready', after=['range_whole_left', 'runquery', 'check']) self.machine.add_transition( 'step', 'left', 'right', conditions='no_hits', after=['range_left_right', 'runquery', 'check']) self.machine.add_transition('step', 'left', 'whole', unless='is_ready', after=['range_shrink']) self.machine.add_transition('step', 'right', 'whole', unless='is_ready', after=['range_shrink']) elif self.kind == self.NEWEST: self.date_from = Arrow.fromdatetime( datetime.datetime(2000, 0o1, 0o1)) self.date_to = Arrow.utcnow() self.date_to += relativedelta(months=12 - self.date_to.month, days=31 - self.date_to.day) self.factor = -1 self.machine.add_transition( 'step', 'whole', 'right', unless='is_ready', after=['range_whole_right', 'runquery', 'check']) self.machine.add_transition( 'step', 'right', 'left', conditions='no_hits', after=['range_right_left', 'runquery', 'check']) self.machine.add_transition('step', 'right', 'whole', unless='is_ready', after=['range_shrink']) self.machine.add_transition('step', 'left', 'whole', unless='is_ready', after=['range_shrink']) else: raise ValueError('kind must be self.OLDEST or self.NEWEST') self.delta = (self.date_to - self.date_from) / 2 def runquery(self): criteria = self.criteria.copy() criteria['pubdate'] = 'within {date_from},{date_to}'.format( date_from=self.date_from.format('YYYY-MM-DD'), date_to=self.date_to.format('YYYY-MM-DD')) query = make_expression_filter({ 'datasource': self.datasource, 'format': 'comfort', 'criteria': criteria, })['expression'] if self.datasource == 'ops': self.response, self.hits = query_ops(query, limit=self.maxcount) elif self.datasource == 'depatisnet': self.response, self.hits = query_depatisnet(query, limit=self.maxcount) elif self.datasource == 'sip': self.response, self.hits = query_sip(query, limit=self.maxcount) else: raise ValueError('Data source "{0}" not implemented'.format( self.datasource)) self.querycount += 1 def no_hits(self): return self.hits == 0 def is_ready(self): return self.hits > 0 and self.hits <= self.maxcount # for "oldest" searches def range_whole_left(self): self.date_to -= self.delta def range_left_right(self): self.date_from += self.delta self.date_to += self.delta # for "newest" searches def range_whole_right(self): self.date_from += self.delta def range_right_left(self): self.date_from -= self.delta self.date_to -= self.delta def range_shrink(self): self.delta /= 2 def range_next_century(self): century = self.date_from.year / 100 century += self.factor year_begin = century * 100 + 00 year_end = century * 100 + 99 self.date_from += relativedelta(years=year_begin - self.date_from.year, months=-self.date_from.month + 1, days=-self.date_from.day + 1) self.date_to += relativedelta(years=year_end - self.date_to.year, months=12 - self.date_to.month, days=31 - self.date_to.day) def is_century_out_of_bounds(self): return self.date_from.year > Arrow.utcnow( ).year or self.date_to.year < 1800 def work(self): debug = False while True: if debug: print('-' * 42) print('state:', self.state) print('delta:', self.delta) print('querycount:', self.querycount) if self.state == 'finished' or self.querycount > 15: break self.step()