# python3

# author lizm

# datetime 2018-02-01 10:00:00

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

'''

    数据起始日期：2015-05-08

    数据库：mysql

'''

import requests

from bs4 import BeautifulSoup

import json

import pymysql

import datetime

import time

import sys

import logging

from selenium import webdriver

from sqlalchemy import Column,Integer, String,DateTime,create_engine

from sqlalchemy.orm import sessionmaker

from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy import and_,func

import configparser

import math

logger = logging.getLogger()

#set loghandler

file = logging.FileHandler(sys.path[0]+"\py_zgjs_log"+time.strftime("%Y%m%d")+".log")

logger.addHandler(file)

#set formater

formatter = logging.Formatter("%(asctime)s %(levelname)s %(message)s")

file.setFormatter(formatter)

#set log level

logger.setLevel(logging.NOTSET)

# 创建对象的基类:

Base = declarative_base()

class Yztzzqktjb(Base):

    # 表名

    __tablename__ = 'py_zgjs_yztzzqktjb'

    # 表结构

    id = Column(Integer,primary_key=True,autoincrement=True)

    mc = Column(String(200),nullable=False)

    begindate = Column(String(45),nullable=False)

    enddate = Column(String(45), nullable=False)

    sjmc = Column(String(200))

    ssjmc = Column(String(200))

    sl = Column(String(45))

    create_time = Column(DateTime,nullable=False)

    update_time = Column(DateTime,nullable=False)

    def __init__(self,mc,begindate,enddate,sjmc,ssjmc,sl,create_time,update_time):

        self.mc = mc

        self.begindate = begindate

        self.enddate = enddate

        self.sjmc = sjmc

        self.ssjmc = ssjmc

        self.sl = sl

        self.create_time = create_time

        self.update_time = update_time

class ZgjsEntry(object):

    def __init__(self, v1, v2,v3,v4,v5,v6):

        self.v1 = v1

        self.v2 = v2

        self.v3 = v3

        self.v4 = v4

        self.v5 = v5

        self.v6 = v6

    def __get__(self, instance, cls):

        if instance is None:

            return self

        else:

            return instance.__dict__[self.name]

    def __set__(self, instance, value):

        instance.__dict__[self.name] = value

    def __delete__(self, instance):

        del instance.__dict__[self.name]

def dbconfig():

    #生成config对象

    cfg = configparser.ConfigParser()

    #用config对象读取配置文件

    path_ = sys.path[0]

    cfg.read(path_+"\dbconfig.ini")

    ip = cfg.get("dbserver", "ip")

    port = cfg.get("dbserver", "port")

    user = cfg.get("dbserver", "user")

    password = cfg.get("dbserver", "password")

    dbname = cfg.get("dbserver", "dbname")

    endtime = cfg.get("dbtime", "endtime")

    initdate = cfg.get("dbtime", "mzkbinitdate")

    interval = cfg.get("dbtime", "interval")

    return (ip,port,user,password,dbname,endtime,initdate,interval)

def savrData(tableName,zgjsList):

    msgcode = 0

    message = '数据保存成功'

    try:

        dbcfg = dbconfig()

        # 初始化数据库连接,

        # 传入参数：数据库类型+连接库+用户名+密码+主机，字符编码，是否打印建表细节

        engine = create_engine('mysql+mysqlconnector://'+dbcfg[2]+':'+dbcfg[3]+'@'+dbcfg[0]+':'+dbcfg[1]+'/'+dbcfg[4],encoding='utf-8')

        # 创建DBSession类型:

        DBSession = sessionmaker(bind=engine)

        session = DBSession()

        try:

            # 增操作

            items = []

            if tableName == 'Yztzzqktjb':

                if len(zgjsList)>0:

                    for i in range(0,len(zgjsList)):

                        results = session.query(Yztzzqktjb).filter(and_(Yztzzqktjb.mc == zgjsList[i].v1,Yztzzqktjb.begindate == zgjsList[i].v2,Yztzzqktjb.enddate==zgjsList[i].v3,Yztzzqktjb.sjmc==zgjsList[i].v4,Yztzzqktjb.ssjmc==zgjsList[i].v5)).all()

                        if len(results) > 0:

                            session.query(Yztzzqktjb).filter(and_(Yztzzqktjb.mc == zgjsList[i].v1,Yztzzqktjb.begindate == zgjsList[i].v2,Yztzzqktjb.enddate==zgjsList[i].v3,Yztzzqktjb.sjmc==zgjsList[i].v4,Yztzzqktjb.ssjmc==zgjsList[i].v5)).update({Yztzzqktjb.sl: zgjsList[i].v6,Yztzzqktjb.update_time:time.strftime('%Y-%m-%d %H:%M:%S')}, synchronize_session=False)

                        else:

                            item = Yztzzqktjb(mc=zgjsList[i].v1,begindate=zgjsList[i].v2,enddate=zgjsList[i].v3,sjmc=zgjsList[i].v4,ssjmc=zgjsList[i].v5,sl=zgjsList[i].v6,create_time=time.strftime('%Y-%m-%d %H:%M:%S'),update_time=time.strftime('%Y-%m-%d %H:%M:%S'))

                            items.append(item)

            else:

                pass

            #print("len(items)>>>>>%s" %len(items))

            if len(items) > 0:

                for i in range(0,len(items)):

                    session.add(items[i])

            #提交数据

            session.commit()

        except Exception as e:

            msgcode = 1

            message = '数据保存失败' + str(e)

            session.rollback()

        finally:

            #关闭

            session.close()

    except Exception as e:

        msgcode = 1

        message = '数据库连接失败'+str(e)

    logger.info(message)

    print(message)

    return msgcode

