TrialCommend/TR/TrialRecommend/Main.py

import csv
import random
import time
import tkinter as tk
from io import BytesIO
from tkinter import messagebox

import requests
from PIL import Image, ImageTk
from TrialRecommend import DataRequest, Notes_Request

import func_collec
class Mainpage:
    def __init__(self,master,name):
        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_csv_file_path = 'data.csv'
        self.notes_csv_file_path = 'notes.csv'
        self.data_request = None
        self.research = tk.StringVar()
        self.user = name
        self.root = master
        self.root.title('推荐中心 v0.0.1')
        # 调用函数来居中窗口
        self.center_window(self.root)
        self.init_main_Page()

    def init_main_Page(self):
        self.page = tk.Frame(self.root)
        self.page.place(relx=0, rely=0, relheight=1, relwidth=1)


        self.btn_exit = tk.Button(self.page, text='退出', command=self.exit, bg='#D7EFFF', fg='red',font='bold')
        self.btn_exit.place(relx=0.9, rely=0.9, relheight=0.05, relwidth=0.07)

        self.lable_user = tk.Label(self.page, text='欢迎您：' + self.user, fg='green', font=24)
        self.lable_user.place(relx=0.86, rely=0, relheight=0.08, relwidth=0.13)

        self.text = tk.Text(self.page, bg='#ADD9FE', font=18)
        self.text.insert(tk.END, self.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.page,text='上一张',bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.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.page, text='下一张', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.next_image)
        self.btn_Image_next.place(relx=0.24, rely=0.01, relheight=0.05, relwidth=0.07)

        self.btn_liked_page = tk.Button(self.page, text='我的收藏夹', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.to_like_page)
        self.btn_liked_page.place(relx=0.9, rely=0.1, relheight=0.05, relwidth=0.1)

        self.btn_next = tk.Button(self.page, text='下一篇', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.next_data)
        self.btn_next.place(relx=0.45, rely=0.45, relheight=0.08, relwidth=0.09)

        self.btn_pre = tk.Button(self.page, text='上一篇', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.back_data)
        self.btn_pre.place(relx=0.45, rely=0.35, relheight=0.08, relwidth=0.09)

        self.btn_liked = tk.Button(self.page, text='收藏攻略', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.func_collect)
        self.btn_liked.place(relx=0.9, rely=0.2, relheight=0.05, relwidth=0.1)

        self.btn_dis_liked = tk.Button(self.page, 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.page,text='喜欢人数:'+self.likes,font=('SimHei',15,'bold'))
        self.lable_like.place(relx=0.05, rely=0.93)

        self.lable_like = tk.Label(self.page, text='收藏人数:'+self.collect,font=('SimHei',15,'bold'))
        self.lable_like.place(relx=0.17, rely=0.93)

        self.lable_like = tk.Label(self.page, text='转发人数:'+self.share,font=('SimHei',15,'bold'))
        self.lable_like.place(relx=0.29, rely=0.93)

        self.entry=tk.Entry(self.page,textvariable=self.research,font=('',20, "bold"))
        self.entry.place(relx=0.35, rely=0.01, relheight=0.05, relwidth=0.3)

        self.btn_research=tk.Button(self.page,text='搜索', bg='#D7EFFF',font=("SimHei",20, "bold"),command=self.researchPage)
        self.btn_research.place(relx=0.66, rely=0.01, relheight=0.05, relwidth=0.1)

        self.lable_time=tk.Label(self.page,text='发布时间：'+self.time,font=('SimHei',15,'bold'))
        self.lable_time.place(relx=0.41, rely=0.93)

    def to_init_page(self):
        self.page2.destroy()
        self.init_text()
        self.init_main_Page()
        self.update_max()
        self.image_url(self.current_data_index, self.current_data_image_index)
    def to_like_page(self):
        self.page.destroy()
        self.init_like_page()

    def init_like_page(self):
        self.page2=tk.Frame(self.root)
        self.page2.place(relx=0,rely=0,relheight=1,relwidth=1)

        self.btn_init_page = tk.Button(self.page2, text='返回主页面',command=self.to_init_page, bg='#D7EFFF',font=('SimHei',15,'bold'))
        self.btn_init_page.place(relx=0.9, rely=0.1, relheight=0.05, relwidth=0.1)

        self.lable_user = tk.Label(self.page2, text='欢迎您：' + self.user, fg='green', font=24)
        self.lable_user.place(relx=0.86, rely=0, relheight=0.08, relwidth=0.13)

        self.btn_exit = tk.Button(self.page2, text='退出', command=self.exit, bg='#D7EFFF', fg='red',font='bold')
        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.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.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.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.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.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_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:
                with open('data.csv', 'r', encoding='utf-8') as f:
                    if self.current_notes_index!=self.flag:
                        self.current_notes_index += 1
                        self.current_data_index += 1
                    else:
                        self.current_notes_index += 1
                        self.current_data_index += 1
                        self.flag=self.current_notes_index
                    csv_reader = csv.reader(f)
                    data=''
                    i=0
                    for row in csv_reader:
                        if i == self.current_data_index:
                            data = row[0]
                            break
                        i += 1
                    if self.is_pre=='False':
                        Notes_Request.notes_request(data)
                    if self.flag==self.current_notes_index:
                        self.is_pre='False'
                time.sleep(random.uniform(0, 1))
                self.update_max()
                #print(self.max_data_count, self.current_data_index, self.current_data_image_index)
                self.init_text()
                self.init_main_Page()
                self.image_show()
            else:
                messagebox.showinfo(message='正在载入新的一批数据，请稍等！')
                self.data_request.dataRequest()
                if self.data_request.success == 'True':
                    time.sleep(random.uniform(1, 2))
                    with open('data.csv', 'r', encoding='utf-8') as f:
                        if self.current_notes_index != self.flag:
                            self.current_notes_index += 1
                            self.current_data_index += 1
                        else:
                            self.current_notes_index += 1
                            self.current_data_index += 1
                            self.flag = self.current_notes_index
                        csv_reader = csv.reader(f)
                        data=''
                        i=0
                        for row in csv_reader:
                            if i == self.current_data_index:
                                data = row[0]
                            i += 1
                        if self.is_pre == 'False':
                            Notes_Request.notes_request(data)
                        if self.flag == self.current_notes_index:
                            self.is_pre = 'True'
                        time.sleep(random.uniform(0, 1))
                    self.update_max()
                    print(self.max_data_count, self.current_data_index, self.current_data_image_index)
                    self.init_text()
                    self.init_main_Page()
                    self.image_show()
                else:
                    messagebox.showinfo(message='网络请求失败，请稍后尝试！')

        if next_back == 'back':
            if self.current_notes_index-1<0:
                messagebox.showinfo(message='这是第一篇文章')
            else:
                self.is_pre = 'True'
                self.current_notes_index -= 1
                self.current_data_index -= 1
                with open('data.csv', 'r', encoding='utf-8') as f:
                    csv_reader = csv.reader(f)
                    first_row = ''
                    i=0
                    for row in csv_reader:
                        if i==self.current_data_index:
                            first_row=row[0]
                        i += 1
                    Notes_Request.notes_request(first_row)
                    time.sleep(random.uniform(0, 1))
                self.update_max()
                print(self.max_data_count, self.current_data_index, self.current_data_image_index)
                self.init_text()
                self.init_main_Page()
                self.image_show()
    # 返回上一篇
    def back_data(self):
        if self.max_data_count != 0:
            self.current_data_image_index = 0
            self.notes_request('back')
        else:
            messagebox.showinfo(message='请先搜索')
    # 下一篇
    def next_data(self):
        if self.max_data_count != 0:
            self.current_data_image_index = 0
            self.notes_request('next')
        else:
            messagebox.showinfo(message='请先搜索')
    # 下一张图片
    def next_image(self):

        if self.max_iamge_count == 0:
            messagebox.showinfo(message='请先搜索')
        elif self.current_data_image_index + 1 >= self.max_iamge_count:
            messagebox.showinfo(message='这是最后一张')
        else:
            self.current_data_image_index += 1
            self.image_url(self.current_data_index, self.current_data_image_index)
    # 上一张图片
    def pre_image(self):
        if self.max_iamge_count == 0:
            messagebox.showinfo(message='请先搜索')
        elif self.current_data_image_index - 1 < 0:
            messagebox.showinfo(title='提示', message='没有上一张了')
        else:
            self.current_data_image_index -= 1
            self.image_url(self.current_data_index, self.current_data_image_index)
    def researchPage(self):
        self.clear_csv_file()
        rs=self.research.get()
        if rs=='':
            messagebox.showinfo(title='提示', message='你还没有输入任何东西！')
        else:
            if DataRequest.DataRequest(str(self.research.get())).success=='True':
                self.data_request = DataRequest.DataRequest(str(self.research.get()))
                time.sleep(random.uniform(0,1))
                self.init_notes_request()
            else:
                messagebox.showinfo(message='网络请求失败，请稍后尝试！')
    def init_notes_request(self):
        with open('data.csv', 'r', encoding='utf-8') as f:
            reader = csv.reader(f)
            for row in reader:
                Notes_Request.notes_request(row[0])
                break
        self.update_max()
        self.init_text()
        self.init_main_Page()
        self.image_show()
    def image_url(self,index,col):
        with open(self.data_csv_file_path,'r',encoding='utf-8') as f:
            reader = csv.reader(f)
            row =''
            i=0
            for rows in reader:
                if i==self.current_data_index:
                    row=rows
                i += 1
            self.load_image_from_url(self.page,eval(row[3])[col])
    def load_image_from_url(self,View,url):
        # 使用requests从URL下载图片
        response = requests.get(url)
        img = Image.open(BytesIO(response.content))
        img = img.resize((450, 600), Image.BICUBIC)  # 调整像素，使用高质量重采样算法
        # 使用PIL的ImageTk将PIL图片转换为Tkinter可以显示的格式
        tk_img = ImageTk.PhotoImage(img)

        # 创建一个标签来显示图片，并添加到Tkinter窗口中
        label = tk.Label(View, image=tk_img)
        label.image = tk_img  # 保持对tk_img的引用，防止被垃圾回收
        label.place(relx=0.05, rely=0.07)
    def image_show(self):
        if self.csv_is_empty():
            messagebox.showinfo('提示',message='数据加载失败')
        else:
            self.image_url(self.current_data_index,0)
    def csv_is_empty(self):
        with open('notes.csv','r',encoding='utf-8') as f2:
            rows = []
            reader = csv.reader(f2)
            for row in reader:
                # 如果找到非空行，则将rows列表设置为非空
                if any(row):
                    rows.append(row)
                    return False # 如果只需要检查是否至少有一行非空，则可以提前退出循环
                # 如果rows列表为空，则CSV文件是空的（或只包含空行）
            if len(rows) == 0:
                print('notes 无数据')
                return True
    def init_text(self):
        with open(self.notes_csv_file_path,'r',encoding='utf-8') as f:
            reader = csv.reader(f)
            index=0
            for row in reader:
                if index==self.current_notes_index:
                    title = row[0]
                    text = row[1].replace('<br/>', '\n').replace('.<br/>', '')
                    self.content=title+text
                    self.time = row[2]
                    self.collect = row[3]
                    self.likes = row[4]
                    self.share = row[5]
                index += 1
    def update_max(self):
        with open(self.data_csv_file_path,'r',encoding='utf-8') as f:
            reader = csv.reader(f)
            index=0
            for row in reader:
                if index==self.current_data_index:
                    self.max_iamge_count = len(eval(row[3]))
                    break
                index+=1

        with open(self.data_csv_file_path, 'r', encoding='utf-8') as f:
            reader = csv.reader(f)
            counts = 0
            for i in reader:
                counts += 1
            self.max_data_count = counts

        with open(self.notes_csv_file_path,'r',encoding='utf-8') as f:
            reader = csv.reader(f)
            counts=0
            for i in reader:
                counts+=1
            self.max_notes_count=counts
    def clear_csv_file(self):
        with open(self.data_csv_file_path, 'w') as file1:  # 使用'w'模式打开文件
            pass  # 不需要执行任何操作，因为打开文件时内容已被清空
        with open(self.notes_csv_file_path, 'w') as file2:
            pass  # 不需要执行任何操作，因为打开文件时内容已被清空
    def exit(self):
        messagebox.showinfo(title='提示', message='欢迎下次使用！')
        self.root.quit()
    def center_window(self, root, width=1300, height=700):
        # 获取屏幕宽度和高度
        screen_width = root.winfo_screenwidth()
        screen_height = root.winfo_screenheight()

        # 计算窗口的左上角应该放置的位置
        left = (screen_width / 2) - (width / 2)
        top = (screen_height / 2) - (height / 2)

        # 设置窗口的位置和大小
        root.geometry("{0}x{1}+{2}+{3}".format(width, height, int(left), int(top)))

    def func_collect(self):
        if self.max_data_count!=0:
            func_collec.collec(self.user, self.current_data_index)
        else:
            messagebox.showinfo('提示','请先搜索')

if __name__ == '__main__':
    root = tk.Tk()
    Mainpage(master=root,name='Null')
    root.mainloop()