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