TrialCommend/TR/TrialRecommend/Main.py

import csv
import os
import random
import shutil
import tempfile
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
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 = ''
        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'),command=self.re)
        # 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_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)
        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.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'),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'),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'),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'),command=self.c_pre_data)
        self.btn_pre.place(relx=0.45, rely=0.35, relheight=0.08, relwidth=0.09)

        self.btn_dis_liked = tk.Button(self.page2, text='取消收藏', bg='#D7EFFF',font=('SimHei',15,'bold'),command=self.remove_rows_with_content)
        self.btn_dis_liked.place(relx=0.9, rely=0.3, relheight=0.05, relwidth=0.1)

    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_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_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()), self.user).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()), self.user)
                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,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(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)
            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]
                    i+=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.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))

    def remove_rows_with_content(self):
        dis_collect.dis_collec(self.user, self.c_data_index)

        # 创建一个临时文件
        with tempfile.NamedTemporaryFile(mode='w', delete=False,encoding='utf-8',newline='') as tmpfile:
            # 打开原始文件并读取内容
            with open('collections/{}.csv'.format(self.user), 'r', newline='',encoding='utf-8') as infile:
                reader = csv.reader(infile)
                writer = csv.writer(tmpfile)

                # 遍历每一行，如果不包含指定内容则写入临时文件
                i=0
                for row in reader:
                    if self.c_data_index != i:
                        writer.writerow(row)
                    i+=1

                        # 替换原始文件为临时文件
            tmpfile.close()
            shutil.move(tmpfile.name, 'collections/{}.csv'.format(self.user))

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