#vim: fileencoding=utf-8 """ One-line output mode. Initial implementation of one-line output mode. [ ] forecast [ ] spark [ ] several locations [ ] location handling [ ] more preconfigured format lines [ ] add information about this mode to /:help """ import sys import re import datetime from astral import Astral from constants import WWO_CODE, WEATHER_SYMBOL, WIND_DIRECTION from weather_data import get_weather_data PRECONFIGURED_FORMAT = { '1': u'%c %t', '2': u'%c šŸŒ”ļø%t šŸŒ¬ļø%w', '3': u'%l: %c %t', '4': u'%l: %c šŸŒ”ļø%t šŸŒ¬ļø%w', } MOON_PHASES = ( u"šŸŒ‘", u"šŸŒ’", u"šŸŒ“", u"šŸŒ”", u"šŸŒ•", u"šŸŒ–", u"šŸŒ—", u"šŸŒ˜" ) def render_temperature(data): """ temperature (t) """ temperature = u'%sā°C' % data['temp_C'] if temperature[0] != '-': temperature = '+' + temperature return temperature def render_condition(data): """ condition (c) """ weather_condition = WEATHER_SYMBOL[WWO_CODE[data['weatherCode']]] return weather_condition def render_wind(data): """ wind (w) """ try: degree = data["winddirDegree"] except KeyError: degree = "" try: degree = int(degree) except ValueError: degree = "" if degree: wind_direction = WIND_DIRECTION[((degree+22)%360)/45] else: wind_direction = "" unit = ' km/h' wind = u'%s%s%s' % (wind_direction, data['windspeedKmph'], unit) return wind def render_location(data): """ location (l) """ return data['location'].title() def render_moonphase(_): """ A symbol describing the phase of the moon """ astral = Astral() moon_index = int( int(32.0*astral.moon_phase(date=datetime.datetime.today())/28+2)%32/4 ) return MOON_PHASES[moon_index] def render_moonday(_): """ An number describing the phase of the moon (days after the New Moon) """ astral = Astral() return str(int(astral.moon_phase(date=datetime.datetime.today()))) FORMAT_SYMBOL = { 'c': render_condition, 't': render_temperature, 'w': render_wind, 'l': render_location, 'm': render_moonphase, 'M': render_moonday, } def render_line(line, data): """ Render format `line` using `data` """ def render_symbol(match): """ Render one format symbol from re `match` using `data` from external scope. """ symbol_string = match.group(0) symbol = symbol_string[-1] if symbol not in FORMAT_SYMBOL: return '' render_function = FORMAT_SYMBOL[symbol] return render_function(data) return re.sub(r'%[^%]*[a-zA-Z]', render_symbol, line) def format_weather_data(format_line, location, data): """ Format information about current weather `data` for `location` with specified in `format_line` format """ current_condition = data['data']['current_condition'][0] current_condition['location'] = location output = render_line(format_line, current_condition) return output def wttr_line(location, query): """ Return 1line weather information for `location` in format `line_format` """ format_line = query.get('format', '') if format_line in PRECONFIGURED_FORMAT: format_line = PRECONFIGURED_FORMAT[format_line] weather_data = get_weather_data(location) output = format_weather_data(format_line, location, weather_data) output = output.rstrip("\n")+"\n" return output def main(): """ Function for standalone module usage """ location = sys.argv[1] query = { 'line': sys.argv[2], } sys.stdout.write(wttr_line(location, query)) if __name__ == '__main__': main()