From e968e835f7e72f90c6cfa106a55e5214e9b1c47d Mon Sep 17 00:00:00 2001 From: William Peebles Date: Thu, 17 Nov 2022 21:43:01 -0500 Subject: [PATCH] add create_events function, integrate with createTask --- app.py | 7 +++++-- create_events.py | 11 +++++++++++ 2 files changed, 16 insertions(+), 2 deletions(-) create mode 100644 create_events.py diff --git a/app.py b/app.py index 6ca4698..87951a0 100644 --- a/app.py +++ b/app.py @@ -8,6 +8,7 @@ from wtforms import (StringField, DateField, TimeField, TextAreaField, IntegerFi RadioField) from wtforms.validators import InputRequired, Length from wtforms_sqlalchemy.orm import QuerySelectField +from create_events import createEvents basedir = os.path.abspath(os.path.dirname(__file__)) @@ -23,6 +24,9 @@ db = SQLAlchemy(app) from auth import auth as auth_blueprint app.register_blueprint(auth_blueprint) +currDay = datetime.now() +currDay = currDay.strftime('%m-%d-%Y') + class Period(db.Model): period = db.Column(db.Integer, primary_key=True) periodTime = db.Column(db.Time) @@ -110,10 +114,9 @@ def newPeriod(): period = Period(periodTime=form.periodTime.data, weekendSchedule=form.weekendSchedule.data ) - print(period.periodTime) - print(period.weekendSchedule) db.session.add(period) db.session.commit() + createEvents(db, currDay, Period, Event) return redirect(f'/period/edit/{period.period}') return render_template('newPeriod.html', form=form) diff --git a/create_events.py b/create_events.py new file mode 100644 index 0000000..19bfc2c --- /dev/null +++ b/create_events.py @@ -0,0 +1,11 @@ +def createEvents(db, currDay, Period, Event): + periods = Period.query.all() + for period in periods: + q = db.session.query(Event).filter(Event.scheduled_date == currDay).filter(Event.period_num == period.period) + isEventsExist = db.session.query(q.exists()).scalar() + if isEventsExist == False: + event = Event(scheduled_date=currDay, + period_num=period.period + ) + db.session.add(event) + db.session.commit()