manage.py

from flask.ext.script import Manager
from webapp import app
from importer.osm_importer import run as osm_import
from importer.dwd_importer import import_dwd_db, import_dwd_json, import_dwd_from_json
from models.map import Osm_Admin, Station, Observation, Country, State, District, Cities, ContribState, ContribDistrict, \
    GFSImport, GFS
from webapp import db
import os
import sys
import json

manager = Manager(app)


@manager.option('--imposm', action='store_true', dest='imposm', default=False,
                help="import data with imposm3 from pfb file")
@manager.option('--simplify', action='store_true', dest='simplify', default=False,
                help="simplify all polygons while preserving topology")
@manager.option('--load', action='store_true', dest='load', default=False,
                help="create map tables for the app from the tables created by imposm")
@manager.option('--drop_tables', action='store_true', dest='drop_tables', default=False,
                help="drop the osm_admin and osm_places tables from the imposm import")
def import_osm(imposm, simplify, load, drop_tables):
    """import data osm map data from pbf file"""
    osm_import(imposm, simplify, load, drop_tables)


@manager.option('-t', '--tables', dest='tables', help="osm, map, dwd, gfs, contrib", required=True)
def drop_tables(tables):
    """"drop the tables for osm (osm_admin, osm_places), map (country, state, district, cities),
    dwd (station, observation) or contrib (contrib_state, contrib_district)"""

    if tables == 'osm':
        Osm_Admin.__table__.drop(db.engine, checkfirst=True)
        Osm_Places.__table__.drop(db.engine, checkfirst=True)
    elif tables == 'map':
        Country.__table__.drop(db.engine, checkfirst=True)
        State.__table__.drop(db.engine, checkfirst=True)
        District.__table__.drop(db.engine, checkfirst=True)
        Cities.__table__.drop(db.engine, checkfirst=True)
    elif tables == 'dwd':
        db.session.query(Observation).delete()
        db.session.query(Station).delete()
        db.session.commit()
        Observation.__table__.drop(db.engine, checkfirst=True)
        Station.__table__.drop(db.engine, checkfirst=True)
    elif tables == 'gfs':
        GFSImport.__table__.drop(db.engine, checkfirst=True)
        GFSImport.__table__.drop(db.engine, checkfirst=True)
    elif tables == 'contrib':
        ContribState.__table__.drop(db.engine, checkfirst=True)
        ContribDistrict.__table__.drop(db.engine, checkfirst=True)

DEFAULT_FILE_NAME = 'data/weather.json'
@manager.option('--to_json', action='store_true', dest='to_json', default=False,
                help="import data to json")
@manager.option('--from_json', action='store_true', dest='load_from_json', default=False,
                help="import from json file")
def import_dwd(to_json, load_from_json, file=DEFAULT_FILE_NAME):
    """import dwd weather data, either directly to the database or to/from a json file"""
    if to_json:
        if os.path.isfile(file):
            override = input('%s already exists. Override? [y/N]' % file)
            if override == 'y' or override == 'yes':
                print("overriding ", file)
            else:
                print("aborting...")
                return
        import_dwd_json()
    elif load_from_json:
        data = json.load(open(file, 'r'))
        import_dwd_from_json(data)
    else:
        import_dwd_db()

@manager.command
def calculate_contrib_area():
    db.create_all()
    ContribState.fill()
    ContribDistrict.fill()


if __name__ == "__main__":
    manager.run()