python 学习笔记 12 -- 写一个脚本获取城市天气信息

最近在玩树莓派，前面写过一篇在树莓派上使用1602液晶显示屏，那么能够显示后最重要的就是显示什么的问题了。最容易想到的就是显示时间啊，CPU利用率啊，IP地址之类的。那么我觉得呢，如果能够显示当前时间、温度也是甚好的，作为一个桌面小时钟还是很精致的。

1. 目前有哪些工具

目前比较好用的应该是 weather-util， 之前我获取天气信息一般都是通过它。

使用起来也很简单：

（1） Debian/Ubuntu 用户使用
sudo apt-get install weather-util 安装

（2） 使用可以直接终端下输入“ weather 城市名拼音” 查询，如

所以我们只要对查询语句进行一点点改进即可得到：

$ weather lanzhou

Searching via name...

[using result Lanzhou / Zhongchuan, China]

Current conditions at Lanzhou / Zhongchuan, China (ZLLL) 36-01-12N 103-45E

Last updated Jul 05, 2014 - 03:00 AM EDT / 2014.07.05 0700 UTC

Temperature: 86 F (30 C)

Relative Humidity: 28%

Wind: Variable at 4 MPH (4 KT)

Sky conditions: partly cloudy

所以，如果我在树莓派的程序/脚本中直接调用，就可很轻松的得到我想要的天气信息。

$ weather "lanzhou" | grep "Temperature" | cut -d "(" -f2 | cut -d ")" -f1

30 C

但是，做很多事都怕都这么一个但是啊！！！在树莓派上调用weather 查询天气耗时太可怕了：

$ time weather "lanzhou" | grep "Temperature" | cut -d "(" -f2 | cut -d ")" -f1

25 C

real 4m30.381s

user 4m27.010s

sys 0m1.560s

而在我的台式机端，耗时虽然比较长，但是也消耗了15s 之久。

$ time weather "lanzhou" | grep "Temperature" | cut -d "(" -f2 | cut -d ")" -f1

30 C

real 0m15.609s

user 0m15.405s

sys 0m0.203s

那么为什么使用weather 查询个天气需要耗时这么久？根据参考资料[1] 中所述，“Weather”工具从METAR(Meteorological Terminal Aviation Routine Weather Report,
航空例行天气报告)， NOAA (the USA National Oceanic and Atmospheric Administration,
美国国家海洋和大气管理局)和NWS (the USA National Weather Service,
美国国家气象服务)检索获取天气状况和预报信息。这使得这个工具主要以美国为中心，然而你也能获取到全球有国际机场的地区的天气信息。

原来这个工具是针对美国国土以及一些国际机场，既然我们不在美国国土，那么我们查一下国际机场的信息试试时间会不会快上很多：

$ time weather egll

Searching via station...

[caching result London / Heathrow Airport, United Kingdom]

Current conditions at London / Heathrow Airport, United Kingdom (EGLL) 51-29N 000-27W 0M

Last updated Jul 05, 2014 - 03:20 AM EDT / 2014.07.05 0720 UTC

Temperature: 62 F (17 C)

..

real 0m3.958s

user 0m2.890s

sys 0m0.072s

我们可以看到，速度确实要快上不少。（而且由于此工具的优化，第二次查询时，耗时可达 0.08s 级别！）看来我们使用现成的 weather-util 工具查询天气信息算是泡汤了。

我们需要更快的速度、更全面的区域，所以我觉得我们需要本土的天气查询。

2. 使用什么接口

上文讨论到，我们需要使用本土的天气查询，百度上找到了一个帖子（见参考资料[2]），上面给出了中国天气网（www.weather.com.cn）提供的查询结构，并且下面还给出了城市编码。在此感谢一下作者。

那么我们在此查看一下这些接口会给我们哪些新息：

接口1 ： http://www.weather.com.cn/data/cityinfo/***.html
（“***” 处替换为你想要查询的城市的编码，下同）

