获取上海地区AQI质量数据Python脚本

一个获取上海地区AQI质量的Python脚本

https://github.com/Teaing/Raspberrypi/blob/master/GetPmData_Shanghai.py

#!/usr/bin/env python

# -*- coding: utf-8 -*-

#Author:Tea

import time

import sys

import json

import random

import requests

reload(sys)

sys.setdefaultencoding('utf8')

def main():

	Get_PmData()

def Convert_Des_Text(aqi_data):

	if aqi_data:

		Pm_Tag = {'0':u'优',

				  '1':u'良',

				  '2':u'轻度污染',

				  '3':u'中度污染',

				  '4':u'重度污染',

				  '5':u'严重污染',

				  '6':u'数据不正常'}

		if (aqi_data >= 0 and aqi_data <= 50):

			Tmp_tag = Pm_Tag['0']

		elif (aqi_data >= 51 and aqi_data <= 100):

			Tmp_tag = Pm_Tag['1']

		elif (aqi_data >= 101 and aqi_data <= 150):

			Tmp_tag = Pm_Tag['2']

		elif (aqi_data >= 151 and aqi_data <= 200):

			Tmp_tag = Pm_Tag['3']

		elif (aqi_data >= 201 and aqi_data <= 300):

			Tmp_tag = Pm_Tag['4']

		elif (aqi_data >= 300):

			Tmp_tag = Pm_Tag['5']

		else:

			Tmp_tag = Pm_Tag['6']

	return Tmp_tag

def Get_PmData():

	try:

		Pm_Json_Data = Get_AmericanEmbassy_PmData()

		Convert_AmericanEmbassy_Data(Pm_Json_Data)

	except:

		Pm_Json_Data = Get_China_PmData()

		Convert_China_Data(Pm_Json_Data)

def Get_China_PmData():

	City_list = {'0':'beijing','1': 'shanghai'}

	Req_City = City_list['1']

	Req_Token = '5j1znBVAsnSf5xQyNQyq'

	Req_Ip = ''.join([random.randint(1,254).__str__(),'.',random.randint(1,254).__str__(),'.',random.randint(1,254).__str__(),'.',random.randint(1,254).__str__()])

	Req_Url = ''.join(['http://www.pm25.in/api/querys/pm2_5.json?city=',Req_City,'&token=',Req_Token])

	Req_headers = {'content-type': 'application/json',

				   'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64; rv:36.0) Gecko/20100101 Firefox/36.0',

				   'X-Forwarded-For': Req_Ip}

	Request_Result = requests.get(Req_Url,headers=Req_headers)

	Request_Code = Request_Result.status_code

	Request_Content = Request_Result.text

	return Request_Content

def Convert_China_Data(json_data):

	if json_data:

		Load_Json_Data = json.loads(json_data)

		Check_Site_Length = Load_Json_Data.__len__()

		print Load_Json_Data[0]["time_point"]

		for i in xrange(Check_Site_Length):

			Pm_Tag_Data = int(Load_Json_Data[i]["aqi"])

			Tmp_tag = Convert_Des_Text(Pm_Tag_Data)

			print Load_Json_Data[i]["position_name"],u'AQI:',Load_Json_Data[i]["aqi"],Tmp_tag

def Get_AmericanEmbassy_PmData():

	City_list = {'0':'Beijing','1': 'Shanghai'}

	Req_City = City_list['1']

	Req_Stime = int(round(time.time() * 1000)).__str__()

	Req_Ip = ''.join([random.randint(1,254).__str__(),'.',random.randint(1,254).__str__(),'.',random.randint(1,254).__str__(),'.',random.randint(1,254).__str__()])

	Req_Url = ''.join(['http://aqicn.org/aqicn/json/android/',Req_City,'/json?',Req_Stime])

	Req_headers = {'content-type': 'application/json',

				   'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64; rv:36.0) Gecko/20100101 Firefox/36.0',

				   'X-Forwarded-For': Req_Ip}

	Request_Result = requests.get(Req_Url,headers=Req_headers)

	Request_Code = Request_Result.status_code

	Request_Content = Request_Result.text

	return Request_Content

def Convert_AmericanEmbassy_Data(json_data):

	if json_data:

		Load_Json_Data = json.loads(json_data)

		#print Load_Json_Data["nearest"][0]["pol"]

		print u'时间:',time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(Load_Json_Data["time"])),u'AQI:',Load_Json_Data["aqi"]

		Check_Site_Length = Load_Json_Data["nearest"].__len__()

		for i in xrange(Check_Site_Length):

			Pm_Tag_Data = int(Load_Json_Data["nearest"][i]["v"])

			Tmp_tag = Convert_Des_Text(Pm_Tag_Data)

			print Load_Json_Data["nearest"][i]["nna"],Load_Json_Data["nearest"][i]["v"],Tmp_tag

if __name__ == '__main__':

	main()