manage.py

from flask_script import Manager, Server, Shell, prompt_pass
from WeatherServer.api import Province, City, Country, WeatherHistory
from WeatherServer.admin import AdminUser
from WeatherServer import app
import json


def _make_context():
    return dict(Province=Province, City=City,
                Country=Country, WeatherHistory=WeatherHistory)

manager = Manager(app)
manager.add_command('runserver', Server(host='0.0.0.0', port='5000',
                                        use_debugger=True))
manager.add_command('shell', Shell(make_context=_make_context))


@manager.command
def make_json():
    provinces = []
    for province in Province.objects:
        province_json = make_province(province)
        provinces.append(province_json)
    res = dict(provinces=provinces)
    with open('locations.json', 'w') as f:
        f.write(json.dumps(res))


@manager.command
def get_weather():
    weather_set = set()
    for country in Country.objects:
        if country.weather_infos:
            weather_info = country.weather_infos[-1]
            realtime = weather_info.realtime
            forecasts = weather_info.forecast
            weather = realtime.weather
            if weather and weather not in weather_set:
                weather_set.add(weather)
            # if forecasts:
            #     for forecast in forecasts:
            #         if forecast.weather not in weather_set:
            #             weather_set.add(forecast.weather)
    print(weather_set)


@manager.command
def save_json():
    c = WeatherHistory.objects.all()[100]
    from collections import defaultdict
    temp = defaultdict()
    for weather in c.history_infos:
        year, month, year_month_str = get_year_month(weather)
        



@manager.command
def superuser():
    """Create a superuser.
       Default account is admin@test.com, password is admin."""
    email = prompt_pass("Input your email.(Default is admin@test.com)",
                        default='admin@test.com')
    password = prompt_pass("Input your passwod.(Default is admin)",
                           default='admin')
    user = AdminUser.objects(email=email).first()
    if user is None:
        user = AdminUser(email=email, username='admin', password=password)
        user.save()
        print('Create successful')
    else:
        print('Fail to create, maybe create a another new one?')


def make_province(province):
    province_json = json.loads(province.to_json())
    del province_json['_id']
    for city in province.cities:
        id = city.location_id
        province_json['cities'].append(make_city(city))
        province_json['cities'].remove(id)
    return province_json


def make_city(city):
    city_json = json.loads(city.to_json())
    del city_json['_id']
    for country in city.countries:
        id = country.location_id
        city_json['countries'].append(make_country(country))
        city_json['countries'].remove(id)
    return city_json


def make_country(country):
    country_json = json.loads(country.to_json())
    del country_json['_id']
    del country_json['weather_id']
    del country_json['weather_infos']
    return country_json


def get_year_month(weather):
    _datetime = weather.datetime
    year, month = _datetime.year, _datetime.month
    year_month_str = '%s%s' % (year, month)
    return year, month, year_month_str


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