import pymysql
import execjs
import os
import re
from datetime import datetime
from pymysql import escape_string
import sys

HOST = '172.18.51.14'
PORT = 3306
USER = 'spider'
PASSWD = 'Gengmei123'
DB = 'spider'
JS_FILE_PATH = '/srv/apps/crawler/crawler_sys/site_crawler/zhihu.js'


class RefreshContent(object):
    
    def __init__(self):
        '''
            初始化数据库，调整js规则
        '''
        self.conn = pymysql.connect(host=HOST, port=PORT, user=USER,
                                    passwd=PASSWD,
                                    db=DB, charset='utf8')
        self.cur = self.conn.cursor()
        
        os.environ["EXECJS_RUNTIME"] = 'Node'
        try:
            with open('./zhihu.js', 'r', encoding='utf-8') as f:
                js = f.read()
        except:
            with open(JS_FILE_PATH, 'r', encoding='utf-8') as f:
                js = f.read()
        self.exec_js = execjs.compile(js)
    
    def refresh_content(self, table, pic_table, key_id):
        '''
            替换url，更新回答内容
        '''
        import re
        sql = """select {}, url, new_url from {}""".format(key_id, pic_table)
        self.cur.execute(sql)
        tuple = self.cur.fetchall()
        self.conn.commit()
        for i in range(len(tuple)):
            if tuple[i][2] == None:
                continue
            find_id = tuple[i][0]
            temp = str(tuple[i][1])
            temp1 = temp.replace("?", "#")
            sql = """select new_content from {} where {} = '{}' """.format(table, key_id, find_id)
            self.cur.execute(sql)
            tuples = self.cur.fetchall()
            # tuples = str(tuples)
            content = tuples[0][0]
            pattern = r'%s(.+?)%s' % ("<noscript><img src=\"" + temp1, "</figure>")
            temp_tuples = content.replace("?", "#")
            new_content = re.sub(pattern, "<noscript><img src=\"" + str(tuple[i][2]) + "\"></noscript></figure>",
                                 temp_tuples)
            new_content = r'%s' % (new_content)
            new_content = escape_string(new_content)
            sql = """update {} set new_content = '{}' WHERE {} = '{}' """.format(table, new_content, key_id, find_id)
            self.cur.execute(sql)
            self.conn.commit()
    
    def answer_refresh_content(self):
        '''
            替换url，更新回答内容
        '''
        sql = """select answer_id, url, new_url from zhihu_answer_picture_url"""
        self.cur.execute(sql)
        tuple = self.cur.fetchall()
        self.conn.commit()
        for i in range(len(tuple)):
            if tuple[i][2] == None:
                continue
            find_id = tuple[i][0]
            temp = str(tuple[i][1])
            temp1 = temp.replace("?", "#")
            sql = """select new_content from zhihu_answer where zhihu_answer.answer_id = '{}' """.format(find_id)
            self.cur.execute(sql)
            tuples = self.cur.fetchall()
            # tuples = str(tuples)
            content = tuples[0][0]
            pattern = r'%s(.+?)%s' % ("<noscript><img src=\"" + temp1, "</figure>")
            temp_tuples = content.replace("?", "#")
            new_content = re.sub(pattern, "<noscript><img src=\"" + str(tuple[i][2]) + "\"></noscript></figure>",
                                 temp_tuples)
            new_content = r'%s' % (new_content)
            new_content = escape_string(new_content)
            sql = """update zhihu_answer set new_content = '{}' WHERE answer_id = '{}' """.format(new_content,
                                                                                                  tuple[i][0])
            self.cur.execute(sql)
            self.conn.commit()
    
    def article_refresh_content(self):
        '''
            替换url，更新文章内容
        '''
        sql = """select article_id, url, new_url from zhihu_article_picture_url"""
        self.cur.execute(sql)
        tuple = self.cur.fetchall()
        self.conn.commit()
        for i in range(len(tuple)):
            find_id = tuple[i][0]
            temp = str(tuple[i][1])
            temp1 = temp.replace("?", "#")
            sql = """select new_content from zhihu_article where zhihu_article.article_id = '{}' """.format(find_id)
            self.cur.execute(sql)
            tuples = self.cur.fetchall()
            # tuples = str(tuples)
            content = tuples[0][0]
            pattern = r'%s(.+?)%s' % ("<noscript><img src=\"" + temp1, "</figure>")
            temp_tuples = content.replace("?", "#")
            new_content = re.sub(pattern, "<noscript><img src=\"" + str(tuple[i][2]) + "\"></noscript></figure>",
                                 temp_tuples)
            new_content = r'%s' % (new_content)
            new_content = escape_string(new_content)
            sql = """update zhihu_article set new_content = '{}' WHERE article_id = '{}' """.format(new_content,
                                                                                                    tuple[i][0])
            self.cur.execute(sql)
            self.conn.commit()


if __name__ == '__main__':
    # print('参数个数为:', len(sys.argv), '个参数。')
    # print('参数列表:', type(sys.argv[0]), sys.argv[0], type(sys.argv[1]), sys.argv[1])
    mark = int(sys.argv[1])
    print(datetime.now())
    refresh = RefreshContent()
    if mark == 0:
        refresh.refresh_content('zhihu_answer', 'zhihu_answer_picture_url', 'answer_id')
    elif mark == 1:
        refresh.refresh_content('zhihu_article', 'zhihu_article_picture_url', 'article_id')
    elif mark == 2:
        refresh.refresh_content('zhihu_thought', 'zhihu_thought_picture_url', 'thought_id')
    print(datetime.now())
