def post(self):
if self.request.get("create_newsfeed_form_submit") == "1":
newsfeed = Alert()
newsfeed.title = self.request.get("alertTitle")
newsfeed.message = self.request.get("alertMessage")
newsfeed.type = self.request.get("alertType")
# newsfeed.time_start = self.request.get('alertStartTime')
# newsfeed.time_duration = self.request.get('alertDuration')
if self.request.get("alertPriority") == "True":
newsfeed.priority = True
else:
newsfeed.priority = False
newsfeed.put()
elif self.request.get("submit_changes") == "1":
# This can be done better but it works for now
entities = Alert.query().order(Alert.title)
for a in entities:
id = a.key.integer_id()
if self.request.get("%d" % (id)) == "1":
a.priority = True
a.put()
elif self.request.get("%d" % (id)) == "2":
a.priority = False
a.put()
elif self.request.get("%d" % (id)) == "3":
a.key.delete()
self.redirect(self.get_submit_path())
def testInsertAlertTitle(self):
mytest = Alert()
result = mytest.put()
self.assertEqual(None, ndb.get(result).title)
mytest1 = Alert(title='title')
result1 = mytest1.put()
self.assertEqual('title', ndb.get(result1).title)
def testInsertMessage(self):
mytest = Alert()
result = mytest.put()
self.assertEqual(None, ndb.get(result).type)
mytest1 = Alert(message='testing')
result1 = mytest1.put()
self.assertEqual('testing', ndb.get(result1).message)
mytest2 = Alert(message='234324')
result2 = mytest2.put()
self.assertEqual('234324', ndb.get(result2).message)
mytest3 = Alert(incident_type='fdjgidgjidjgidjgirgjrigjfigjdfgijdfigjdfigjfigjfigjfijgifjgifjgfigj')
result3 = mytest3.put()
self.assertEqual('fdjgidgjidjgidjgirgjrigjfigjdfgijdfigjdfigjfigjfigjfijgifjgifjgfigj', ndb.get(result3).message)
def testInsertAlertType(self):
mytest = Alert()
result = mytest.put()
self.assertEqual(None, ndb.get(result).type)
mytest1 = Alert(type='Bus')
result1 = mytest1.put()
self.assertEqual('Bus', ndb.get(result1).type)
mytest2 = Alert(type='Route')
result2 = mytest2.put()
self.assertEqual('Route, ndb.get(result2).type)
mytest3 = Alert(type='Regional')
result3 = mytest3.put()
self.assertEqual('Regional', ndb.get(result3).type)
