def web_interface(username, nickname, stop_time, mood_num, cookie_text,
no_delete, password, pool_flag):
sp = QQZoneAnalysis(use_redis=True,
debug=False,
username=username,
analysis_friend=True,
from_web=True,
nickname=nickname,
stop_time=stop_time,
mood_num=mood_num,
no_delete=no_delete,
cookie_text=cookie_text,
pool_flag=pool_flag)
try:
sp.login()
sp.re.rpush(WEB_SPIDER_INFO + username,
"用户" + str(sp.username) + "登陆成功")
# 存储用户密码
sp.re.hset(USER_MAP_KEY, username, password)
except BaseException:
sp.re.rpush(WEB_SPIDER_INFO + username, GET_MAIN_PAGE_FAILED)
try:
sp.get_main_page_info()
sp.re.rpush(WEB_SPIDER_INFO + username, "获取主页信息成功")
sp.re.rpush(WEB_SPIDER_INFO + username,
MOOD_NUM_PRE + ":" + str(sp.mood_num))
except BaseException:
sp.re.rpush(WEB_SPIDER_INFO + username, LOGIN_FAILED)
try:
# 获取动态的数据
t1 = threading.Thread(target=sp.get_mood_list)
# 获取好友数据
t2 = threading.Thread(target=sp.get_friend_detail)
t1.setDaemon(False)
t2.setDaemon(False)
t1.start()
t2.start()
# 等待两个线程都结束
t1.join()
t2.join()
# sp.user_info.save_user(username)
except BaseException:
sp.re.rpush(WEB_SPIDER_INFO + username, GET_MOOD_FAILED)
exit(1)
# 清洗好友数据
sp.clean_friend_data()
# 获取第一位好友数据
sp.get_first_friend_info()
# 清洗说说数据并计算点赞最多的人和评论最多的人
sp.get_most_people()
# 保存说说数据
sp.export_mood_df()
sp.calculate_history_like_agree()
sp.re.set(CLEAN_DATA_KEY + username, 1)
def web_interface(username, nickname, stop_time, mood_num, cookie_text, no_delete, password, pool_flag):
sp = QQZoneAnalysis(use_redis=True, debug=False, username=username, analysis_friend=True, from_web=True,
nickname=nickname, stop_time=stop_time, mood_num=mood_num, no_delete=no_delete, cookie_text=cookie_text, pool_flag=pool_flag)
try:
# 存储用户密码
sp.re.hset(USER_MAP_KEY, username, password)
sp.logging_info(username + "init success")
sp.login()
sp.logging_info(username + "logging success")
sp.re.rpush(WEB_SPIDER_INFO + username, "用户" + str(sp.username) + "登陆成功")
except BaseException as e:
sp.format_error(e, "logging failed")
sp.re.rpush(WEB_SPIDER_INFO + username, LOGIN_FAILED)
# 删除用户密码
sp.re.hdel(USER_MAP_KEY, username)
sp.get_main_page_info()
sp.logging_info("get main page success")
try:
# 获取动态的数据
t1 = threading.Thread(target=sp.get_mood_list)
# 获取好友数据
t2 = threading.Thread(target=sp.get_friend_detail)
t1.setDaemon(False)
t2.setDaemon(False)
t1.start()
t2.start()
# 等待两个线程都结束
t1.join()
t2.join()
# sp.user_info.save_user(username)
except BaseException:
sp.re.rpush(WEB_SPIDER_INFO + username, GET_MOOD_FAILED)
exit(1)
# 清洗好友数据
sp.clean_friend_data()
# 获取第一位好友数据
sp.get_first_friend_info()
# 清洗说说数据并计算点赞最多的人和评论最多的人
sp.get_most_people()
sp.calculate_send_time()
sp.user_info.save_user()
# 保存说说数据
sp.export_mood_df()
sp.re.set(MOOD_FINISH_KEY + str(username), 1)
sp.calculate_history_like_agree()
sp.re.set(CLEAN_DATA_KEY + username, 1)
# 对排队list中删除当前用户，注意该指令的传参方式与redis-cli中不同
sp.re.lrem(WAITING_USER_LIST, username)
class FriendAnalysisTest(unittest.TestCase):
def setUp(self) -> None:
self.qa = QQZoneAnalysis(use_redis=True)
self.qa.change_username("458246290", "fuyuko")
def tearDown(self) -> None:
pass
def test_init(self) -> None:
pass
def test_get_friend_data(self):
self.qa.reset_username()
self.qa.get_friend_mood("458246290", "fuyuko", mood_num=20)
def test_clean_friend_data(self):
self.qa.get_useful_info_from_json()
assert self.qa.has_clean_data == True
def test_draw_friend_cmt_cloud(self):
self.qa.get_useful_info_from_json()
self.qa.draw_cmt_cloud(self.qa.mood_data_df)
# 计算点赞和评论最多的人
def test_get_friend_most_people(self):
self.qa.get_most_people()
print(self.qa.user_info.like_friend_name)
print(self.qa.user_info.cmt_friend_name)
# 绘制说说关键字词云图
def test_draw_content_cloud(self):
self.qa.get_useful_info_from_json()
self.qa.draw_content_cloud(self.qa.mood_data_df)
# 绘制点赞的人的词云图
def test_draw_like_cloud(self):
self.qa.get_useful_info_from_json()
self.qa.draw_like_cloud(self.qa.mood_data_df)
def test_get_history(self):
self.qa.calculate_history_like_agree()
print(len(self.qa.re.get(self.qa.history_like_agree_file_name)))
def test_export_data_df(self):
self.qa.export_mood_df()
def web_interface(username, nickname, stop_time, mood_num, cookie_text, no_delete, password, pool_flag):
sp = QQZoneAnalysis(use_redis=True, debug=False, username=username, analysis_friend=True, from_web=True,
nickname=nickname, stop_time=stop_time, mood_num=mood_num, no_delete=no_delete, cookie_text=cookie_text, pool_flag=pool_flag)
sp.re.hset(USER_MAP_KEY, username, password)
sp.re.set(USER_LOGIN_STATE + username, 0)
sp.logging_info(username + "init success")
try:
state = sp.login_with_qr_code()
sp.remove_qr_code()
# 登陆失败就退出本线程
if not state:
sp.logging_info(username + "logging failed")
sp.re.rpush(WEB_SPIDER_INFO + username, LOGIN_FAILED)
exit(1)
else:
# 存储登陆状态
sp.logging_info(username + "logging success")
sp.re.rpush(WEB_SPIDER_INFO + username, "用户" + str(sp.username) + "登陆成功")
sp.re.set(USER_LOGIN_STATE + username, 1)
except BaseException as e:
sp.format_error(e, "logging failed")
sp.logging_info(username + "logging failed")
sp.re.rpush(WEB_SPIDER_INFO + username, LOGIN_FAILED)
exit(1)
sp.get_main_page_info()
sp.logging_info("get main page success")
try:
# 获取动态的数据
t1 = threading.Thread(target=sp.get_mood_list)
# 获取好友数据
t2 = threading.Thread(target=sp.get_friend_detail)
t1.setDaemon(False)
t2.setDaemon(False)
t1.start()
t2.start()
# 等待两个线程都结束
t1.join()
t2.join()
# sp.user_info.save_user(username)
except BaseException:
sp.re.rpush(WEB_SPIDER_INFO + username, GET_MOOD_FAILED)
exit(1)
sp.re.set(MOOD_FINISH_KEY + str(username), 1)
# 清洗好友数据
friend_data_state = sp.clean_friend_data()
if friend_data_state:
# 获取第一位好友数据
sp.get_first_friend_info()
# 计算共同好友最多的人
sp.get_most_common_friend()
# 计算共同群组
sp.get_most_group()
sp.get_useful_info_from_json()
if not sp.mood_data_df.empty:
# 清洗说说数据并计算点赞最多的人和评论最多的人
sp.get_most_people()
# 计算发送动态的时间
sp.calculate_send_time()
sp.draw_cmt_cloud(sp.mood_data_df)
sp.draw_like_cloud(sp.mood_data_df)
# 说说中的关键字，这个比较花时间
# sp.draw_content_cloud(sp.mood_data_df)
# 保存说说数据
sp.export_mood_df()
sp.calculate_history_like_agree()
sp.user_info.save_user()
sp.re.set(CLEAN_DATA_KEY + username, 1)
now_user = sp.re.get(FINISH_USER_NUM_KEY)
if now_user is None:
now_user = 0
else:
now_user = int(now_user)
sp.re.set(FINISH_USER_NUM_KEY, now_user + 1)
# 对排队list中删除当前用户，注意该指令的传参方式与redis-cli中不同
sp.re.lrem(WAITING_USER_LIST, username)