123 lines
4.6 KiB
Python
123 lines
4.6 KiB
Python
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)
|