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

import sys, os
import requests
from bs4 import BeautifulSoup
from workflow import Workflow3

reload(sys)
sys.setdefaultencoding('utf-8')

def save_img(url):
    """下载图片"""
    file_name = url.split('/')[-1]
    try:
        r = requests.get(url)
    except:
        print "远程连接错误"
    try:
        filepath = "picture/" + file_name
        with open(filepath, 'wb') as f:
            f.write(r.content)
    except:
        print "文件保存错误"
        
        
def main(wf):
    """输入日期"""
    base_url = "https://www.lssdjt.com"
    if not os.path.exists("picture"):
         os.makedirs("picture")
    if not wf.args:
        url = base_url
    else:
        try:
            today = str(wf.args[0])
            month = today.split('/')[0]
            day = today.split('/')[1]
            url = "https://www.lssdjt.com/{}/{}/".format(month, day)
        except:
           print "输入参数错误"
        
    resp = requests.request("GET", url)
    resp.raise_for_status()
    soup = BeautifulSoup(resp.content, "html.parser")
    today_in_history = soup.find_all("li", attrs={"class": "gong"})
    """加载资源"""
    try:
        threads = []
        for li in today_in_history:
            h_url = base_url + li.a.attrs['href']
            if li.a.has_attr('rel'):
                pic_url = li.a.attrs['rel'][0]
                pic_name = pic_url.split('/')[-1]
                h_jpg = "picture/" + pic_name
                if not os.path.exists(h_jpg):
                    t = threading.Thread(target=save_img,args={pic_url})
                    threads.append(t)
            else:
                h_jpg = 'default.png'
            h_date = li.a.em.get_text("", strip=True)
            h_title = li.a.i.get_text("", strip=True)
            wf.add_item(title=h_title, subtitle=h_date, arg=h_url, valid=True, icon=h_jpg)
        for i in threads:
            i.start()
        for i in threads:
            i.join()
    except:
         print "资源加载失败"
         
    wf.send_feedback()
    
if __name__ == '__main__':
    wf = Workflow3()
    sys.exit(wf.run(main))