def getData(jsDate, channelIdStr,tableName):

    zgjsList = []

    dateStr = jsDate[0:4]+'.'+jsDate[5:7]+'.'+jsDate[8:10]

    # 查询按钮跳转url：

    # http://www.******.cn/cms-search/view.action?action=china

    url = "http://www.******.cn/cms-search/view.action?action=china"

    headerDict = {'Host': 'www.*******.cn',

                  'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.31 Safari/537.36',

                  'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8',

                  'Accept-Language': 'zh-CN,zh;q=0.8',

                  'Accept-Encoding': 'gzip, deflate',

                  'Referer': 'http://www.******.cn/cms-search/view.action?action=china',

                  'Connection': 'keep-alive'}

    data = {'dateType': '', 'dateStr': dateStr,

            'channelIdStr': channelIdStr}

    # psot 传递参数

    res = requests.post(url, data=data, headers=headerDict)

    # 获取跳转后的页面源码

    soup = BeautifulSoup(res.content, "html.parser")

    #获取周报的起始日期

    SettlementTitle = soup.find('div',class_='SettlementTitle')

    if SettlementTitle is None:

        return zgjsList

    h2 = SettlementTitle.find('h2').text

    if h2 == '搜索结果':

        return zgjsList

    weekdate = h2.strip().split('（')[1].split('）')[0]

    begindate = weekdate.split('-')[0].replace('.','-')

    enddate = weekdate.split('-')[1].replace('.','-')

    settlementList = soup.find(id='settlementList')

    # print(settlementList)

    if settlementList is None:

        return zgjsList

    if settlementList.find('table') is None:

        return zgjsList

    table_ = settlementList.find('table')

    tr_list = table_.find('table').find_all('tr')

    # 上级名称

    sjmc_1 = ''

    sjmc_2 = ''

    sjmc_3 = ''

    sjmc_4 = ''

    sjmc_5 = ''

    sjmc_6 = ''

    # 上上级名称

    ssjmc_1 = ''

    for n in range(1,len(tr_list)):

        td_list = tr_list[n].find_all('td')

        if tableName == 'Yztzzqktjb':

            if n == 1:

                sjmc_1 = td_list[0].get_text().replace('一、','').strip()

            if n == 4:

                sjmc_2 = td_list[0].get_text().replace('二、','').strip()

                ssjmc_1 = td_list[0].get_text().replace('二、','').strip()

            if n == 5:

                sjmc_3 = td_list[0].get_text().replace('1、','').strip()

            if n == 9:

                sjmc_4 = td_list[0].get_text().replace('2、','').strip()

            if n == 13:

                sjmc_5 = td_list[0].get_text().replace('三、','').strip()

            if n == 17:

                sjmc_6 = td_list[0].get_text().replace('四、','').strip()

        if tableName == 'Yztzzqktjb':

            if n in (6,10,14,18):

                continue

        zgjs = ZgjsEntry('','','','','','')

        zgjs.v2 = begindate

        zgjs.v3 = enddate

        if tableName == 'Yztzzqktjb':

            # 上级名称

            if n in (2,3):

                zgjs.v4 = sjmc_1

            if n in (5,9):

                zgjs.v4 = sjmc_2

            if n in (7,8):

                zgjs.v4 = sjmc_3

                zgjs.v5 = ssjmc_1

            if n in (11,12):

                zgjs.v4 = sjmc_4

                zgjs.v5 = ssjmc_1

            if n in (15,16):

                zgjs.v4 = sjmc_5

            if n in (19,20):

                zgjs.v4 = sjmc_6

        for i in range(0,len(td_list)):

            if i == 0:

                zgjs.v1 =td_list[i].get_text().replace('一、','').replace('二、','').replace('三、','').replace('四、','').replace('1、','').replace('2、','').strip()

            if i == 1:

                zgjs.v6 =td_list[i].get_text().strip().replace(',','')

        if zgjs is not None:

            zgjsList.append(zgjs)

    return zgjsList

# 获取开始日期：

