add create_events function, integrate with createTask
This commit is contained in:
7
app.py
7
app.py
@@ -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
11
create_events.py
Normal 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()
|
Reference in New Issue
Block a user