comment.py

# -*- coding: utf-8 -*-
import requests

from util import doubanutil
from util import tools
from verifycode import wordrecognition


def comment_topic(topic_url, comment_dict):
    # 在一个帖子下发表回复

    r = requests.post(topic_url, cookies=doubanutil.get_cookies(),
                      data=comment_dict)
    doubanutil.logger.info("in func comment_topic(), " +
                           str(comment_dict) + ", status_code: " + str(r.status_code))
    return r


def make_comment_dict(topic_url, rv_comment):
    # 组装回帖的参数

    pic_url, pic_id = doubanutil.get_verify_code_pic(topic_url)
    verify_code = ""
    if len(pic_url):
        pic_path = tools.save_pic_to_disk(pic_url)
        verify_code = wordrecognition.get_word_in_pic(pic_path)
    comment_dict = {
        "ck": doubanutil.get_form_ck_from_cookie(),
        "rv_comment": rv_comment,
        "start": 0,
        "captcha-solution": verify_code,
        "captcha-id": pic_id,
        "submit_btn": "加上去"
    }
    return comment_dict