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 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 get(self):
page_title = "Push Newsfeed"
entities = Alert.query().order(Alert.title)
template_values = self.make_template_dict(page_title=page_title, entities=entities)
template = JINJA_ENVIRONMENT.get_template("templates/pushnewsfeed.html")
self.response.write(template.render(template_values))
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)
def get(self):
# For endpoints to work, https must be used. Redirect to https
# if not running on localhost.
# parsed = urlparse(self.request.url)
# if parsed.scheme == 'http' and parsed.hostname != 'localhost':
# # Redirect to https.
# self.redirect("https" + self.request.url[4:])
# return
page_title = "Home"
entities = Alert.query().order(Alert.title)
current_bus = main.current_bus()
template_values = {
"page_title": page_title,
"entities": entities,
"headsigns": TripsHeadsigns.find(),
"current_bus": current_bus,
}
template = JINJA_ENVIRONMENT.get_template("templates/home.html")
self.response.write(template.render(template_values))