这个网址会给我们如下信息（我感觉有用的信息已用颜色标出）：

{"weatherinfo":{"city":"兰州","cityid":"101160101","temp1":"19℃","temp2":"33℃","weather":"晴","img1":"n0.gif","img2":"d0.gif","ptime":"18:00"}}

所以我们可以通过此接口得到城市的气候（"weather"），低温（"temp1"），高温("temp2")。

接口2 ： http://www.weather.com.cn/data/sk/***.html
这个接口会给我们如下信息：

{"weatherinfo":{"city":"兰州","cityid":"101160101","temp":"26","WD":"北风","WS":"1 级","SD":"36%","WSE":"1","time":"21:00","isRadar":"1","Radar":"JC_RADAR_AZ9931_JB"}}

所以通过此结构我们可以得到的是一个城市的实时温度("temp")。

接口3：http://m.weather.com.cn/data/***.html

感觉这个接口给的信息比较杂乱：

{"weatherinfo":{"city":"兰州","city_en":"lanzhou","date_y":"2014年3月4日","date":"","week":"星期二","fchh":"11","cityid":"101160101","temp1":"11℃~0℃","temp2":"12℃~0℃","temp3":"11℃~2℃","temp4":"13℃~2℃","temp5":"15℃~2℃","temp6":"18℃~4℃","tempF1":"51.8℉~32℉","tempF2":"53.6℉~32℉","tempF3":"51.8℉~35.6℉","tempF4":"55.4℉~35.6℉","tempF5":"59℉~35.6℉","tempF6":"64.4℉~39.2℉","weather1":"晴","weather2":"多云","weather3":"多云转小雨","weather4":"小雨转多云","weather5":"多云转晴","weather6":"晴","img1":"0","img2":"99","img3":"1","img4":"99","img5":"1","img6":"7","img7":"7","img8":"1","img9":"1","img10":"0","img11":"0","img12":"99","img_single":"0","img_title1":"晴","img_title2":"晴","img_title3":"多云","img_title4":"多云","img_title5":"多云","img_title6":"小雨","img_title7":"小雨","img_title8":"多云","img_title9":"多云","img_title10":"晴","img_title11":"晴","img_title12":"晴","img_title_single":"晴","wind1":"微风","wind2":"微风","wind3":"微风","wind4":"微风","wind5":"微风","wind6":"微风","fx1":"微风","fx2":"微风","fl1":"小于3级","fl2":"小于3级","fl3":"小于3级","fl4":"小于3级","fl5":"小于3级","fl6":"小于3级","index":"冷","index_d":"天气冷，建议着棉服、羽绒服、皮夹克加羊毛衫等冬季服装。年老体弱者宜着厚棉衣、冬大衣或厚羽绒服。","index48":"冷","index48_d":"天气冷，建议着棉服、羽绒服、皮夹克加羊毛衫等冬季服装。年老体弱者宜着厚棉衣、冬大衣或厚羽绒服。","index_uv":"中等","index48_uv":"弱","index_xc":"较适宜","index_tr":"适宜","index_co":"较舒适","st1":"10","st2":"1","st3":"12","st4":"0","st5":"11","st6":"2","index_cl":"适宜","index_ls":"基本适宜","index_ag":"极易发"}}

3. 实现

有了这些接口信息，外加城市代码文件，我们可以很轻松滤清所要做的顺序：
根据用户输入的城市在代码文件中查找城市代码，使用城市代码替换结构中的代码，使用python的urllib2 获取此结构的内容，然后解析并输出。

3.1 搜索城市代码

城市代码文件中的内容格式为：

101010100=北京

101010200=海淀

101010300=朝阳

而整个文件有2000多行，如果逐行读取并匹配，个人觉得效率还是不高，所以我使用 os.popen 调用shell 中的grep 来进行全文匹配查找。

