def city2code(city_name):
	with open('city_code.json', 'r') as f:
		data = json.load(f)
	f.close()
	try:
		return data[city_name]
	except:
		sys.exit('no city {} information'.format(city_name))

city = input("city:")
city_code = city2code(city)
url = "http://www.weather.com.cn/weather/{}.shtml".format(city_code)
res = requests.get(url)
res.encoding = 'utf-8'
bs = BeautifulSoup(res.text,"html.parser")
date = bs.select("li.sky > h1")
desc = bs.select("li.sky > p.wea")
temp = bs.select("li.sky > p.tem")
dir = bs.select("li > p.win > em")
level = bs.select("li > p.win > i")
result = []
for i in range(date.__len__()):
    date1 = date[i].text
    desc1 = desc[i].text
    temp1 = temp[i].stripped_strings
    temp1 = "".join(temp1)
    dir1 = dir[i]
    dir2 = dir1.select("span")
    if len(dir2) == 1:
        direction = "无持续风向"
    else:
        direction = dir2[0].get("title")+"-"+dir2[1].get("title")
    level1 = level[i].text
    result.append([city,date1,desc1,temp1,direction,level1])

conn = sqlite3.connect('weather.db')
cursor = conn.cursor()
try:
    cursor.execute("delete from week_weather where city='{}'".format(city))
    print("delete old data")
except:
    cursor.execute('''create table week_weather (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        city char(100),
        date char(100),
        desc char(100),
        temp char(100),
        direction char(100),
        level char(100)
        );''')
    print("create new table")
for result1 in result:
    print(result1)
    insertsql = "insert into week_weather (city,date,desc,temp,direction,level) VALUES ('%s','%s','%s','%s','%s','%s')"
    cursor.execute(insertsql % (result1[0],result1[1],result1[2],result1[3],result1[4],result1[5]))
 
conn.commit()
cursor.close()
conn.close()

dic1 = ['河北省', '山西省', '辽宁省','吉林省', '黑龙江省', '江苏省', '浙江省', '安徽省', '福建省', '江西省', '山东省', '河南省', '广东省', '湖南省', '湖北省', '海南省', '四川省', '贵州省', '云南省', '陕西省', '甘肃省', '青海省', '台湾省', '内蒙古自治区', '广西壮族自治区', '西藏自治区', '宁夏回族自治区', '新疆维吾尔自治区']
dic2 = ['石家庄', '太原', '沈阳', '长春', '哈尔滨', '南京', '杭州', '合肥', '福州', '南昌', '济南', '郑州', '广州', '长沙', '武汉', '海口', '成都', '贵阳', '昆明', '西安', '兰州', '西宁', '台北', '呼和浩特', '南宁', '拉萨', '银川', '乌鲁木齐']
dic3 = ['53698', '53772', '54342', '54161', '50953', '58238', '58457', '58321', '58847', '58606', '54823', '57083', '59287', '57687', '57494', '59758', '56294', '57816', '56778', '57036', '52889', '52866', '71294', '53463', '59431', '55591', '53614','51463']
dic4 = ['shijiazhuang', 'taiyuan', 'shenyang', 'changchun', 'haerbin', 'nanjing', 'hangzhou', 'hefei', 'fuzhou', 'nanchang', 'jinan', 'zhengzhou', 'guangzhou', 'changsha', 'wuhan', 'haikou', 'chengdu', 'guiyang', 'kunming', 'xian', 'lanzhou', 'xining', 'taibei', 'huhehaote', 'nanning', 'lasa', 'yinchuan', 'wulumuqi']

result = []
for i in range(dic1.__len__()):
    url = 'http://tianqi.eastday.com/{}/{}.html'.format(dic4[i],dic3[i])
    res = requests.get(url)
    bs = BeautifulSoup(res.text,"html.parser")
    data = bs.select("div.tempBox > span")
    temp = int(data[0].text)
    result.append([dic1[i],dic2[i],temp])

conn = sqlite3.connect('weather.db')
cursor = conn.cursor()
try:
    cursor.execute("drop table live_weather;")
    print("delete old data")
