wttr.in/lib/wttr.py

# vim: set encoding=utf-8

import gevent
from gevent.wsgi import WSGIServer
from gevent.queue import Queue
from gevent.monkey import patch_all
from gevent.subprocess import Popen, PIPE, STDOUT
patch_all()

import os
import re
import time

from translations import get_message, FULL_TRANSLATION, PARTIAL_TRANSLATION, SUPPORTED_LANGS
from globals import WEGO, PYPHOON, CACHEDIR, ANSI2HTML, \
                    NOT_FOUND_LOCATION, DEFAULT_LOCATION, \
                    log, error

def _is_invalid_location(location):
    if '.png' in location:
        return True

def get_wetter(location, ip, html=False, lang=None, query=None, location_name=None, full_address=None):

    def get_filename(location, lang=None, query=None, location_name=None):
        location = location.replace('/', '_')
        timestamp = time.strftime( "%Y%m%d%H", time.localtime() )

        imperial_suffix = ''
        if query.get('use_imperial', False):
            imperial_suffix = '-imperial'

        lang_suffix = ''
        if lang is not None:
            lang_suffix = '-lang_%s' % lang

        if query != None:
            query_line = "_" + "_".join("%s=%s" % (key, value) for (key, value) in query.items())
        else:
            query_line = ""

        if location_name is None:
            location_name = ""
        return "%s/%s/%s%s%s%s%s" % (CACHEDIR, location, timestamp, imperial_suffix, lang_suffix, query_line, location_name)

    def save_weather_data(location, filename, lang=None, query=None, location_name=None, full_address=None):
        ansi_escape = re.compile(r'(\x9B|\x1B\[)[0-?]*[ -\/]*[@-~]')
        def remove_ansi(sometext):
            return ansi_escape.sub('', sometext)
     
        if _is_invalid_location( location ):
            error("Invalid location: %s" % location)
        
        NOT_FOUND_MESSAGE_HEADER = ""
        while True:
            if location == NOT_FOUND_LOCATION:
                location_not_found = True
                location = DEFAULT_LOCATION
            else:
                location_not_found = False
            
            cmd = [WEGO, '--city=%s' % location]

            if query.get('inverted_colors'):
                cmd += ['-inverse']

            if query.get('use_ms_for_wind'):
                cmd += ['-wind_in_ms']

            if query.get('narrow'):
                cmd += ['-narrow']

            if lang and lang in SUPPORTED_LANGS:
                cmd += ['-lang=%s'%lang]

            if query.get('use_imperial', False):
                cmd += ['-imperial']

            if location_name:
                cmd += ['-location_name', location_name]

            p = Popen(cmd, stdout=PIPE, stderr=PIPE)
            stdout, stderr = p.communicate()
            if p.returncode != 0:
                print "ERROR: location not found: %s" % location
                if 'Unable to find any matching weather location to the query submitted' in stderr:
                    if location != NOT_FOUND_LOCATION:
                        NOT_FOUND_MESSAGE_HEADER = u"ERROR: %s: %s\n---\n\n" % (get_message('UNKNOWN_LOCATION', lang), location)
                        location = NOT_FOUND_LOCATION
                        continue
                error(stdout + stderr)
            break

        dirname = os.path.dirname(filename)
        if not os.path.exists(dirname):
            os.makedirs(dirname)
        
        if location_not_found:
            stdout += get_message('NOT_FOUND_MESSAGE', lang).encode('utf-8')
            stdout = NOT_FOUND_MESSAGE_HEADER.encode('utf-8') + stdout

        if 'days' in query:
            if query['days'] == '0':
                stdout = "\n".join(stdout.splitlines()[:7]) + "\n"
            if query['days'] == '1':
                stdout = "\n".join(stdout.splitlines()[:17]) + "\n"
            if query['days'] == '2':
                stdout = "\n".join(stdout.splitlines()[:27]) + "\n"

        if query.get('no-caption', False):
            first = stdout.splitlines()[0].decode('utf-8')
            rest = stdout.splitlines()[1:]

            separator = None
            if ':' in first:
                separator = ':'
            if u'：' in first:
                separator = u'：'

            if separator:
                first = first.split(separator,1)[1]
                stdout = "\n".join([first.strip().encode('utf-8')] + rest) + "\n"

        if query.get('no-terminal', False):
            stdout = remove_ansi(stdout)

        if query.get('no-city', False):
            stdout = "\n".join(stdout.splitlines()[2:]) + "\n"

        if full_address:
            line = "%s: %s [%s]\n" % (get_message('LOCATION', lang).encode('utf-8'), full_address.encode('utf-8'), location)
            stdout += line

        if query.get('padding', False):
            lines = [x.rstrip() for x in stdout.splitlines()]
            max_l = max(len(remove_ansi(x).decode('utf8')) for x in lines)
            last_line = " "*max_l + "   .\n"
            stdout = " \n" + "\n".join("  %s  " %x for x in lines) + "\n" + last_line 

        open(filename, 'w').write(stdout)

        cmd = ["bash", ANSI2HTML, "--palette=solarized"]
        if not query.get('inverted_colors'):
            cmd += ["--bg=dark"]

        p = Popen(cmd,  stdin=PIPE, stdout=PIPE, stderr=PIPE )
        stdout, stderr = p.communicate(stdout)
        if p.returncode != 0:
            error(stdout + stderr)

        if query.get('inverted_colors'):
            stdout = stdout.replace('<body class="">', '<body class="" style="background:white;color:#777777">')

        open(filename+'.html', 'w').write(stdout)

    filename = get_filename(location, lang=lang, query=query, location_name=location_name)
    if not os.path.exists(filename):
        save_weather_data(location, filename, lang=lang, query=query, location_name=location_name, full_address=full_address)
    if html:
        filename += '.html'

    return open(filename).read()

def get_moon(location, html=False):
    date = None
    if '@' in location:
        date = location[location.index('@')+1:]
        location = location[:location.index('@')]

    cmd = [PYPHOON]
    if date:
        try:
            dateutil.parser.parse(date)
        except:
            pass
        else:
            cmd += [date]

    p = Popen(cmd, stdout=PIPE, stderr=PIPE)
    stdout, stderr = p.communicate()

    if html:
        p = Popen(["bash", ANSI2HTML, "--palette=solarized", "--bg=dark"],  stdin=PIPE, stdout=PIPE, stderr=PIPE)
        stdout, stderr = p.communicate(stdout)
        if p.returncode != 0:
            error(stdout + stderr)

    return stdout