commit008

TR/TrialRecommend/Main.py

@@ -1,5 +1,7 @@
 import csv
+import os
 import random
+import tempfile
 import time
 import tkinter as tk
 from io import BytesIO
@@ -13,6 +15,12 @@ import func_collec
 import dis_collect
 class Mainpage:
     def __init__(self,master,name):
+        self.c_data_index=0
+        self.c_image_index=0
+        self.c_content='暂无数据!'
+        self.c_image_max=0
+        self.c_data_max=0
+
         self.content = ''
         self.time = ''
         self.collect = ''
@@ -100,11 +108,101 @@ class Mainpage:
         self.init_text()
         self.init_main_Page()
         self.update_max()
-        self.image_url(self.current_data_index, self.current_data_image_index)
+        self.image_url(self.current_data_image_index)
 
     def to_like_page(self):
+        data=0
+        with open('collections/{}.csv'.format(self.user),'r+',newline='',encoding='utf-8') as f:
+            reader = csv.reader(f)
+            if len(list(reader))>0:
+                data=1
+
+        if data==1:
+            self.c_data_index = 0
+            self.c_image_index = 0
+            self.c_content = '暂无数据!'
+            self.c_image_max = 0
+            self.c_data_max = 0
+
             self.page.destroy()
+            self.c_init_text()
+            self.init_like_page()
+            self.c_image_url()
+        else:
+            self.init_like_page()
+    def c_next_data(self):
+        if self.c_data_index+1 < self.c_data_max:
+            self.c_data_index += 1
+            self.c_image_index = 0
+            self.c_init_text()
+            self.init_like_page()
+            self.c_image_url()
+        else:
+            messagebox.showinfo(message='这是最后一篇')
+    def c_pre_data(self):
+        if self.c_data_index-1<0:
+            messagebox.showinfo(message='这是第一篇')
+        else:
+            self.c_data_index -= 1
+            self.c_image_index=0
+            self.c_init_text()
             self.init_like_page()
+            self.c_image_url()
+
+    def c_image_url(self):
+        with open('collections/{}.csv'.format(self.user),'r+',encoding='utf-8') as f:
+            reader = csv.reader(f)
+            row =''
+            i=0
+            for rows in reader:
+                if i==self.c_data_index:
+                    row=rows
+                    break
+                i += 1
+            self.load_image_from_url(self.page2,eval(row[2])[self.c_image_index])
+
+    def c_pre_image(self):
+        if self.c_image_index - 1 < 0:
+            messagebox.showinfo(title='提示', message='没有上一张了')
+        else:
+            self.c_image_index -= 1
+            self.c_image_url()
+
+    def c_next_image(self):
+
+        if self.c_image_index + 1 >= self.c_image_max:
+            messagebox.showinfo(message='这是最后一张')
+        else:
+            self.c_image_index += 1
+            self.c_image_url()
+
+    def c_init_text(self):
+        with open('collections/{}.csv'.format(self.user),'r+',encoding='utf-8') as f:
+            reader = csv.reader(f)
+            i=0
+            for row in reader:
+                i+=1
+            self.c_data_max=i
+        flag=0
+        with open('collections/{}.csv'.format(self.user),'r+',encoding='utf-8') as f:
+            reader = csv.reader(f)
+            if not next(reader):
+                self.c_content='暂无收藏数据'
+            else:
+                flag=1
+
+        if flag==1:
+            with open('collections/{}.csv'.format(self.user), 'r+', encoding='utf-8') as f:
+                reader = csv.reader(f)
+                index = 0
+                for row in reader:
+                    if index == self.c_data_index:
+                        title = row[1]
+                        text = row[3].replace('<br/>', '\n').replace('.<br/>', '\n')
+                        self.c_content = title + text
+                        self.c_image_max=len(eval(row[2]))
+                        break
+                    index += 1
 
     def init_like_page(self):
         self.page2=tk.Frame(self.root)
@@ -120,37 +218,25 @@ class Mainpage:
         self.btn_exit.place(relx=0.9, rely=0.9, relheight=0.05, relwidth=0.07)
 
         self.text = tk.Text(self.page2, bg='#ADD9FE', font=18)
-        self.text.insert(tk.END, '收藏页面')
+        self.text.insert(tk.END, self.c_content)
         self.text.config(state=tk.DISABLED, cursor='')
         self.text.place(relx=0.58, rely=0.07, relheight=0.86, relwidth=0.3)
 