def getBeginDate(bgdate,tableName):

    r_date = bgdate

    try:

        dbcfg = dbconfig()

        # 初始化数据库连接,

        # 传入参数：数据库类型+连接库+用户名+密码+主机，字符编码，是否打印建表细节

        engine = create_engine('mysql+mysqlconnector://'+dbcfg[2]+':'+dbcfg[3]+'@'+dbcfg[0]+':'+dbcfg[1]+'/'+dbcfg[4],encoding='utf-8')

        # 创建DBSession类型:

        DBSession = sessionmaker(bind=engine)

        session = DBSession()

        try:

            if tableName == 'Yztzzqktjb':

                results = session.query(func.max(Yztzzqktjb.enddate)).all()

                if len(results) != 0:

                    r_date = results[0]

            else:

                pass

        except Exception as e:

            print('获取开始日期，查询异常；%s'%str(e))

            logger('获取开始日期，查询异常；%s'%str(e))

            session.rollback()

        finally:

            #关闭

            session.close()

    except Exception as e:

        print('获取开始日期，数据库连接失败；%s'%str(e))

        logger('获取开始日期，数据库连接失败；%s'%str(e))

    if r_date[0] is None:

        r_date = bgdate

    else:

        begin = time.strptime(r_date[0], "%Y-%m-%d")

        y,m,d = begin[0:3]

        r_date =  datetime.date(y,m,d) + datetime.timedelta(days=7)

        r_date = r_date .strftime('%Y-%m-%d')

    return r_date

def isCheckData(date_):

    r_code = 0

    try:

        dbcfg = dbconfig()

        # 初始化数据库连接,

        # 传入参数：数据库类型+连接库+用户名+密码+主机，字符编码，是否打印建表细节

        engine = create_engine('mysql+mysqlconnector://'+dbcfg[2]+':'+dbcfg[3]+'@'+dbcfg[0]+':'+dbcfg[1]+'/'+dbcfg[4],encoding='utf-8')

        # 创建DBSession类型:

        DBSession = sessionmaker(bind=engine)

        session = DBSession()

        try:

            #

            results = session.query(Yztzzqktjb).filter(and_(func.datediff(Yztzzqktjb.enddate,date_)<6,func.datediff(Yztzzqktjb.enddate,date_)>-2)).all()

            if len(results) == 0:

                r_code = 1

            else:

                r_code = 0

        except Exception as e:

            r_code = 1

            print('判断是否有数据异常；%s'%str(e))

            logger('判断是否有数据异常；%s'%str(e))

            session.rollback()

        finally:

            #关闭

            session.close()

    except Exception as e:

        r_code = 1

        print('判断是否有数据，数据库连接异常；%s'%str(e))

        logger('判断是否有数据，数据库连接异常；%s'%str(e))

    return r_code

# 执行入口

def main(initdate_):

    req_list = [

    {'report':'6ac54ce22db4474abc234d6edbe53ae7','table':'Yztzzqktjb'}

    ]

    for req in req_list:

        #字符转日期

        begin = time.strptime(getBeginDate(initdate_,req['table']), "%Y-%m-%d")

        y,m,d = begin[0:3]

        #日期格式：2018-01-18

        begin = datetime.date(y,m,d)

        #获取当前日期

        end = datetime.date.today()

        if (end- begin).days < 0:

            pass

        else:

            for i in range(math.ceil((end - begin).days/7)+1):

                list_szzj = []

                # 日期转字符

                date_ = (begin+datetime.timedelta(days=i*7)).strftime('%Y-%m-%d')

                list_mzkb = getData(date_,req['report'],req['table'])

                if len(list_mzkb):

                    savrData(req['table'],list_mzkb)

                else:

                    pass

                time.sleep(0.5)

                if i % 350 == 0:

                    time.sleep(15)

if __name__ == '__main__':

    vrg_date = ''

    dbcfg = dbconfig()

    vrg_endtime = dbcfg[5][0:2]+":"+dbcfg[5][2:4]+":"+dbcfg[5][4:6]

    var_initdate = dbcfg[6][0:4]+"-"+dbcfg[6][4:6]+"-"+dbcfg[6][6:8]

    var_interval = int(dbcfg[7])

    if len(vrg_date) ==8:

        vrg_date = str(vrg_date[0:4]) + "-" + str(vrg_date[4:6]) + "-" + str(vrg_date[6:8])

        end_time = time.strptime(vrg_endtime, "%H:%M:%S")

        y,m,d = end_time[3:6]

        end_time = datetime.time(y,m,d)

        # 循环采集

        while True:

            now_time = time.strftime("%H%M%S")

            main(var_initdate)

            if isCheckData(vrg_date,) == 0:

                logger.info("采集数据结束")

                print("采集数据结束")

                break

            # 时间到停止采集

            if int(end_time.strftime('%H%M%S')) - int(now_time) <= 0:

                logger.info("采集数据结束")

                print("采集数据结束")

                break

            # 间隔执行时间

            logger.info("**********************（%s）:没有采集到数据，任务继续执行**********************" %vrg_date)

            print("********************（%s）:没有采集到数据，任务继续执行**********************" %vrg_date)

            time.sleep(var_interval)

        else:

            logger.info("日期参数格式不正确，请用格式：20180205")

            print("日期参数格式不正确，请用格式：20180205")