from flask import Flask, render_template, request, flash, redirect import datetime from sqlalchemy import create_engine, Column, Integer, String from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker app = Flask(__name__) app.config['SECRET_KEY'] = 'fa6166b0218186fb852429060105aca18e0f979240be8ebf' Base = declarative_base() class Dates(Base): __tablename__ = 'Dates' id = Column(Integer, primary_key=True, autoincrement=True) dateTypeID = Column(Integer) dateValue = Column(String) class DatesMetadata(Base): __tablename__ = 'DatesMetadata' id = Column(Integer, primary_key=True, autoincrement=True) dateTypeID = Column(Integer) DateTypeName = Column(String) class Settings(Base): __tablename__ = 'Settings' id = Column(Integer, primary_key=True, autoincrement=True) settingType = Column(Integer) settingValue = Column(String) class SettingsMetadata(Base): __tablename__ = 'SettingsMetadata' id = Column(Integer, primary_key=True, autoincrement=True) settingTypeID = Column(Integer) settingTypeName = Column(String) engine = create_engine('sqlite:///static/db.db') Dates.__table__.create(bind=engine, checkfirst=True) Settings.__table__.create(bind=engine, checkfirst=True) DatesMetadata.__table__.create(bind=engine, checkfirst=True) SettingsMetadata.__table__.create(bind=engine, checkfirst=True) Session = sessionmaker(bind=engine) @app.route('/', methods=('GET', 'POST')) @app.route('/index.html', methods=('GET', 'POST')) def index(): session = Session() installCheck = session.query(Dates).first() is None if installCheck is True: # session.close() return redirect('install.html') # Init the dates dict dates = {'plant': '', 'last': '', 'next': '', 'freq': ''} alerts = {'text': '', 'email': ''} # alerts = {'text': False, 'email': True} # Get the feeding frequency and set it to an int dates['freq'] = int(session.query(Dates).filter_by(dateTypeID=4).first().dateValue) alerts = {'text': False, 'email': True} # If clicked the update button for dates, grab the new values and save them to the db. Then render the template if request.method == 'POST': dates['plant'] = request.form['plantDate'] dates['last'] = request.form['lastDate'] dates['next'] = datetime.datetime.strptime(dates['last'],'%Y-%m-%d') + datetime.timedelta(days=dates['freq']) session.query(Dates).filter_by(dateTypeID=1).update({Dates.dateValue: dates['plant']}) session.query(Dates).filter_by(dateTypeID=2).update({Dates.dateValue: dates['last']}) session.query(Dates).filter_by(dateTypeID=3).update({Dates.dateValue: dates['next']}) session.commit() return render_template("index.html", plant=dates, alerts=alerts) # If regular page load, grab all the dates from the db and render the template else: # result = session.query(Dates).filter_by(dateTypeID=1).first() dates['plant'] = session.query(Dates).filter_by(dateTypeID=1).first().dateValue dates['last'] = session.query(Dates).filter_by(dateTypeID=2).first().dateValue dates['next'] = session.query(Dates).filter_by(dateTypeID=3).first().dateValue # dates['freq'] = session.query(Dates).filter_by(dateTypeID=4).first().dateValue # print(datetime.datetime.strptime(dates['plant'], '%Y-%m-%d')) return render_template("index.html", plant=dates, alerts=alerts) @app.route('/install.html') def install(): session = Session() populateDate = datetime.datetime.strftime(datetime.datetime.now(), '%Y-%m-%d') # Populate Dates Table session.add(Dates(dateTypeID=1, dateValue=populateDate)) session.add(Dates(dateTypeID=2, dateValue=populateDate)) session.add(Dates(dateTypeID=3, dateValue=populateDate)) session.add(Dates(dateTypeID=4, dateValue='14')) #Populate DatesMetadata session.add(DatesMetadata(dateTypeID=1, DateTypeName='Plant Date')) session.add(DatesMetadata(dateTypeID=1, DateTypeName='Last Feeding Date')) session.add(DatesMetadata(dateTypeID=1, DateTypeName='Next Feeding Date')) session.add(DatesMetadata(dateTypeID=1, DateTypeName='Feeding Frequency')) # Populate Settings session.add(Settings(settingType=1, settingValue='False')) session.add(Settings(settingType=2, settingValue='False')) # Populate SettingsMetadata session.add(SettingsMetadata(settingTypeID=1, settingTypeName='Text')) session.add(SettingsMetadata(settingTypeID=2, settingTypeName='Email')) session.commit() return redirect('index.html') if __name__ == '__main__': serve(app, host='0.0.0.0', port=8000)