|
''' 从网络抓取各个城市气温信息,并依次显示:
北京:15~20
天津:17~22
长春:12~18
......
如果一次抓取所有城市天气再显示,显示第一个城市气温时,有很高的延迟,并且
浪费存储空间,我们期望以“用时访问”的策略,并且能把所有城市气温封装到一个
对象里,可用for语句进行迭代,如何解决?
'''
from collections import Iterable, Iterator
from urllib import request
import urllib
import json
import gzip
'''
1.实现一个迭代器对象WeatherIterator,__next__方法每次返回一个城市的气温,
迭代器对象的实现需要继承 Iterator
'''
class WeatherIterator(Iterator):
def __init__(self, cities):
'''
构造器需要接收一个可迭代的对象,在这里这个cities是一个城市列表
index用于记录迭代次数,也用于充当下标索引
'''
self.cities = cities
self.index = 0
def getWeather(self, city):
'''
抓取城市天气信息
'''
url_str = 'http://wthrcdn.etouch.cn/weather_mini?city=' + city
# 转换URL的中文,safe参数指定哪些符号不转换
url = urllib.parse.quote(url_str, safe='/:?=')
resp = request.urlopen(url).read()
# 由于返回的的数据是gzip的压缩格式,所以需要解压返回的数据
resp = gzip.decompress(resp)
# 将字节转换成字符串,编码设置为utf-8
json_data = str(resp, encoding = 'utf-8')
# 解析json数据
data = json.loads(json_data)['data']['forecast'][0]
return '%s: %s , %s ' % (city, data['low'], data['high'])
def __next__(self):
if self.index == len(self.cities):
# 被迭代的次数等于迭代对象的长度时就需要抛异常
raise StopIteration
# 从城市列表中迭代取出城市名称,每被取出一个,就需要增加一次迭代次数
city = self.cities[self.index]
self.index += 1
# 返回城市的天气数据
return self.getWeather(city)
'''
2.实现一个可迭代对象WeatherIterable,__iter__方法返回一个迭代器对象
可迭代对象的实现需要继承 Iterable
'''
class WeatherIterable(Iterable):
def __init__(self, cities):
self.cities = cities
def __iter__(self):
# 返回迭代器对象的实例
return WeatherIterator(self.cities)
cities = ['北京', '上海', '广州', '深圳']
for x in WeatherIterable(cities):
print(x) |
|
|