except:
    pass
cursor.execute('''create table live_weather (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    province char(100),
    city char(100),
    temp int
    );''')
print("create new table")
for result1 in result:
    print(result1)
    insertsql = "insert into live_weather (province,city,temp) VALUES ('%s','%s','%d')"
    cursor.execute(insertsql % (result1[0],result1[1],result1[2]))

conn = sqlite3.connect('weather.db')
c = conn.cursor()
print("opened database successfully")
try:
    cursor = c.execute("SELECT * from live_weather")
    print("data selected successfully")
except:
    sys.exit("No information")
city = []
temp = []
for row in cursor:
    city.append(row[2])
    temp.append(row[3])

def geo_base() -> Geo:
    c = (
        Geo()
        .add_schema(maptype="china")
        .add("省会温度", [list(z) for z in zip(city, temp)])
        .set_series_opts(label_opts=opts.LabelOpts(is_show=False))
        .set_global_opts(
            visualmap_opts=opts.VisualMapOpts(max_=30),
            title_opts=opts.TitleOpts(title="省会城市实时温度", subtitle="made by Ringfa1l"),
        )
    )
    return c
geo_base().render("result/live_temp.html")

year = str(int(time.strftime("%Y", time.localtime()))-1)
mon = time.strftime("%m", time.localtime())
day = time.strftime("%d", time.localtime())

result = []
for i in range(dic1.__len__()):
    for j in range(1,13):
        monn = int(mon)+j
        if monn > 12:
            monn = monn-12
            yearr = int(year) + 1
        else:
            yearr = int(year)
        url = 'http://tianqi.eastday.com/{}_history/{}_{}{:0>2d}.html'.format(dic4[i],dic3[i],yearr,monn)
        #print(url)
        res = requests.get(url)
        bs = BeautifulSoup(res.text,"html.parser")
        if bs.find_all(class_="item-value") != []:
            data = bs.find_all(class_="item-value")
            high = re.findall(r"\d+",data[0].text)
            low = re.findall(r"\d+",data[1].text)
            #print(high,low)
            try:
                result.append([dic2[i],yearr,monn,int(high[0]),int(low[0])])
            except:
                print(dic2[i]+'-'+str(yearr)+'-'+str(monn)+"data lost")
        else:
            data = bs.select("ul.history2_left > li")
            high = re.findall(r"\d+",data[0].text)
            low = re.findall(r"\d+",data[1].text)
            #print(high,low)
            try:
                result.append([dic2[i],yearr,monn,int(high[0]),int(low[0])])
            except:
                print(dic2[i]+'-'+str(yearr)+'-'+str(monn)+"data lost")

conn = sqlite3.connect('weather.db')
cursor = conn.cursor()
try:
    cursor.execute("drop table his_weather;")
    print("delete old data")
except:
    pass
cursor.execute('''create table his_weather (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    city char(100),
    mon char(100),
    high int,
    low int
    );''')
print("create new table")
for result1 in result:
    print(result1)
    insertsql = "insert into his_weather (city,mon,high,low) VALUES ('%s','%s','%d','%d')"
    cursor.execute(insertsql % (result1[0],str(result1[1])+'年'+str(result1[2])+'月',result1[3],result1[4]))

city = input("请输入一个省会城市:")
conn = sqlite3.connect('weather.db')
c = conn.cursor()
print("opened database successfully")
try:
    cursor = c.execute("SELECT * from his_weather where city='{}'".format(city))
    print("data selected successfully")
except:
    sys.exit("No information about {}".format(city))
time = []
high = []
low = []
for row in cursor:
    time.append(row[2])
    high.append(row[3])
    low.append(row[4])

def line_base() -> Line:
    c = (
        Line()
        .add_xaxis(time)
        .add_yaxis("月平均最高温度", high)
        .add_yaxis("月平均最低温度", low)
        .set_global_opts(title_opts=opts.TitleOpts(title="{}历史气温比较".format(city), subtitle="made by Ringfa1l"))
    )
    return c
line_base().render("result/{}_his_compare.html".format(city))