#!/usr/bin/env python2.7
# -*- coding: utf-8 -*-

# Author		 : Ryan Yuan
# Date  		 : Dec 31, 2015
# Python Version : 2.7

import requests
from bs4 import BeautifulSoup
import json

import sys
from PyQt4 import QtGui
from PyQt4 import QtCore
class weather_predictor(QtGui.QWidget):
	def __init__(self):
		super(weather_predictor, self).__init__()
		self.city = None
		self.province = None
		self.initUI()
		self.Done = False
		self.err_msg = None

	def initUI(self):
		self.city_btn = QtGui.QPushButton(u'城市', self)
		self.province_btn = QtGui.QPushButton(u'省份', self)
		self.lookup_btn = QtGui.QPushButton(u'今日天气', self)

		# Create a Table to show the result of weather
		self.res_table = QtGui.QTableWidget(self)
		self.res_table.setColumnCount(0)
		self.res_table.setRowCount(5)
		self.res_table.resize(200,200)

		self.city_btn.clicked.connect(self.city_holder)
		self.province_btn.clicked.connect(self.province_holder)
		self.lookup_btn.clicked.connect(self.run)
		# layout ---------------------*
		hbox = QtGui.QHBoxLayout()
		hbox.addStretch(1)
		hbox.addWidget(self.city_btn)
		hbox.addWidget(self.province_btn)
		hbox.addWidget(self.lookup_btn)

		# 把水平布局放到垂直布局中
		vbox = QtGui.QVBoxLayout()
		vbox.addStretch(1)
		vbox.addLayout(hbox)
		# layout ---------------------*

		self.setLayout(vbox)
		self.resize(250,300)
		self.setWindowTitle(u'Ryan的天气预报')
	

	def city_holder(self):
		text, ok = QtGui.QInputDialog.getText(self, u'城市', 
			u'输入你的城市')
		self.city = unicode(text.toUtf8(), encoding="UTF-8")


	def province_holder(self):
		text, ok = QtGui.QInputDialog.getText(self, u'省份', 
			u'输入你的省份')
		self.province = unicode(text.toUtf8(), encoding="UTF-8")



	def find_city_code(self):

		res = open('citylist.xml').read()
		soup = BeautifulSoup(res)

		return soup.find_all(d4=self.province, d2=self.city)[0]['d1']


	def predict(self):

		s = requests.Session()
		info = s.get('http://wthrcdn.etouch.cn/weather_mini?citykey='+self.find_city_code()).content
		data = json.loads(info)

		self.city = data['data']['city']
		self.date = data['data']['forecast'][0]['date']
		self.high = data['data']['forecast'][0]['high']
		self.low = data['data']['forecast'][0]['low']
		self.type = data['data']['forecast'][0]['type']

	def run(self):
		
		try:
			self.predict()
		except IndexError:
			self.err_msg = u'无法找到该省份中的该城市'
			
		# 每运行一次run()，就insert a new Column
		column = self.res_table.columnCount()
		self.res_table.insertColumn(column)

		if self.err_msg:
			self.res_table.setItem(0, column, QtGui.QTableWidgetItem(self.err_msg))
			self.err_msg = None
		else:
			# 第一个参数是row, 第二参数是column
			self.res_table.setItem(0, column, QtGui.QTableWidgetItem(self.city))
			self.res_table.setItem(1, column, QtGui.QTableWidgetItem(self.date))
			self.res_table.setItem(2, column, QtGui.QTableWidgetItem(self.high))
			self.res_table.setItem(3, column, QtGui.QTableWidgetItem(self.low))
			self.res_table.setItem(4, column, QtGui.QTableWidgetItem(self.type))


def main():
	app = QtGui.QApplication(sys.argv)
	app_icon = QtGui.QIcon()
	app_icon.addFile('icons/web.png', QtCore.QSize(32,32))
	app.setWindowIcon(app_icon)
	icon = weather_predictor()
	icon.show()
	sys.exit(app.exec_())

if __name__ == '__main__':
	main()