def get_city_code(test_city, citycode_f = "citycode"):
    if os.path.exists(citycode_f):      # 先检查城市代码文件是否存在
        ret = os.system("grep %s %s >/dev/null" % (test_city,citycode_f))   
        if ret == 0:        # 检测os.system 的返回值从而得知城市名是否在代码文件中存在
            grep_content = os.popen("grep %s %s" % (test_city,citycode_f)).read()
            city_code = grep_content.split("=")[0]  # 如果存在获取匹配行,如"101010100=北京"，使用‘=’作为分隔符获取第一列
            return city_code
        else:
            print "\t搜索\"%s\"失败，请重新搜索（市级城市）" % test_city
            sys.exit(-1)
    else:
        print "当前目录没有\"%s\"文件，无法搜索" % citycode_f
        sys.exit(-1)

3.2 替换结构代码获取内容

通过调用上述函数得到城市代码，使用如下的简单字符串方法即可修改城市url ：

weather_url = "http://www.weather.com.cn/data/sk/%s.html" % city_code

info_url = "http://www.weather.com.cn/data/cityinfo/%s.html" % city_code

所以我们此时只需要使用urllib2 模块获取url 的内容，并对内容进行解析。

获取url 内容，这里使用了一个简单的函数：

def use_urllib2(url):  
    try:  
        f = urllib2.urlopen(url, timeout=5).read()  
    except urllib2.URLError, e:    
        print e.reason  
        return -1
    else:
        return f

所以使用此函数打开上面的url ，打开后分析返回值如果是 -1 则打开出错，给出错误信息并调用 sys.exit 退出程序。如果正常打开，对内容进行解析获取关心的内容即可：

weather_url = "http://www.weather.com.cn/data/sk/%s.html" % city_code   
    wea_str = use_urllib2(weather_url)
    if wea_str == -1:
        print "Sorry that I can't find the weather info of this city now, please check or retry"
        sys.exit(-1)
    else:
        wea_list = wea_str.split(",")
        for item in wea_list:
            if "city" in item and "cityid" not in item:
                city_name = get_content(item)
            if "temp" in item:
                city_temp = get_content(item)
            if "time" in item:
                update_time = get_content(item)

3.3 中文输入or英文输入

对于城市的代码文件，我们可以看出：如果想要匹配，我们只能使用中文城市名。那么如果我想用城市名的拼音怎么办？所以只能从根本处借决问题。那么怎么解决？一个个手动打进去？2000多行哦！ 这时候我想到了上面的接口3 中有这么一条有用的信息："city":"兰州","city_en":"lanzhou"

貌似一切都要变得简单。我将原来原来拷贝过来的 citycode 改名为citycode1。并写了一个脚本：

#!/usr/bin/python
# -*- coding: utf-8 -*-
import urllib2
import sys
import os
def file_append(filename,string):
    try:
        f = open(filename,"a")
        try:
            f.write(string+"\n")
        except IOError:
            print("Append %s occurs some errors" % filename)
            return 0
        except:
            print("Append %s occurs some errors" % filename)
            return 0
        else:
            return 1
    except IOError:
        print("open %s file error" % filename)
        return 0
        traceback.print_exc()
    finally:
        f.close()

def use_urllib2(url):
    try:
        f = urllib2.urlopen(url, timeout=5).read()
    except urllib2.URLError, e:
        print e.reason
        return -1
    else:
        return f

def deal_with_file(sour_file, tar_file):
    try:
        open_file = file(sour_file)
        file_len = len(open(sour_file,'rU').readlines())    # 获取文件长度

        for cur_line in range(0, file_len):
            content = open_file.readline().strip("\n")
            print "No.%d\tNow dealing with %s" %(cur_line, content)
            if len(content) == 0:
                file_append(tar_file, content)
            else:
                city_code = content.split("=")[0]
                get_city_content_url = "http://m.weather.com.cn/data/%s.html" % city_code
                web_content = use_urllib2(get_city_content_url)
                if web_content == -1:
                    print "%s has no the web content, pass it " % content
                else:
                    web_list = web_content.split(",")
                    for item in web_list:
                        if "city_en" in item:
                            dealed_content = content+"="+item.split(":")[1].replace("\"", "")
                            file_append(tar_file, dealed_content)
            
    except IOError:
        print("open %s file error" % sour_file)
        sys.exit(-1)
    finally:
        print("we will close file %s" % sour_file)
        open_file.close()

