def week_logview():
try:
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
allCats = p3.getCats(conn, userID)
dFormat = "MMM/DD/YYYY"
intervalGap = 7
intType = 'day'
logDict = {}
logDict['all'] = p3.checkLog(conn, 'week', userID, 'all')
for eachCat in allCats:
cat = eachCat['name']
catInfo = p3.checkLog(conn, 'week', userID,
cat) #list of dictionary
logDict[str(cat)] = p3.checkLog(conn, 'week', userID, cat)
colorRows = p3.checkCatColor(conn, userID)
colorDict = {}
colorDict['all'] = 'black'
if colorRows != None:
for eachRow in colorRows:
colorDict[str(eachRow['name'])] = str(eachRow['color'])
return render_template('base_log_week.html',
allCats=allCats,
database=DATABASE,
logs=logDict,
dFormat=dFormat,
intervalGap=intervalGap,
colorDict=colorDict)
else:
return render_template('login.html')
except:
flash('Error! Redirecting to login page')
return render_template('login.html')
def changeView():
try:
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
allCats = p3.getCats(conn, userID)
timeSelection = request.form['time']
dataSelection = request.form['views']
if (dataSelection == "checklist"):
dataStruct = p3.rightPanelTask(conn, userID)
return jsonify(dataStruct)
if (dataSelection == "events"):
dataStruct = p3.rightPanelEvent(conn, userID)
return jsonify(dataStruct)
if (dataSelection == "log"):
print "at dataselectionlog!"
dataStruct = p3.allLog(conn, userID, allCats)
print dataStruct
return jsonify(dataStruct)
else:
return render_template('login.html')
except:
flash('Error! Redirecting to login page')
return render_template('login.html')
def addLog():
try:
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
allCats = p3.getCats(conn, userID)
cat = request.form['catName']
hours = request.form['hour']
taskDate = request.form['taskDate']
try:
if p3.legaldate(
taskDate) and int(hours) > 0: #check if input is legal
p3.addLog(conn, cat, hours, userID, taskDate)
else:
# print "check input!!!"
flash('check your input')
return jsonify({"error": "Empty inputs!"})
dataStruct = p3.allLog(conn, userID, allCats)
return jsonify(dataStruct)
except:
flash('check your input')
return jsonify({"error": "Not a valid number!"})
else:
return render_template('login.html')
except:
flash('Error! Redirecting to login page')
return render_template('login.html')
def month_event():
try:
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
allCats = p3.getCats(conn, userID) #need to take care of userID
data = p3.rightPanelEvent(conn, userID)
return render_template('base_event_month.html',
allCats=allCats,
dataStruct=data,
database=DATABASE)
else:
return render_template('login.html')
except:
flash('Error! Redirecting to login page')
return render_template('login.html')
def week_checklist():
try:
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
allCats = p3.getCats(conn, userID)
data = p3.rightPanelTask(conn, userID)
return render_template('base_task_week.html',
allCats=allCats,
dataStruct=data,
database=DATABASE)
else:
return render_template('login.html')
except:
flash('Error! Redirecting to login page')
return render_template('login.html')
def tickedCats():
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
if request.method == 'POST':
dataSelection = request.form['catHiddenRedirect']
if (dataSelection == "checklist"):
dataStruct = p3.rightPanelTask(conn, userID)
return jsonify(dataStruct)
if (dataSelection == "events"):
dataStruct = p3.rightPanelEvent(conn, userID)
return jsonify(dataStruct)
if (dataSelection == "log"):
allCats = p3.getCats(conn, userID)
dataStruct = p3.allLog(conn, userID, allCats)
return jsonify(dataStruct)
return jsonify({"error": "Please refresh!"})
def addEvent():
try:
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
allCats = p3.getCats(conn, userID)
if (request.form['eventName']
== "") or (request.form['eventDate']
== "") or (request.form['eventName']
== "") or (request.form['eventName']
== ""):
return jsonify({"error": "Empty inputs!"})
startCheck = request.form['startTime'].split(':')
startCheckFormat = int(startCheck[0] + startCheck[1])
endCheck = request.form['endTime'].split(':')
endCheckFormat = int(endCheck[0] + endCheck[1])
if (endCheckFormat < startCheckFormat):
return jsonify({"error": "Time continuum violation!"})
eventName = request.form[
'eventName'] # we should change the name of this varchar
eventDate = request.form['eventDate']
start = request.form['startTime'] + ':00'
end = request.form['endTime'] + ':00'
if eventName.split() != [] and p3.legaldate(eventDate):
checkError = p3.addEvent(conn, userID, eventName, eventDate,
start, end)
print checkError
if (checkError):
return jsonify({"error": "Scheduling conflict!"})
dataStruct = p3.rightPanelEvent(conn, userID)
return jsonify(dataStruct)
else:
print "break1"
return render_template('login.html')
except:
flash('Error! Redirecting to login page')
print "break2"
return render_template('login.html')
def deleteTask():
try:
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
allCats = p3.getCats(userID) #need to take care of userID
taskName = request.form[
'catName'] #should we change the name of this varchar
start = request.form['startDate']
end = request.form['endDate']
cat = request.form['catOpt']
p3.deleteTask(userID, taskName, start, end, cat)
return render_template('base.html',
database=DATABASE,
allCats=allCats)
else:
flash('not currently logged in')
return render_template('login.html')
except:
flash('Error! Redirecting to login page')
return render_template('login.html')
def csvUpload():
if 'userID' in session:
userID = session['userID']
conn = p3.getConnection()
allCats = p3.getCats(conn, userID)
f = request.files['file']
if not f:
flash('No file is selected')
#return render_template('base.html',allCats=allCats)
fn = secure_filename(f.filename)
if allowed_file(fn):
csv_input = pandas.read_csv(f, header=None)
data = csv_input.values
for eachRow in data:
try:
if eachRow[0] == 'log':
cat = eachRow[1]
hours = int(eachRow[3])
rawDate = eachRow[2].split('/')
logDate = rawDate[2] + '-' + rawDate[
0] + '-' + rawDate[1]
p3.addLog(conn, cat, hours, userID, logDate)
elif eachRow[0] == 'event':
eventName = eachRow[1]
rawDate = eachRow[2].split('/')
eventDate = rawDate[2] + '-' + rawDate[
0] + '-' + rawDate[1]
start = eachRow[3] + ':00'
end = eachRow[4] + ':00'
p3.addEvent(conn, userID, eventName, eventDate, start,
end)
elif eachRow[0] == 'task':
cat = eachRow[1]
isFinished = int(eachRow[2])
taskName = eachRow[3]
rawStart = eachRow[4].split('/')
start = rawStart[2] + '-' + rawStart[
0] + '-' + rawStart[1]
rawEnd = eachRow[5].split('/')
end = rawEnd[2] + '-' + rawEnd[0] + '-' + rawEnd[1]
if taskName.split() != [] and p3.legaldate(
start) and p3.legaldate(end):
p3.addTask(conn, isFinished, userID, taskName,
start, end, cat)
checkID = p3.checkTaskID(conn, taskName, start,
end)
if checkID != None:
parID = checkID['taskID']
numSubtask = len(
eachRow) - 6 #check number of subtasks
print numSubtask
if (numSubtask == 0):
p3.addSubtaskNull(conn, userID, parID)
elif (numSubtask > 0):
for i in range(6, len(eachRow)):
sub = eachRow[i]
if sub.split() != []:
startDT = dt.datetime.strptime(
start, '%Y-%m-%d')
print startDT
endDT = dt.datetime.strptime(
end, '%Y-%m-%d')
numDays = endDT - startDT
length = numDays / numSubtask
daysFromStart = (length *
(i - 5)).days
endFormatted = startDT + dt.timedelta(
days=daysFromStart)
endFormat = str(endFormatted)[:-9]
p3.addTask(conn, isFinished,
userID, sub, start,
endFormat, cat)
print "subtaskAdded"
childID = p3.checkTaskID(
conn, sub, start,
endFormat)['taskID']
p3.addSubtask(
conn, userID, parID, childID)
except:
flash("invalid input")
return render_template('base.html', allCats=allCats)
else:
return render_template('login.html')
def addTask():
try:
conn = p3.getConnection()
if 'userID' in session:
userID = session['userID']
allCats = p3.getCats(conn, userID)
isFinished = 0
taskName = request.form['catName']
start = request.form['startDate']
end = request.form['endDate']
cat = request.form['catOpt']
try:
if taskName.split() != [] and p3.legaldate(
start) and p3.legaldate(end):
checkError = p3.addTask(conn, isFinished, userID, taskName,
start, end, cat)
if (checkError):
return jsonify({"error": "Duplicate task!"})
checkID = p3.checkTaskID(conn, taskName, start, end)
if checkID != None:
parID = checkID['taskID']
numSubtask = int(
request.form['num']) #check number of subtasks
print numSubtask
if (numSubtask == 0):
p3.addSubtaskNull(conn, userID, parID)
for i in range(1, numSubtask + 1):
sub = request.form['subtask' + str(i)]
print sub
if sub.split() != []:
startDT = dt.datetime.strptime(
start, '%Y-%m-%d')
endDT = dt.datetime.strptime(end, '%Y-%m-%d')
numDays = endDT - startDT
length = numDays / numSubtask
daysFromStart = (length * i).days
endFormatted = startDT + dt.timedelta(
days=daysFromStart)
endFormat = str(endFormatted)[:-9]
p3.addTask(conn, isFinished, userID, sub,
start, endFormat, cat)
print "subtaskAdded"
childID = p3.checkTaskID(
conn, sub, start, endFormat)['taskID']
p3.addSubtask(conn, userID, parID, childID)
else:
return jsonify({"error": "Empty inputs!"})
jsonData = p3.rightPanelTask(conn, userID)
return jsonify(jsonData)
except:
flash('something is wrong: check your entries!')
return render_template('base.html',
allCats=allCats,
database=DATABASE)
else:
print "omg 2"
return render_template('login.html')
except:
print "omg3"
flash('Error! Redirecting to login page')
return render_template('login.html')