-        self.btn_Image_pre = tk.Button(self.page2, text='上一张', bg='#D7EFFF',font=('SimHei',15,'bold'))
+        self.btn_Image_pre = tk.Button(self.page2, text='上一张', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.c_pre_image)
         self.btn_Image_pre.place(relx=0.14, rely=0.01, relheight=0.05, relwidth=0.07)
 
-        self.btn_Image_next = tk.Button(self.page2, text='下一张', bg='#D7EFFF',font=('SimHei',15,'bold'))
+        self.btn_Image_next = tk.Button(self.page2, text='下一张', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.c_next_image)
         self.btn_Image_next.place(relx=0.24, rely=0.01, relheight=0.05, relwidth=0.07)
 
-        self.btn_next = tk.Button(self.page2, text='下一篇', bg='#D7EFFF',font=('SimHei',15,'bold'))
+        self.btn_next = tk.Button(self.page2, text='下一篇', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.c_next_data)
         self.btn_next.place(relx=0.45, rely=0.45, relheight=0.08, relwidth=0.09)
 
-        self.btn_pre = tk.Button(self.page2, text='上一篇', bg='#D7EFFF',font=('SimHei',15,'bold'))
+        self.btn_pre = tk.Button(self.page2, text='上一篇', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.c_pre_data)
         self.btn_pre.place(relx=0.45, rely=0.35, relheight=0.08, relwidth=0.09)
 
-        self.btn_liked = tk.Button(self.page2, text='收藏攻略', bg='#D7EFFF',font=('SimHei',15,'bold'))
+        self.btn_dis_liked = tk.Button(self.page2, text='取消收藏', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.func_discollect)
-        self.btn_liked.place(relx=0.9, rely=0.2, relheight=0.05, relwidth=0.1)
-
-        self.btn_dis_liked = tk.Button(self.page2, text='取消收藏', bg='#D7EFFF',font=('SimHei',15,'bold'))
         self.btn_dis_liked.place(relx=0.9, rely=0.3, relheight=0.05, relwidth=0.1)
 
-        self.lable_like = tk.Label(self.page2, text='喜欢人数:', font=('SimHei',15,'bold'))
-        self.lable_like.place(relx=0.05, rely=0.93)
-
-        self.lable_like = tk.Label(self.page2, text='收藏人数:', font=('SimHei',15,'bold'))
-        self.lable_like.place(relx=0.17, rely=0.93)
-
-        self.lable_like = tk.Label(self.page2, text='转发人数:', font=('SimHei',15,'bold'))
-        self.lable_like.place(relx=0.29, rely=0.93)
-
     def notes_request(self,next_back):
         if next_back == 'next':
             if self.current_data_index + 1 <= self.max_data_count:
@@ -261,7 +347,7 @@ class Mainpage:
             messagebox.showinfo(message='这是最后一张')
         else:
             self.current_data_image_index += 1
-            self.image_url(self.current_data_index, self.current_data_image_index)
+            self.image_url(self.current_data_image_index)
 
     # 上一张图片
     def pre_image(self):
@@ -280,6 +366,23 @@ class Mainpage:
             messagebox.showinfo(title='提示', message='你还没有输入任何东西！')
         else:
             if DataRequest.DataRequest(str(self.research.get())).success=='True':
+
+                self.content = ''
+                self.time = ''
+                self.collect = ''
+                self.likes = ''
+                self.share = ''
+
+                self.flag = 0
+                self.is_pre = 'False'
+
+                self.current_data_index = 0
+                self.current_data_image_index = 0
+                self.current_notes_index = 0
+                self.max_data_count = 0
+                self.max_notes_count = 0
+                self.max_iamge_count = 0
+
                 self.data_request = DataRequest.DataRequest(str(self.research.get()))
                 time.sleep(random.uniform(0,1))
                 self.init_notes_request()
@@ -325,7 +428,7 @@ class Mainpage:
         if self.csv_is_empty():
             messagebox.showinfo('提示',message='数据加载失败')
         else:
-            self.image_url(self.current_data_index,0)
+            self.image_url(0)
 
     def csv_is_empty(self):
         with open('notes.csv','r',encoding='utf-8') as f2:
@@ -405,11 +508,57 @@ class Mainpage:
     def func_collect(self):
         if self.max_data_count!=0:
             func_collec.collec(self.user, self.current_data_index)