#### Real start here
file_name = "citycode1"        # 原始文件，格式为"101010100=北京"
result_f = "citycode"            # 最终将改完的格式为"101010100=北京=beijing"内容存入此文件
if os.path.exists(result_f):
    os.remove(result_f)
deal_with_file(file_name, result_f)

此时我们打开城市代码文件，里面的内容都是这个格式的

：

101010100=北京=beijing

101010200=海淀=haidian

101010300=朝阳=chaoyang

注： 修改完的的城市代码信息，本来我上传csdn下载，但是苦于被删除，所以大家可以克隆此仓库。

3.4 中文输出or英文输出

那么这个程序打印到底应该是中文还是英文呢？我采取了本系列前文《python 学习笔记 11 -- 使用参数使你的程序变得更性感
》中的方法，添加参数选择"-zh"/"-en"。如果使用的是 "./getweather.py -zh **" 则以中文输出（不填加选项，直接使用 "./getweather.py ***" 查询采取的默认输出也是中文，不过可通过修改代码中的初始值设置进行修改），如果使用"./getweather.py -en **" 则以英文输出。

使用的方法很简单：设置一个全局变量叫 "chinese_flag" 并设初始值为1。获取用户参数时，如果有 "-zh" 则不变，如果有"-en" 则改为0。任何printf 打印时对此变量进行判断后使用不同的输出。

4. 总结

本文中虽然总共写了不到300 行的python 代码，但是最终效果以及健壮性还是很不错的：

$ ./getweather.py 兰州

你只使用一个参数"兰州"（且不是"-h"），将其作为城市名搜索

timed out

对不起，暂时搜寻不到此城市信息，请检查或稍后再试

$ ./getweather.py 兰州

你只使用一个参数"兰州"（且不是"-h"），将其作为城市名搜索

今天"兰州市"天气是:

========================

今日天气 : 晴

实时温度 : 32℃

今日的温度区间 : 19℃ ~ 33℃

更新时间: 16:10

========================

更多请参见：http://www.weather.com.cn/weather/101160101.shtml

$ ./getweather.py lanzhou

你只使用一个参数"lanzhou"（且不是"-h"），将其作为城市名搜索

今天"兰州市"天气是:

========================

今日天气 : 晴

实时温度 : 32℃

今日的温度区间 : 19℃ ~ 33℃

更新时间: 16:10

========================

更多请参见：http://www.weather.com.cn/weather/101160101.shtml

$ ./getweather.py -en lanzhou

The weather info of "兰州":

========================

The weather today is : 晴

Currenttly the temperature is : 32℃

The weather section today is : 19℃ ~ 33℃

Update time: 16:10

========================

See more: http://en.weather.com.cn/weather/101160101.shtml
$ ./getweather.py -en lanzhou1

Search city "lanzhou1" fail, please check the city and retry(at lease city size)

$ ./getweather.py -zg lanzhou1

你选择了错误的结果选项，请重新输入或者使用'-h'查看如何使用

$ ./getweather.py -zh lanzhou1

搜索"lanzhou1"失败，请重新搜索（市级城市）

关于本文代码，我已经提交在前文《Shell 脚本小试牛刀（番外） -- 捷报
》中提到的github 仓库中。且git 仓库已更名： cool_script 。

可直接克隆仓库： git clone https://github.com/longerzone/easy_script .然后进入 get_weather 目录查看。

================

参考资料：

[1] 关于weather-util ： http://www.iraspberrypi.com/archives/539
[2] 关于中国天气网的结构： http://www.iteye.com/topic/1106241
================================================

注： 转载注明出处： http://blog.csdn.net/longerzone