utils.py

#!/usr/bin/env python
#
# Copyright 2007 Google Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import webapp2
import shared
from datamodel import *

class Clear(webapp2.RequestHandler):
    def get(self):
        if not shared.is_weekend():
            for spot in Spot.all().filter("future = ", False):
                spot.Release()
        
class InitSpots(webapp2.RequestHandler):
    def get(self):

        self.response.out.write('Removing all spots...')
        for spot in Spot.all():
            spot.delete()
        self.response.out.write(' OK!\r\n<br>')

        self.response.out.write('Creating %d inside spots...' % shared.INSIDE_SPOTS_COUNT)
        for i in xrange(shared.INSIDE_SPOTS_COUNT):
            spot = Spot.get_or_insert(str(i+1), number=i+1, free=True, reserved=None, car=None, comments='', future=False, outside=False)
            spot.put()
        self.response.out.write(' OK!\r\n<br>')

        self.response.out.write('Creating %d outside spots...' % shared.OUTSIDE_SPOTS_COUNT)
        for j in xrange(shared.OUTSIDE_SPOTS_COUNT):
            i = shared.INSIDE_SPOTS_COUNT + j
            spot = Spot.get_or_insert(str(i+1), number=i+1, free=True, reserved=None, car=None, comments='', future=False, outside=True)
            spot.put()
        self.response.out.write(' OK!\r\n<br>')

class InitCars(webapp2.RequestHandler):
    def get(self):
        self.response.out.write('Removing all cars...')
        for car in Car.all():
            car.delete()
        self.response.out.write(' OK!\r\n<br>')

class MigrateConfigSchema(webapp2.RequestHandler):
    def get(self):
        self.response.out.write('Upgrading configurations...')
        updated = Configuration.MigrateConfigurationSchema()
        self.response.out.write(' OK!\r\n<br>')
        self.response.out.write('Upgraded %d configurations!' % updated)

class TestTime(webapp2.RequestHandler):
    def get(self):
        self.response.out.write(shared.get_current_time())

app = webapp2.WSGIApplication([
                               ('/tasks/clear', Clear),
                               ('/tasks/initspots', InitSpots),
                               ('/tasks/initcars', InitCars),
                               ('/tasks/migrateconfigschema', MigrateConfigSchema),
                               ('/tasks/testtime', TestTime),
                               ],
                              debug=True)