+            dic={}
+            with open(self.data_csv_file_path,'r',encoding='utf-8') as f:
+                reader = csv.reader(f)
+                i=0
+                for row in reader:
+                    if i==self.current_data_index:
+                        dic={
+                            '文章ID':row[0],
+                            '文章标题':row[1],
+                            '图片资源':eval(row[3])
+                        }
+                        break
+                    i+=1
+
+            with open(self.notes_csv_file_path,'r',encoding='utf-8') as f:
+                reader = csv.reader(f)
+                i=0
+                for row in reader:
+                    if i==self.current_data_index:
+                        dic['文章内容']=row[1]
+
+            with open('collections/{}.csv'.format(self.user),'a+',encoding='utf-8',newline='') as f:
+                writer = csv.DictWriter(f, fieldnames=['文章ID','文章标题','图片资源','文章内容'])
+                writer.writerow(dic)
+
         else:
             messagebox.showinfo('提示','请先搜索')
 
     def func_discollect(self):
-        dis_collect.dis_collec(self.user,self.current_data_index)
+        dis_collect.dis_collec(self.user,self.c_data_index)
+
+        row_to_delete = self.c_data_index
+
+        # 使用临时文件来存储修改后的内容
+        with open('collections/{}.csv'.format(self.user), newline='', encoding='utf-8') as csvfile, \
+                tempfile.NamedTemporaryFile(mode='w+t', newline='', encoding='utf-8', delete=False) as temp:
+
+            reader = csv.reader(csvfile)
+            writer = csv.writer(temp)
+
+            # 读取并写入除了要删除的行之外的所有行
+            for i, row in enumerate(reader):
+                if i != row_to_delete:
+                    writer.writerow(row)
+        try:
+            os.remove('collections/{}.csv'.format(self.user))
+        except FileNotFoundError:
+            pass
+
+        # 将临时文件重命名为原始文件名
+        os.rename(temp.name, 'collections/{}.csv'.format(self.user))
 
 
 if __name__ == '__main__':

TR/TrialRecommend/Notes_Request.py

@@ -19,11 +19,11 @@ def notes_request(url):
         'a1':'18f9b67ecdbh41bri2la0dgguminrd7dmmrk4pxf750000215521',
         'webId':'6d97ebd3e6051489537e7a6aa1d1cf99',
         'gid':'yYijDKYydJ6JyYijDKWdS6UCfDx4yDF3J108FiWfkk763T28IFfWf4888Jy22Jy8WYf0SyjD',
-        'webBuild':'4.16.1',
+        'webBuild':'4.17.2',
         'web_session':'040069b3643d32585fff79ee79344bd4b89bc6',
-        'acw_tc':'25f299cc55249ae472ab6eab4f8404ce338ad4b4544c7f87a592375d787ede3a',
+        'acw_tc':'459a6388281f21d6215ff80f1402afbfdf47eb5a1ef949b426ace99565a6fc50',
-        'websectiga':'16f444b9ff5e3d7e258b5f7674489196303a0b160e16647c6c2b4dcb609f4134',
+        'websectiga':'7750c37de43b7be9de8ed9ff8ea0e576519e8cd2157322eb972ecb429a7735d4',
-        'sec_poison_id':'9e6926ce-0bb8-4335-992a-0ec7875fff69',
+        'sec_poison_id':'b1e1329b-735f-427c-bfd0-698a912c0366',
         'xsecappid':'xhs-pc-web'
 
     }

TR/TrialRecommend/dis_collect.py

@@ -4,12 +4,13 @@ from tkinter import messagebox
 
 def dis_collec(user,data_index):
     id=''
-    with open('data.csv', 'r', encoding='utf-8', newline='') as file:
+    with open('collections/{}.csv'.format(user), 'r', encoding='utf-8', newline='') as file:
         reader = csv.reader(file)
         index=0
         for row in reader:
             if index == data_index:
                 id=row[0]
+                break
             index=index+1
     db = mysql.connector.connect(
         host="localhost",

TR/TrialRecommend/test.py

@@ -0,0 +1,7 @@
+import csv
+
+with open('data.csv','r',encoding='utf-8') as csvfile:
+    reader = csv.reader(csvfile)
+    for row in reader:
+        print(len(eval(row[3])))
+        break