diff --git a/peewee_test.py b/peewee_test.py index b206a7f..284424e 100644 --- a/peewee_test.py +++ b/peewee_test.py @@ -14,11 +14,11 @@ class Schedule(BaseModel): priority = IntegerField(default=0) def add_step(self, s_time, s_target, s_comment=None): - Step(time=s_time, - target=s_target, - comment=s_comment, - schedule=self, - ).save() + Step.create(time=s_time, + target=s_target, + comment=s_comment, + schedule=self, + ) class Step(BaseModel): @@ -65,12 +65,10 @@ if __name__ == "__main__": db.create_tables([Schedule, Step, Event]) - s1 = Schedule(name='semaine', description='Les jours de semaine') - s1.save() + s1 = Schedule.create(name='semaine', description='Les jours de semaine') [s1.add_step(hr, 18.5) for hr in [time(i) for i in range(10)]] - s2 = Schedule(name='weekend', description='Les jours de weekend') - s2.save() + s2 = Schedule.create(name='weekend', description='Les jours de weekend') [s2.add_step(hr, 20.0) for hr in [time(i) for i in range(14)]] schedules = Schedule.select() @@ -80,18 +78,18 @@ if __name__ == "__main__": schedules[0].steps[0].target = 22.0 schedules[0].steps[0].save() - Event(name='test event', - start_date=datetime.now(), - schedule=s1, - _repeat=int_from_list([True for _ in range(7)]), - ).save() + Event.create(name='test event', + start_date=datetime.now(), + schedule=s1, + _repeat=int_from_list([True for _ in range(7)]), + ) - Event(name='test event 2', - start_date=datetime.now(), - stop_date=datetime.now()+timedelta(days=2), - schedule=s2, - _repeat=int_from_list([True] + [False for _ in range(6)]), - ).save() + Event.create(name='test event 2', + start_date=datetime.now(), + stop_date=datetime.now()+timedelta(days=2), + schedule=s2, + _repeat=int_from_list([True] + [False for _ in range(6)]), + ) events = Event.select() for event in events: