#!/isan/bin/nxpython
# -*- coding:utf-8 -*-

################################################################
# File:   Show Weather
#
##################################################################

import signal
import time
import threading
import sys
import commands
import json

### Imports NX-OS SDK package
import nx_sdk_py

def get_location_id(location = "tokyo"):
    location_code = {
        "tokyo": "130010",
        "sapporo": "016010",
        "sendai": "040010",
        "nagoya": "230010",
        "osaka": "270000",
        "hiroshima": "340010",
        "fukuoka": "400010"
    }

    return location_code.get(location, "130010")

def get_weather(location = "tokyo", proxy = "", vrf = "default"):

    global tmsg
    command = "ip netns exec " + vrf

    base_url = "http://weather.livedoor.com/forecast/webservice/json/v1?city="
    url = base_url + get_location_id(location)

    command += " curl -s " + url

    if proxy != "":
        command += " -x " + proxy

    result = commands.getoutput(command)

    if len(result) == 0:
        return None

    return json.loads(result)

def weather_detail(result):
    location = result["location"]["city"].encode('utf-8')
    summary = result["forecasts"][0]['telop'].encode('utf-8')
    max_temp = result["forecasts"][0]['temperature']['max']
    if max_temp is not None:
        max_temp = float(max_temp['celsius'])

    min_temp = result["forecasts"][0]['temperature']['min']
    if min_temp is not None:
        min_temp = float(min_temp['celsius'])

    link = result["link"].encode('utf-8')

    message = "今日の%sの天気は、%s\n" % (location, summary)

    if max_temp is not None:
        message += "最高気温は、%.2f度\n" % max_temp
    if min_temp is not None:
        message += "最低気温は、%.2f度\n" % min_temp

    message += link + '\n'

    return message

def weather_summary(result):
    return result["forecasts"][0]['telop'].encode('utf-8')

class pyCmdHandler(nx_sdk_py.NxCmdHandler):
    location = "tokyo"
    vrf = "default"
    proxy = ""

    def postCliCb(self,clicmd):
        global cliP, location

        if "show_weather" in clicmd.getCmdName():
            weather_result = get_weather(self.location, self.proxy, self.vrf)

            if weather_result is None:
                clicmd.printConsole("Could Not access weather API\n")
                return False

            message = ""

            if "detail" in clicmd.getCmdLineStr():
                message = weather_detail(weather_result)
            else:
                message = weather_summary(weather_result) + "\n"

            clicmd.printConsole(message)
        elif "config_weather_location" in clicmd.getCmdName():
            if "no" in clicmd.getCmdLineStr():
                self.location = "tokyo";
            else:
                self.location = nx_sdk_py.void_to_string(clicmd.getParamValue("<location>"))

        elif "config_weather_vrf" in clicmd.getCmdName():
            if "no" in clicmd.getCmdLineStr():
                self.vrf = "default";
            else:
                temp_vrf = nx_sdk_py.void_to_string(clicmd.getParamValue("<vrf>"))
                if temp_vrf == "all":
                    clicmd.printConsole("Can not configure VRF to all")
                    tmsg.event("Can not configure VRF to all")
                    return False
                self.vrf = temp_vrf

        elif "config_weather_proxy" in clicmd.getCmdName():
            if "no" in clicmd.getCmdLineStr():
                self.proxy = "";
            else:
                self.proxy = nx_sdk_py.void_to_string(clicmd.getParamValue("<proxy>"))

        return True

def sdkThread(name,val):
    global cliP, sdk, event_hdlr, tmsg

    sdk = nx_sdk_py.NxSdk.getSdkInst(len(sys.argv), sys.argv)
    if not sdk:
        return

    sdk.setAppDesc('Weather App for Nexus')

    tmsg = sdk.getTracer()

    tmsg.event("[%s] Started service" % sdk.getAppName())

    cliP = sdk.getCliParser()

    nxcmd = cliP.newShowCmd("show_weather", "[detail]")
    nxcmd.updateKeyword("detail", "For Detail Weather Information")

    nxcmd1 = cliP.newConfigCmd("config_weather_location", "location <location>")
    nxcmd1.updateKeyword("location", "Location to check weather")

    str_attr = nx_sdk_py.cli_param_type_string_attr()
    str_attr.length = 25;
    str_attr.regex_pattern = "^[a-zA-Z]+$";
    nxcmd1.updateParam("<location>", "Location Name", nx_sdk_py.P_STRING, str_attr, len(str_attr))

    nxcmd2 = cliP.newConfigCmd("config_weather_vrf", "use-vrf <vrf>")
    nxcmd2.updateKeyword("use-vrf", "Configures Weather to use the selected VRF to fetch weather information")
    nxcmd2.updateParam("<vrf>", "VRF name", nx_sdk_py.P_VRF)

    nxcmd3 = cliP.newConfigCmd("config_weather_proxy", "proxy <proxy>")
    nxcmd3.updateKeyword("proxy", "Configures Weather to use the proxy server Ex) proxy.esl.cisco.com:8080 ")

    str_attr_url = nx_sdk_py.cli_param_type_string_attr()
    str_attr_url.length = 128;
    str_attr_url.regex_pattern = "^([^/ :]+):?([0-9]*)$";
    nxcmd3.updateParam("<proxy>", "Proxy Server", nx_sdk_py.P_STRING, str_attr_url, len(str_attr_url))

    mycmd = pyCmdHandler()
    cliP.setCmdHandler(mycmd)

    cliP.addToParseTree()

    sdk.startEventLoop()

    tmsg.event("Service Quitting...!")

    nx_sdk_py.NxSdk.__swig_destroy__(sdk)

cliP = 0
sdk  = 0
tmsg = 0

### create a new sdkThread to setup SDK service and handle events.
sdk_thread = threading.Thread(target=sdkThread, args=("sdkThread",0))
sdk_thread.start()

sdk_thread.join()