add create_events function, integrate with createTask

This commit is contained in:
2022-11-17 21:43:01 -05:00
parent f27749009c
commit e968e835f7
2 changed files with 16 additions and 2 deletions

7
app.py
View File

@@ -8,6 +8,7 @@ from wtforms import (StringField, DateField, TimeField, TextAreaField, IntegerFi
RadioField) RadioField)
from wtforms.validators import InputRequired, Length from wtforms.validators import InputRequired, Length
from wtforms_sqlalchemy.orm import QuerySelectField from wtforms_sqlalchemy.orm import QuerySelectField
from create_events import createEvents
basedir = os.path.abspath(os.path.dirname(__file__)) basedir = os.path.abspath(os.path.dirname(__file__))
@@ -23,6 +24,9 @@ db = SQLAlchemy(app)
from auth import auth as auth_blueprint from auth import auth as auth_blueprint
app.register_blueprint(auth_blueprint) app.register_blueprint(auth_blueprint)
currDay = datetime.now()
currDay = currDay.strftime('%m-%d-%Y')
class Period(db.Model): class Period(db.Model):
period = db.Column(db.Integer, primary_key=True) period = db.Column(db.Integer, primary_key=True)
periodTime = db.Column(db.Time) periodTime = db.Column(db.Time)
@@ -110,10 +114,9 @@ def newPeriod():
period = Period(periodTime=form.periodTime.data, period = Period(periodTime=form.periodTime.data,
weekendSchedule=form.weekendSchedule.data weekendSchedule=form.weekendSchedule.data
) )
print(period.periodTime)
print(period.weekendSchedule)
db.session.add(period) db.session.add(period)
db.session.commit() db.session.commit()
createEvents(db, currDay, Period, Event)
return redirect(f'/period/edit/{period.period}') return redirect(f'/period/edit/{period.period}')
return render_template('newPeriod.html', form=form) return render_template('newPeriod.html', form=form)

11
create_events.py Normal file
View File

@@ -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()