779 lines
28 KiB
Python
779 lines
28 KiB
Python
# -*- coding: utf-8 -*-
|
||
# 自动带图评价、追评、服务评价，需电脑端CK
|
||
# @Time : 2022/11/4
|
||
# @Author : @qiu-lzsnmb and @Dimlitter @Dylan
|
||
# @File : auto_comment.py
|
||
# 多账号评价，异常处理
|
||
# 2023/3/28 修复乱码
|
||
# 2023/4/19 当存在OPENAI_API_KEY环境变量时，启用AI评价；网络支持：1.环境变量OPENAI_API_BASE_URL反向代理、2.ProxyUrl代理、3.环境支持直连；
|
||
'''
|
||
new Env('自动评价');
|
||
8 8 2 1 * https://raw.githubusercontent.com/6dylan6/auto_comment/main/jd_comment.py
|
||
'''
|
||
import argparse
|
||
import copy
|
||
import logging
|
||
import os
|
||
import random
|
||
import sys
|
||
import time, re
|
||
import urllib.parse
|
||
|
||
import toml
|
||
|
||
try:
|
||
import jieba # just for linting
|
||
import jieba.analyse
|
||
import requests
|
||
# import yaml
|
||
from lxml import etree
|
||
import zhon.hanzi
|
||
|
||
except:
|
||
print('解决依赖问题...稍等')
|
||
os.system('pip3 install lxml &> /dev/null')
|
||
os.system('pip3 install jieba &> /dev/null')
|
||
os.system('pip3 install zhon &> /dev/null')
|
||
os.system('pip3 install requests &> /dev/null')
|
||
os.system('pip3 install urllib3==1.25.11 &> /dev/null')
|
||
import jieba
|
||
import jieba.analyse
|
||
# import yaml
|
||
from lxml import etree
|
||
import requests
|
||
import urllib.parse
|
||
import jdspider
|
||
|
||
# constants
|
||
CONFIG_PATH = './config.yml'
|
||
USER_CONFIG_PATH = './config.user.yml'
|
||
ORDINARY_SLEEP_SEC = 10
|
||
SUNBW_SLEEP_SEC = 5
|
||
REVIEW_SLEEP_SEC = 10
|
||
SERVICE_RATING_SLEEP_SEC = 15
|
||
|
||
## logging with styles
|
||
## Reference: https://stackoverflow.com/a/384125/12002560
|
||
_COLORS = {
|
||
'black': 0,
|
||
'red': 1,
|
||
'green': 2,
|
||
'yellow': 3,
|
||
'blue': 4,
|
||
'magenta': 5,
|
||
'cyan': 6,
|
||
'white': 7
|
||
}
|
||
|
||
_RESET_SEQ = '\033[0m'
|
||
_COLOR_SEQ = '\033[1;%dm'
|
||
_BOLD_SEQ = '\033[1m'
|
||
_ITALIC_SEQ = '\033[3m'
|
||
_UNDERLINED_SEQ = '\033[4m'
|
||
|
||
_FORMATTER_COLORS = {
|
||
'DEBUG': _COLORS['blue'],
|
||
'INFO': _COLORS['green'],
|
||
'WARNING': _COLORS['yellow'],
|
||
'ERROR': _COLORS['red'],
|
||
'CRITICAL': _COLORS['red']
|
||
}
|
||
|
||
|
||
def format_style_seqs(msg, use_style=True):
|
||
if use_style:
|
||
msg = msg.replace('$RESET', _RESET_SEQ)
|
||
msg = msg.replace('$BOLD', _BOLD_SEQ)
|
||
msg = msg.replace('$ITALIC', _ITALIC_SEQ)
|
||
msg = msg.replace('$UNDERLINED', _UNDERLINED_SEQ)
|
||
else:
|
||
msg = msg.replace('$RESET', '')
|
||
msg = msg.replace('$BOLD', '')
|
||
msg = msg.replace('$ITALIC', '')
|
||
msg = msg.replace('$UNDERLINED', '')
|
||
|
||
|
||
class StyleFormatter(logging.Formatter):
|
||
def __init__(self, fmt=None, datefmt=None, use_style=True):
|
||
logging.Formatter.__init__(self, fmt, datefmt)
|
||
self.use_style = use_style
|
||
|
||
def format(self, record):
|
||
rcd = copy.copy(record)
|
||
levelname = rcd.levelname
|
||
if self.use_style and levelname in _FORMATTER_COLORS:
|
||
levelname_with_color = '%s%s%s' % (
|
||
_COLOR_SEQ % (30 + _FORMATTER_COLORS[levelname]),
|
||
levelname, _RESET_SEQ)
|
||
rcd.levelname = levelname_with_color
|
||
return logging.Formatter.format(self, rcd)
|
||
|
||
|
||
# 评价生成
|
||
def generation(pname, _class=0, _type=1, opts=None):
|
||
if "OPENAI_API_KEY" in os.environ:
|
||
return generation_ai(pname, opts)
|
||
opts = opts or {}
|
||
items = ['商品名']
|
||
items.clear()
|
||
items.append(pname)
|
||
opts['logger'].debug('Items: %s', items)
|
||
loop_times = len(items)
|
||
opts['logger'].debug('Total loop times: %d', loop_times)
|
||
for i, item in enumerate(items):
|
||
opts['logger'].debug('Loop: %d / %d', i + 1, loop_times)
|
||
opts['logger'].debug('Current item: %s', item)
|
||
spider = jdspider.JDSpider(item, ck)
|
||
opts['logger'].debug('Successfully created a JDSpider instance')
|
||
# 增加对增值服务的评价鉴别
|
||
if "赠品" in pname or "非实物" in pname or "京服无忧" in pname or "权益" in pname or "非卖品" in pname or "增值服务" in pname:
|
||
result = [
|
||
"赠品挺好的。",
|
||
"很贴心，能有这样免费赠送的赠品!",
|
||
"正好想着要不要多买一份增值服务，没想到还有这样的赠品。",
|
||
"赠品正合我意。",
|
||
"赠品很好，挺不错的。",
|
||
"本来买了产品以后还有些担心。但是看到赠品以后就放心了。",
|
||
"不论品质如何，至少说明店家对客的态度很好！",
|
||
"我很喜欢这些商品！",
|
||
"我对于商品的附加值很在乎，恰好这些赠品为这件商品提供了这样的的附加值，这令我很满意。"
|
||
"感觉现在的网购环境环境越来越好了，以前网购的时候还没有过么多贴心的赠品和增值服务",
|
||
"第一次用京东，被这种赠品和增值服物的良好态度感动到了。",
|
||
"赠品还行。"
|
||
]
|
||
else:
|
||
result = spider.getData(4, 3) # 这里可以自己改
|
||
opts['logger'].debug('Result: %s', result)
|
||
|
||
# class 0是评价 1是提取id
|
||
try:
|
||
name = jieba.analyse.textrank(pname, topK=5, allowPOS='n')[0]
|
||
opts['logger'].debug('Name: %s', name)
|
||
except Exception as e:
|
||
# opts['logger'].warning(
|
||
# 'jieba textrank analysis error: %s, name fallback to "宝贝"', e)
|
||
name = "宝贝"
|
||
if _class == 1:
|
||
opts['logger'].debug('_class is 1. Directly return name')
|
||
return name
|
||
else:
|
||
if _type == 1:
|
||
num = 6
|
||
elif _type == 0:
|
||
num = 4
|
||
num = min(num, len(result))
|
||
# use `.join()` to improve efficiency
|
||
comments = ''.join(random.sample(result, num))
|
||
opts['logger'].debug('_type: %d', _type)
|
||
opts['logger'].debug('num: %d', num)
|
||
opts['logger'].debug('Raw comments: %s', comments)
|
||
|
||
return 5, comments.replace("$", name)
|
||
|
||
|
||
# ChatGPT评价生成
|
||
def generation_ai(pname, _class=0, _type=1, opts=None):
|
||
# 当存在 OPENAI_API_BASE_URL 时，使用反向代理
|
||
api_base_url = os.environ.get("OPENAI_API_BASE_URL", "https://api.openai.com")
|
||
api_key = os.environ["OPENAI_API_KEY"]
|
||
prompt = f"{pname} 写一段此商品的评价，简短、口语化"
|
||
response = requests.post(
|
||
f"{api_base_url}/v1/chat/completions",
|
||
headers={
|
||
"Content-Type": "application/json",
|
||
"Authorization": f"Bearer {api_key}",
|
||
},
|
||
json={
|
||
"model": "gpt-3.5-turbo",
|
||
"messages": [{"role": "user", "content": prompt}],
|
||
"max_tokens": 1024,
|
||
}
|
||
)
|
||
response_text = response.json()
|
||
if "error" in response_text:
|
||
print("\nOpenAI API 调用错误：\n", response_text["error"]["message"])
|
||
exit()
|
||
else:
|
||
return 5, response_text["choices"][0]["message"]["content"].strip()
|
||
|
||
|
||
# 查询全部评价
|
||
def all_evaluate(opts=None):
|
||
try:
|
||
opts = opts or {}
|
||
N = {}
|
||
url = 'https://club.jd.com/myJdcomments/myJdcomment.action?'
|
||
opts['logger'].debug('URL: %s', url)
|
||
opts['logger'].debug('Fetching website data')
|
||
req = requests.get(url, headers=headers)
|
||
opts['logger'].debug(
|
||
'Successfully accepted the response with status code %d',
|
||
req.status_code)
|
||
if not req.ok:
|
||
opts['logger'].warning(
|
||
'Status code of the response is %d, not 200', req.status_code)
|
||
req_et = etree.HTML(req.text)
|
||
opts['logger'].debug('Successfully parsed an XML tree')
|
||
evaluate_data = req_et.xpath('//*[@id="main"]/div[2]/div[1]/div/ul/li')
|
||
loop_times = len(evaluate_data)
|
||
opts['logger'].debug('Total loop times: %d', loop_times)
|
||
for i, ev in enumerate(evaluate_data):
|
||
opts['logger'].debug('Loop: %d / %d', i + 1, loop_times)
|
||
na = ev.xpath('a/text()')[0]
|
||
opts['logger'].debug('na: %s', na)
|
||
# print(ev.xpath('b/text()')[0])
|
||
try:
|
||
num = ev.xpath('b/text()')[0]
|
||
opts['logger'].debug('num: %s', num)
|
||
except IndexError:
|
||
# opts['logger'].warning('Can\'t find num content in XPath, fallback to 0')
|
||
num = 0
|
||
N[na] = int(num)
|
||
return N
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
|
||
# 评价晒单
|
||
def sunbw(N, opts=None):
|
||
try:
|
||
opts = opts or {}
|
||
Order_data = []
|
||
req_et = []
|
||
loop_times = 2
|
||
opts['logger'].debug('Fetching website data')
|
||
opts['logger'].debug('Total loop times: %d', loop_times)
|
||
for i in range(loop_times):
|
||
url = (f'https://club.jd.com/myJdcomments/myJdcomment.action?sort=0&'
|
||
f'page={i + 1}')
|
||
opts['logger'].debug('URL: %s', url)
|
||
req = requests.get(url, headers=headers)
|
||
opts['logger'].debug(
|
||
'Successfully accepted the response with status code %d',
|
||
req.status_code)
|
||
if not req.ok:
|
||
opts['logger'].warning(
|
||
'Status code of the response is %d, not 200', req.status_code)
|
||
req_et.append(etree.HTML(req.text))
|
||
opts['logger'].debug('Successfully parsed an XML tree')
|
||
opts['logger'].debug('Fetching data from XML trees')
|
||
opts['logger'].debug('Total loop times: %d', loop_times)
|
||
for idx, i in enumerate(req_et):
|
||
opts['logger'].debug('Loop: %d / %d', idx + 1, loop_times)
|
||
opts['logger'].debug('Fetching order data in the default XPath')
|
||
elems = i.xpath(
|
||
'//*[@id="main"]/div[2]/div[2]/table/tbody')
|
||
opts['logger'].debug('Count of fetched order data: %d', len(elems))
|
||
Order_data.extend(elems)
|
||
# if len(Order_data) != N['待评价订单']:
|
||
# opts['logger'].debug(
|
||
# 'Count of fetched order data doesn\'t equal N["待评价订单"]')
|
||
# opts['logger'].debug('Clear the list Order_data')
|
||
# Order_data = []
|
||
# opts['logger'].debug('Total loop times: %d', loop_times)
|
||
# for idx, i in enumerate(req_et):
|
||
# opts['logger'].debug('Loop: %d / %d', idx + 1, loop_times)
|
||
# opts['logger'].debug('Fetching order data in another XPath')
|
||
# elems = i.xpath(
|
||
# '//*[@id="main"]/div[2]/div[2]/table')
|
||
# opts['logger'].debug('Count of fetched order data: %d', len(elems))
|
||
# Order_data.extend(elems)
|
||
|
||
opts['logger'].info(f"当前共有{N['待评价订单']}个评价。")
|
||
opts['logger'].debug('Commenting on items')
|
||
for i, Order in enumerate(reversed(Order_data)):
|
||
if i + 1 > 10:
|
||
opts['logger'].info(f'\n已评价10个订单，跳出')
|
||
break
|
||
try:
|
||
oid = Order.xpath('tr[@class="tr-th"]/td/span[3]/a/text()')[0]
|
||
opts['logger'].debug('oid: %s', oid)
|
||
oname_data = Order.xpath(
|
||
'tr[@class="tr-bd"]/td[1]/div[1]/div[2]/div/a/text()')
|
||
opts['logger'].debug('oname_data: %s', oname_data)
|
||
pid_data = Order.xpath(
|
||
'tr[@class="tr-bd"]/td[1]/div[1]/div[2]/div/a/@href')
|
||
opts['logger'].debug('pid_data: %s', pid_data)
|
||
except IndexError:
|
||
opts['logger'].warning(f"第{i + 1}个订单未查找到商品，跳过。")
|
||
continue
|
||
loop_times1 = min(len(oname_data), len(pid_data))
|
||
opts['logger'].debug('Commenting on orders')
|
||
opts['logger'].debug('Total loop times: %d', loop_times1)
|
||
idx = 0
|
||
for oname, pid in zip(oname_data, pid_data):
|
||
pid = re.findall('(?<=jd.com/)[(0-9)*?]+', pid)
|
||
if len(pid) == 0:
|
||
continue
|
||
pid = pid[0]
|
||
opts['logger'].info(f'\n开始第{i + 1}个订单: {oid}')
|
||
opts['logger'].debug('pid: %s', pid)
|
||
opts['logger'].debug('oid: %s', oid)
|
||
xing, Str = generation(oname, opts=opts)
|
||
opts['logger'].info(f'评价信息：{xing}星 ' + Str)
|
||
# 获取图片
|
||
url1 = (f'https://club.jd.com/discussion/getProductPageImageCommentList'
|
||
f'.action?productId={pid}')
|
||
opts['logger'].debug('Fetching images using the default URL')
|
||
opts['logger'].debug('URL: %s', url1)
|
||
req1 = requests.get(url1, headers=headers)
|
||
opts['logger'].debug(
|
||
'Successfully accepted the response with status code %d',
|
||
req1.status_code)
|
||
if not req1.ok:
|
||
opts['logger'].warning(
|
||
'Status code of the response is %d, not 200', req1.status_code)
|
||
imgdata = req1.json()
|
||
opts['logger'].debug('Image data: %s', imgdata)
|
||
if imgdata["imgComments"]["imgCommentCount"] > 10:
|
||
pnum = random.randint(2, int(imgdata["imgComments"]["imgCommentCount"] / 10) + 1)
|
||
opts['logger'].debug('Count of fetched image comments is 0')
|
||
opts['logger'].debug('Fetching images using another URL')
|
||
url1 = (f'https://club.jd.com/discussion/getProductPageImage'
|
||
f'CommentList.action?productId={pid}&page={pnum}')
|
||
opts['logger'].debug('URL: %s', url1)
|
||
time.sleep(1)
|
||
req1 = requests.get(url1, headers=headers)
|
||
opts['logger'].debug(
|
||
'Successfully accepted the response with status code %d',
|
||
req1.status_code)
|
||
if not req1.ok:
|
||
opts['logger'].warning(
|
||
'Status code of the response is %d, not 200',
|
||
req1.status_code)
|
||
imgdata2 = req1.json()
|
||
opts['logger'].debug('Image data: %s', imgdata2)
|
||
try:
|
||
imgurl = random.choice(imgdata["imgComments"]["imgList"])["imageUrl"]
|
||
if ('imgdata2' in dir()):
|
||
imgurl2 = random.choice(imgdata2["imgComments"]["imgList"])["imageUrl"]
|
||
else:
|
||
imgurl2 = ''
|
||
except Exception:
|
||
imgurl = ''
|
||
imgurl2 = ''
|
||
opts['logger'].debug('Image URL: %s', imgurl)
|
||
|
||
opts['logger'].info(f'图片：{imgurl + "," + imgurl2}')
|
||
# 提交晒单
|
||
opts['logger'].debug('Preparing for commenting')
|
||
url2 = "https://club.jd.com/myJdcomments/saveProductComment.action"
|
||
opts['logger'].debug('URL: %s', url2)
|
||
headers['Referer'] = ('https://club.jd.com/myJdcomments/orderVoucher.action')
|
||
headers['Origin'] = 'https://club.jd.com'
|
||
headers['Content-Type'] = 'application/x-www-form-urlencoded'
|
||
opts['logger'].debug('New header for this request: %s', headers)
|
||
data = {
|
||
'orderId': oid,
|
||
'productId': pid,
|
||
'score': str(xing), # 商品几星
|
||
'content': urllib.parse.quote(Str), # 评价内容
|
||
'imgs': imgurl + ',' + imgurl2,
|
||
'saveStatus': 2,
|
||
'anonymousFlag': 1
|
||
}
|
||
opts['logger'].debug('Data: %s', data)
|
||
if not opts.get('dry_run'):
|
||
opts['logger'].debug('Sending comment request')
|
||
pj2 = requests.post(url2, headers=headers, data=data)
|
||
if pj2.ok:
|
||
opts['logger'].info(f'提交成功！')
|
||
else:
|
||
opts['logger'].debug(
|
||
'Skipped sending comment request in dry run')
|
||
opts['logger'].debug('Sleep time (s): %.1f', ORDINARY_SLEEP_SEC)
|
||
time.sleep(ORDINARY_SLEEP_SEC)
|
||
idx += 1
|
||
N['待评价订单'] -= 1
|
||
return N
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
|
||
# 追评
|
||
def review(N, opts=None):
|
||
try:
|
||
opts = opts or {}
|
||
req_et = []
|
||
Order_data = []
|
||
loop_times = 2
|
||
opts['logger'].debug('Fetching website data')
|
||
opts['logger'].debug('Total loop times: %d', loop_times)
|
||
for i in range(loop_times):
|
||
opts['logger'].debug('Loop: %d / %d', i + 1, loop_times)
|
||
url = (f"https://club.jd.com/myJdcomments/myJdcomment.action?sort=3"
|
||
f"&page={i + 1}")
|
||
opts['logger'].debug('URL: %s', url)
|
||
req = requests.get(url, headers=headers)
|
||
opts['logger'].debug(
|
||
'Successfully accepted the response with status code %d',
|
||
req.status_code)
|
||
if not req.ok:
|
||
opts['logger'].warning(
|
||
'Status code of the response is %d, not 200', req.status_code)
|
||
req_et.append(etree.HTML(req.text))
|
||
opts['logger'].debug('Successfully parsed an XML tree')
|
||
opts['logger'].debug('Fetching data from XML trees')
|
||
opts['logger'].debug('Total loop times: %d', loop_times)
|
||
for idx, i in enumerate(req_et):
|
||
opts['logger'].debug('Loop: %d / %d', idx + 1, loop_times)
|
||
opts['logger'].debug('Fetching order data in the default XPath')
|
||
elems = i.xpath(
|
||
'//*[@id="main"]/div[2]/div[2]/table/tr[@class="tr-bd"]')
|
||
opts['logger'].debug('Count of fetched order data: %d', len(elems))
|
||
Order_data.extend(elems)
|
||
# if len(Order_data) != N['待追评']:
|
||
# opts['logger'].debug(
|
||
# 'Count of fetched order data doesn\'t equal N["待追评"]')
|
||
# # NOTE: Need them?
|
||
# # opts['logger'].debug('Clear the list Order_data')
|
||
# # Order_data = []
|
||
# opts['logger'].debug('Total loop times: %d', loop_times)
|
||
# for idx, i in enumerate(req_et):
|
||
# opts['logger'].debug('Loop: %d / %d', idx + 1, loop_times)
|
||
# opts['logger'].debug('Fetching order data in another XPath')
|
||
# elems = i.xpath(
|
||
# '//*[@id="main"]/div[2]/div[2]/table/tbody/tr[@class="tr-bd"]')
|
||
# opts['logger'].debug('Count of fetched order data: %d', len(elems))
|
||
# Order_data.extend(elems)
|
||
opts['logger'].info(f"当前共有{N['待追评']}个需要追评。")
|
||
opts['logger'].debug('Commenting on items')
|
||
for i, Order in enumerate(reversed(Order_data)):
|
||
if i + 1 > 10:
|
||
opts['logger'].info(f'\n已评价10个订单，跳出')
|
||
break
|
||
oname = Order.xpath('td[1]/div/div[2]/div/a/text()')[0]
|
||
_id = Order.xpath('td[3]/div/a/@href')[0]
|
||
opts['logger'].debug('_id: %s', _id)
|
||
url1 = ("https://club.jd.com/afterComments/"
|
||
"saveAfterCommentAndShowOrder.action")
|
||
opts['logger'].debug('URL: %s', url1)
|
||
pid, oid = _id.replace(
|
||
'http://club.jd.com/afterComments/productPublish.action?sku=',
|
||
"").split('&orderId=')
|
||
opts['logger'].debug('pid: %s', pid)
|
||
opts['logger'].debug('oid: %s', oid)
|
||
opts['logger'].info(f'\n开始第{i + 1}个订单: {oid}')
|
||
_, context = generation(oname, _type=0, opts=opts)
|
||
opts['logger'].info(f'追评内容：{context}')
|
||
data1 = {
|
||
'orderId': oid,
|
||
'productId': pid,
|
||
'content': urllib.parse.quote(context),
|
||
'anonymousFlag': 1,
|
||
'score': 5
|
||
}
|
||
opts['logger'].debug('Data: %s', data1)
|
||
if not opts.get('dry_run'):
|
||
opts['logger'].debug('Sending comment request')
|
||
req_url1 = requests.post(url1, headers=headers, data=data1)
|
||
if req_url1.ok:
|
||
opts['logger'].info(f'提交成功！')
|
||
else:
|
||
opts['logger'].debug('Skipped sending comment request in dry run')
|
||
opts['logger'].debug('Sleep time (s): %.1f', REVIEW_SLEEP_SEC)
|
||
time.sleep(REVIEW_SLEEP_SEC)
|
||
N['待追评'] -= 1
|
||
return N
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
|
||
# 服务评价
|
||
def Service_rating(N, opts=None):
|
||
try:
|
||
opts = opts or {}
|
||
Order_data = []
|
||
req_et = []
|
||
loop_times = 2
|
||
opts['logger'].debug('Fetching website data')
|
||
opts['logger'].debug('Total loop times: %d', loop_times)
|
||
for i in range(loop_times):
|
||
opts['logger'].debug('Loop: %d / %d', i + 1, loop_times)
|
||
url = (f"https://club.jd.com/myJdcomments/myJdcomment.action?sort=4"
|
||
f"&page={i + 1}")
|
||
opts['logger'].debug('URL: %s', url)
|
||
req = requests.get(url, headers=headers)
|
||
opts['logger'].debug(
|
||
'Successfully accepted the response with status code %d',
|
||
req.status_code)
|
||
if not req.ok:
|
||
opts['logger'].warning(
|
||
'Status code of the response is %d, not 200', req.status_code)
|
||
req_et.append(etree.HTML(req.text))
|
||
opts['logger'].debug('Successfully parsed an XML tree')
|
||
opts['logger'].debug('Fetching data from XML trees')
|
||
opts['logger'].debug('Total loop times: %d', loop_times)
|
||
for idx, i in enumerate(req_et):
|
||
opts['logger'].debug('Loop: %d / %d', idx + 1, loop_times)
|
||
opts['logger'].debug('Fetching order data in the default XPath')
|
||
elems = i.xpath(
|
||
'//*[@id="main"]/div[2]/div[2]/table/tbody/tr[@class="tr-th"]')
|
||
opts['logger'].debug('Count of fetched order data: %d', len(elems))
|
||
Order_data.extend(elems)
|
||
# if len(Order_data) != N['服务评价']:
|
||
# opts['logger'].debug(
|
||
# 'Count of fetched order data doesn\'t equal N["服务评价"]')
|
||
# opts['logger'].debug('Clear the list Order_data')
|
||
# Order_data = []
|
||
# opts['logger'].debug('Total loop times: %d', loop_times)
|
||
# for idx, i in enumerate(req_et):
|
||
# opts['logger'].debug('Loop: %d / %d', idx + 1, loop_times)
|
||
# opts['logger'].debug('Fetching order data in another XPath')
|
||
# elems = i.xpath(
|
||
# '//*[@id="main"]/div[2]/div[2]/table/tr[@class="tr-bd"]')
|
||
# opts['logger'].debug('Count of fetched order data: %d', len(elems))
|
||
# Order_data.extend(elems)
|
||
opts['logger'].info(f"当前共有{N['服务评价']}个需要服务评价。")
|
||
opts['logger'].debug('Commenting on items')
|
||
for i, Order in enumerate(reversed(Order_data)):
|
||
if i + 1 > 10:
|
||
opts['logger'].info(f'\n已评价10个订单，跳出')
|
||
break
|
||
# oname = Order.xpath('td[1]/div[1]/div[2]/div/a/text()')[0]
|
||
oid = Order.xpath('td[1]/span[3]/a/text()')[0]
|
||
opts['logger'].info(f'\n开始第{i + 1}个订单: {oid}')
|
||
opts['logger'].debug('oid: %s', oid)
|
||
url1 = (f'https://club.jd.com/myJdcomments/insertRestSurvey.action'
|
||
f'?voteid=145&ruleid={oid}')
|
||
opts['logger'].debug('URL: %s', url1)
|
||
data1 = {
|
||
'oid': oid,
|
||
'gid': '32',
|
||
'sid': '186194',
|
||
'stid': '0',
|
||
'tags': '',
|
||
'ro591': f'591A{random.randint(4, 5)}', # 商品符合度
|
||
'ro592': f'592A{random.randint(4, 5)}', # 店家服务态度
|
||
'ro593': f'593A{random.randint(4, 5)}', # 快递配送速度
|
||
'ro899': f'899A{random.randint(4, 5)}', # 快递员服务
|
||
'ro900': f'900A{random.randint(4, 5)}' # 快递员服务
|
||
}
|
||
opts['logger'].debug('Data: %s', data1)
|
||
if not opts.get('dry_run'):
|
||
opts['logger'].debug('Sending comment request')
|
||
pj1 = requests.post(url1, headers=headers, data=data1)
|
||
if pj1.ok:
|
||
opts['logger'].info(f'提交成功！')
|
||
else:
|
||
opts['logger'].debug('Skipped sending comment request in dry run')
|
||
# opts['logger'].info("\n " + pj1.text)
|
||
opts['logger'].debug('Sleep time (s): %.1f', SERVICE_RATING_SLEEP_SEC)
|
||
time.sleep(SERVICE_RATING_SLEEP_SEC)
|
||
N['服务评价'] -= 1
|
||
return N
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
|
||
def No(opts=None):
|
||
opts = opts or {}
|
||
opts['logger'].info('')
|
||
N = all_evaluate(opts)
|
||
s = '----'.join(['{} {}'.format(i, N[i]) for i in N])
|
||
opts['logger'].info(s)
|
||
opts['logger'].info('')
|
||
return N
|
||
|
||
|
||
def main(opts=None):
|
||
opts = opts or {}
|
||
# opts['logger'].info("开始京东自动评价！")
|
||
N = No(opts)
|
||
opts['logger'].debug('N value after executing No(): %s', N)
|
||
if not N:
|
||
opts['logger'].error('CK错误，请确认是否电脑版CK！')
|
||
return
|
||
if N['待评价订单'] != 0:
|
||
opts['logger'].info("1.开始评价晒单")
|
||
N = sunbw(N, opts)
|
||
opts['logger'].debug('N value after executing sunbw(): %s', N)
|
||
N = No(opts)
|
||
opts['logger'].debug('N value after executing No(): %s', N)
|
||
if N['待追评'] != 0:
|
||
opts['logger'].info("2.开始追评！")
|
||
N = review(N, opts)
|
||
opts['logger'].debug('N value after executing review(): %s', N)
|
||
N = No(opts)
|
||
opts['logger'].debug('N value after executing No(): %s', N)
|
||
if N['服务评价'] != 0:
|
||
opts['logger'].info('3.开始服务评价')
|
||
N = Service_rating(N, opts)
|
||
opts['logger'].debug('N value after executing Service_rating(): %s', N)
|
||
N = No(opts)
|
||
opts['logger'].debug('N value after executing No(): %s', N)
|
||
opts['logger'].info("该账号运行完成！")
|
||
|
||
|
||
def get_ck():
|
||
cks = []
|
||
jd_cooke = ""
|
||
if "PC_COOKIE" in os.environ:
|
||
jd_cooke = os.environ["PC_COOKIE"]
|
||
else:
|
||
# 获取当前脚本所在目录
|
||
current_directory = os.path.dirname(__file__)
|
||
# 构建配置文件路径
|
||
config_file_path = os.path.join(current_directory, 'jd_config.toml')
|
||
# 检查配置文件是否存在
|
||
if os.path.exists(config_file_path):
|
||
# 读取配置文件
|
||
with open(config_file_path, 'r') as file:
|
||
config_data = toml.load(file)
|
||
# 获取 cookie 下的 PC_COOKIE 变量
|
||
jd_cooke = config_data.get('cookie', {}).get('PC_COOKIE')
|
||
else:
|
||
print(f"Error: The file '{config_file_path}' was not found.")
|
||
if len(jd_cooke) > 200:
|
||
if '&' in jd_cooke:
|
||
cks = jd_cooke.split('&')
|
||
else:
|
||
cks.append(jd_cooke)
|
||
else:
|
||
logger.info("CK错误，请确认是否电脑版CK！")
|
||
sys.exit(1)
|
||
logger.info("已获取环境变量 CK")
|
||
return cks
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# parse arguments
|
||
parser = argparse.ArgumentParser()
|
||
parser.add_argument('--dry-run',
|
||
help='have a full run without comment submission',
|
||
action='store_true')
|
||
parser.add_argument('--log-level',
|
||
help='specify logging level (default: info)',
|
||
default='INFO')
|
||
parser.add_argument('-o', '--log-file', help='specify logging file')
|
||
args = parser.parse_args()
|
||
if args.log_level.upper() not in [
|
||
'DEBUG', 'WARN', 'INFO', 'ERROR', 'FATAL'
|
||
# NOTE: `WARN` is an alias of `WARNING`. `FATAL` is an alias of
|
||
# `CRITICAL`. Using these aliases is for developers' and users'
|
||
# convenience.
|
||
# NOTE: Now there is no logging on `CRITICAL` level.
|
||
]:
|
||
args.log_level = 'INFO'
|
||
else:
|
||
args.log_level = args.log_level.upper()
|
||
opts = {
|
||
'dry_run': args.dry_run,
|
||
'log_level': args.log_level
|
||
}
|
||
if "DEBUG" in os.environ and os.environ["DEBUG"] == 'true':
|
||
opts = {
|
||
'dry_run': args.dry_run,
|
||
'log_level': 'DEBUG'
|
||
}
|
||
if hasattr(args, 'log_file'):
|
||
opts['log_file'] = args.log_file
|
||
else:
|
||
opts['log_file'] = None
|
||
|
||
# logging on console
|
||
_logging_level = getattr(logging, opts['log_level'])
|
||
logger = logging.getLogger('comment')
|
||
logger.setLevel(level=_logging_level)
|
||
# NOTE: `%(levelname)s` will be parsed as the original name (`FATAL` ->
|
||
# `CRITICAL`, `WARN` -> `WARNING`).
|
||
# NOTE: The alignment number should set to 19 considering the style
|
||
# controling characters. When it comes to file logger, the number should
|
||
# set to 8.
|
||
formatter = StyleFormatter('%(asctime)s %(levelname)-19s %(message)s', "%F %T")
|
||
rawformatter = StyleFormatter('%(asctime)s %(levelname)-8s %(message)s', use_style=False)
|
||
console = logging.StreamHandler()
|
||
console.setLevel(_logging_level)
|
||
console.setFormatter(logging.Formatter('%(message)s'))
|
||
logger.addHandler(console)
|
||
opts['logger'] = logger
|
||
# It's a hack!!!
|
||
jieba.default_logger = logging.getLogger('jieba')
|
||
jieba.default_logger.setLevel(level=_logging_level)
|
||
jieba.default_logger.addHandler(console)
|
||
# It's another hack!!!
|
||
jdspider.default_logger = logging.getLogger('spider')
|
||
jdspider.default_logger.setLevel(level=_logging_level)
|
||
jdspider.default_logger.addHandler(console)
|
||
|
||
logger.debug('Successfully set up console logger')
|
||
logger.debug('CLI arguments: %s', args)
|
||
logger.debug('Opening the log file')
|
||
if opts['log_file']:
|
||
try:
|
||
handler = logging.FileHandler(opts['log_file'])
|
||
except Exception as e:
|
||
logger.error('Failed to open the file handler')
|
||
logger.error('Error message: %s', e)
|
||
sys.exit(1)
|
||
handler.setLevel(_logging_level)
|
||
handler.setFormatter(rawformatter)
|
||
logger.addHandler(handler)
|
||
jieba.default_logger.addHandler(handler)
|
||
jdspider.default_logger.addHandler(handler)
|
||
logger.debug('Successfully set up file logger')
|
||
logger.debug('Options passed to functions: %s', opts)
|
||
logger.debug('Builtin constants:')
|
||
logger.debug(' CONFIG_PATH: %s', CONFIG_PATH)
|
||
logger.debug(' USER_CONFIG_PATH: %s', USER_CONFIG_PATH)
|
||
logger.debug(' ORDINARY_SLEEP_SEC: %s', ORDINARY_SLEEP_SEC)
|
||
logger.debug(' SUNBW_SLEEP_SEC: %s', SUNBW_SLEEP_SEC)
|
||
logger.debug(' REVIEW_SLEEP_SEC: %s', REVIEW_SLEEP_SEC)
|
||
logger.debug(' SERVICE_RATING_SLEEP_SEC: %s', SERVICE_RATING_SLEEP_SEC)
|
||
|
||
# parse configurations
|
||
# logger.debug('Reading the configuration file')
|
||
# if os.path.exists(USER_CONFIG_PATH):
|
||
# logger.debug('User configuration file exists')
|
||
# _cfg_path = USER_CONFIG_PATH
|
||
# else:
|
||
# logger.debug('User configuration file doesn\'t exist, fallback to the default one')
|
||
# _cfg_path = CONFIG_PATH
|
||
# with open(_cfg_path, 'r', encoding='utf-8') as f:
|
||
# cfg = yaml.safe_load(f)
|
||
# print()
|
||
# logger.debug('Closed the configuration file')
|
||
# logger.debug('Configurations in Python-dict format: %s', cfg)
|
||
cks = get_ck()
|
||
if "OPENAI_API_KEY" in os.environ:
|
||
logger.info('已启用AI评价')
|
||
if "OPENAI_API_BASE_URL" in os.environ:
|
||
logger.info(' - 使用 OpenAI API 代理：' + os.environ["OPENAI_API_BASE_URL"])
|
||
elif os.environ.get("ProxyUrl").startswith("http"):
|
||
os.environ['http_proxy'] = os.getenv("ProxyUrl")
|
||
os.environ['https_proxy'] = os.getenv("ProxyUrl")
|
||
logger.info(' - 使用QL配置文件ProxyUrl代理：' + os.environ["ProxyUrl"])
|
||
else:
|
||
logger.info(' - 未使用代理，请确认当前网络环境可直连：api.openai.com')
|
||
try:
|
||
i = 1
|
||
for ck in cks:
|
||
headers = {
|
||
'cookie': ck.encode("utf-8"),
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.82 Safari/537.36',
|
||
'Connection': 'keep-alive',
|
||
'Cache-Control': 'max-age=0',
|
||
'sec-ch-ua': '" Not A;Brand";v="99", "Chromium";v="98", "Google Chrome";v="98"',
|
||
'sec-ch-ua-mobile': '?0',
|
||
'sec-ch-ua-platform': '"Windows"',
|
||
'DNT': '1',
|
||
'Upgrade-Insecure-Requests': '1',
|
||
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
|
||
'Sec-Fetch-Site': 'same-site',
|
||
'Sec-Fetch-Mode': 'navigate',
|
||
'Sec-Fetch-User': '?1',
|
||
'Sec-Fetch-Dest': 'document',
|
||
'Referer': 'https://order.jd.com/',
|
||
'Accept-Encoding': 'gzip, deflate, br',
|
||
'Accept-Language': 'zh-CN,zh;q=0.9',
|
||
}
|
||
logger.debug('Builtin HTTP request header: %s', headers)
|
||
logger.debug('Starting main processes')
|
||
logger.info('\n开始第 ' + str(i) + ' 个账号评价...\n')
|
||
main(opts)
|
||
i += 1
|
||
# NOTE: It needs 3,000 times to raise this exception. Do you really want to
|
||
# do like this?
|
||
except RecursionError:
|
||
logger.error("多次出现未完成情况，程序自动退出")
|