cacahuete/python-snippet
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

peewee_test: changed called to Object().save() to Object.create() to by 'more pythonic'?

Browse source
This commit is contained in:
Cacahuete Lenovo Ubuntu 2018-10-04 21:24:15 +02:00
parent 807e434987
commit 6c38e6cbcb
1 changed files with 18 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
peewee_test.py
View file

@ -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: