commit_0529

filesys/login.py

@@ -1,5 +1,7 @@
 import tkinter as tk
-from tkinter import filedialog
+from tkinter import messagebox
+from upload import *
+from test import *
 def login_1():
 
     new_window = tk.Tk()
@@ -10,43 +12,26 @@ def login_1():
     new_label = tk.Label(new_window, text="登陆成功!")
     new_label.pack()
 
-    button_frame = tk.Frame(new_window)
-    button_frame.pack()
-
-    upload_button = tk.Button(button_frame, text="上传文件", command=upload_file)
-    download_button = tk.Button(button_frame, text="下载文件", command=download_file)
-    delete_button = tk.Button(button_frame, text="删除文件", command=delete_file)
-
-    # 使用grid()布局管理器在button_frame内
-    upload_button.grid(row=4, column=0,padx=10, pady=10)
-    download_button.grid(row=4, column=3,padx=10, pady=10)
-    delete_button.grid(row=4, column=6,padx=10, pady=10)
-
-    # 退出按钮
-    exit_button = tk.Button(new_window, text="返回", command=new_window.destroy)
+    root = tk.Frame(new_window)
+    root.pack()
+
+    # 创建按钮
+    create_table_button = tk.Button(root, text="创建表", command=create_table)
+    upload_image_button = tk.Button(root, text="上传图片", command=upload_image)
+    download_button = tk.Button(root, text="下载图片", command=download_image)
+    delete_button = tk.Button(root, text="删除图片", command=delete_image)
+    exit_button = tk.Button(root, text="返回", command=new_window.destroy)
+
+    # 布局
+    create_table_button.pack()
+    upload_image_button.pack()
+    download_button.pack()
+    delete_button.pack()
     exit_button.pack()
 
     new_window.mainloop()
 
 
-
-def upload_file():
-
-    selected_file_path = filedialog.askopenfilename()
-    new_window = tk.Tk()
-    new_window.title("欢迎来到上传界面")
-    new_window.geometry("600x200")
-    # 更新界面，显示选定的文件路径
-    selected_file_label = tk.Label(new_window, text=f"选定文件: {selected_file_path}")
-    selected_file_label.pack()
-
-    new_window.mainloop()
-
-
-def download_file():
-    return None
-
-
-def delete_file():
-    return None
+if __name__ == "__main__":
+    login_1()
 

filesys/main.py

@@ -42,5 +42,5 @@ def main_0():
         erro_label = tk.Label(root, text="账号或密码错误")
         erro_label.pack()
 
-
-main()
+if __name__ == "__main__":
+    main()

filesys/register.py

@@ -10,13 +10,13 @@ def register():
 
     username_label = tk.Label(register_view, text="用户名:")
     username_label.pack()
-    username_entry_1 = tk.Entry(register_view)
-    username_entry_1.pack()
+    username_entry = tk.Entry(register_view)
+    username_entry.pack()
 
     password_label = tk.Label(register_view, text="密码:")
     password_label.pack()
-    password_entry_1 = tk.Entry(register_view)
-    password_entry_1.pack()
+    password_entry = tk.Entry(register_view)
+    password_entry.pack()
 
     register_button = tk.Button(register_view, text="注册", command=register_0)
     register_button.pack()
@@ -36,3 +36,6 @@ def register_0():
         add_user(entered_username, entered_password)
         ok_label = tk.Label(register_view, text="注册成功")
         ok_label.pack()
+
+if __name__ == "__main__":
+    register()

filesys/sql.py

@@ -1,6 +1,14 @@
 from flask import *
 from flask_sqlalchemy import SQLAlchemy
 import pymysql
+import os
+from tkinter import filedialog
+import pymysql.cursors
+
+DB_HOST = 'localhost'  # 数据库主机
+DB_USER = 'root'  # 数据库用户名
+DB_PASSWORD = '123456'  # 数据库密码
+DB_NAME = 'file'  # 数据库名
 
 app = Flask(__name__)
 app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:123456@localhost:3306/file'
@@ -22,6 +30,44 @@ def add_user(username, password):
         db.session.add(user)
         db.session.commit()
 
+def create_table():
+    connection = pymysql.connect(host=DB_HOST, user=DB_USER, password=DB_PASSWORD, db=DB_NAME)
+    try:
+        with connection.cursor() as cursor:
+            cursor.execute(CREATE_TABLE_SQL)
+        connection.commit()
+        print("Table 'images' created successfully.")
+    finally:
+        connection.close()
+
+def path_to_bytes():
+    return filedialog.askopenfilename(filetypes=(("Image Files", "*.jpg;*.png;*.gif"), ("All Files", "*.*")))
+def upload_image():
+    selected_image_path = path_to_bytes()
+    if selected_image_path:
+        with open(selected_image_path, 'rb') as image_file:
+            image_content = image_file.read()
+
+        connection = pymysql.connect(host=DB_HOST, user=DB_USER, password=DB_PASSWORD, db=DB_NAME)
+        try:
+            with connection.cursor() as cursor:
+                sql = "INSERT INTO images (filename, content) VALUES (%s, %s)"
+                cursor.execute(sql, (os.path.basename(selected_image_path), pymysql.Binary(image_content)))
+                connection.commit()
+                print(f"Image '{selected_image_path}' uploaded successfully.")
+        finally:
+            connection.close()
+
+CREATE_TABLE_SQL = """
+CREATE TABLE IF NOT EXISTS images (
+    id INT AUTO_INCREMENT PRIMARY KEY,
+    filename VARCHAR(255),
+    content LONGBLOB,
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+);
+"""
+
+
 try:
     with app.app_context():
         # 尝试连接数据库

filesys/test.py

@@ -1,69 +1,114 @@
-import pymysql
+import pymysql.cursors
+import tkinter as tk
+from tkinter import filedialog
 import os
 
+# 数据库连接参数
+DB_HOST = 'localhost'  # 数据库主机
+DB_USER = 'root'  # 数据库用户名
+DB_PASSWORD = '123456'  # 数据库密码
+DB_NAME = 'file'  # 数据库名
 
-def create_connection(host, user, password, db):
-    connection = None
-    try:
-        connection = pymysql.connect(
-            host=host,
-            user=user,
-            password=password,
-            db=db,
-            charset='utf8mb4',
-            cursorclass=pymysql.cursors.DictCursor
-        )
-        print("Connection to MySQL DB successful")
-    except pymysql.MySQLError as e:
-        print(f"The error '{e}' occurred")
-
-    return connection
-
-
-def upload_file_to_db(connection, file_path):
-    if not os.path.exists(file_path):
-        print("File does not exist.")
-        return
+# 创建表的SQL语句
+CREATE_TABLE_SQL = """
+CREATE TABLE IF NOT EXISTS images (
+    id INT AUTO_INCREMENT PRIMARY KEY,
+    filename VARCHAR(255),
+    content LONGBLOB,
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+);
+"""
 
+def create_table():
+    connection = pymysql.connect(host=DB_HOST, user=DB_USER, password=DB_PASSWORD, db=DB_NAME)
     try:
-        with open(file_path, 'rb') as file:
-            # 读取文件二进制内容
-            file_content = file.read()
-
-            cursor = connection.cursor()
-            sql_query = """
-                INSERT INTO file_storage (file_name, file_content)
-                VALUES (%s, %s)
-            """
-            # 获取文件名
-            file_name = os.path.basename(file_path)
-            # 执行SQL语句
-            cursor.execute(sql_query, (file_name, pymysql.Binary(file_content)))
-            connection.commit()
-            print(f"File {file_name} has been uploaded successfully.")
-    except pymysql.MySQLError as e:
-        print(f"Error uploading file: {e}")
+        with connection.cursor() as cursor:
+            cursor.execute(CREATE_TABLE_SQL)
+        connection.commit()
+        print("Table 'images' created successfully.")
+    finally:
+        connection.close()
+def path_to_bytes():
+    return filedialog.askopenfilename(filetypes=(("Image Files", "*.jpg;*.png;*.gif"), ("All Files", "*.*")))
+def upload_image():
+    selected_image_path = path_to_bytes()
+    if selected_image_path:
+        with open(selected_image_path, 'rb') as image_file:
+            image_content = image_file.read()
 
+        connection = pymysql.connect(host=DB_HOST, user=DB_USER, password=DB_PASSWORD, db=DB_NAME)
+        try:
+            with connection.cursor() as cursor:
+                sql = "INSERT INTO images (filename, content) VALUES (%s, %s)"
+                cursor.execute(sql, (os.path.basename(selected_image_path), pymysql.Binary(image_content)))
+                connection.commit()
+                print(f"Image '{selected_image_path}' uploaded successfully.")
+        finally:
+            connection.close()
 
-# 数据库连接配置
-host = 'localhost'
-database = 'file'
-user = 'root'
-password = '123456'
+def delete_image():
+    # 获取用户输入的图像ID
+    image_id = tk.simpledialog.askinteger("删除图片", "请输入需要删除的图像ID:")
+    if image_id is not None:
+        connection = pymysql.connect(host=DB_HOST, user=DB_USER, password=DB_PASSWORD, db=DB_NAME)
+        try:
+            with connection.cursor() as cursor:
+                # 查询图像信息
+                cursor.execute("SELECT * FROM images WHERE id = %s", (image_id,))
+                image_data = cursor.fetchone()
+                if image_data:
+                    # 输出图像信息
+                    print(f"Image Details:")
+                    for index, value in enumerate(image_data):
+                        column_name = cursor.description[index][0]
+                        print(f"{column_name}: {value}")
+                    # 提示用户确认删除
+                    confirmation = tk.messagebox.askyesno("Confirm Delete", f"是否删除图片ID为 {image_id}的图片?")
+                    if confirmation:
+                        # 执行删除操作
+                        sql = "DELETE FROM images WHERE id = %s"
+                        cursor.execute(sql, (image_id,))
+                        connection.commit()
+                        print(f"成功删除图片ID为 {image_id} 的图片.")
+                else:
+                    print(f"没有发现图片ID为 {image_id}的图片.")
+        except Exception as e:
+            print(f"Error deleting image: {e}")
+        finally:
+            connection.close()
 
-# 文件路径
-file_path = 'file_paths.txt'
+def download_image():
+    # 获取用户输入的图像ID
+    image_id = tk.simpledialog.askinteger("下载图片", "请输入需要下载的图像ID:")
+    if image_id is not None:
+        connection = pymysql.connect(host=DB_HOST, user=DB_USER, password=DB_PASSWORD, db=DB_NAME)
+        try:
+            with connection.cursor() as cursor:
+                # 查询图像内容和文件名
+                cursor.execute("SELECT content, filename FROM images WHERE id = %s", (image_id,))
+                image_data = cursor.fetchone()
+                if image_data:
+                    content, original_filename = image_data
+                    # 构建默认的保存文件名
+                    default_filename = os.path.splitext(original_filename)[0] + ".jpg"
+                    # 使用默认扩展名并允许用户选择保存位置
+                    save_path = filedialog.asksaveasfilename(initialfile=default_filename, defaultextension=".jpg", filetypes=[("JPEG Images", "*.jpg"), ("All Files", "*.*")])
+                    if save_path:
+                        try:
+                            with open(save_path, 'wb') as image_file:
+                                image_file.write(content)
+                            print(f"Image downloaded successfully to '{save_path}'.")
+                        except IOError as e:
+                            print(f"Error writing to file: {e}")
+                    else:
+                        print("Download cancelled.")
+                else:
+                    print(f"No image found with ID {image_id}.")
+        except Exception as e:
+            print(f"Error downloading image: {e}")
+        finally:
+            connection.close()
 
-# 创建数据库连接
-conn = create_connection(host, user, password, database)
 
-if conn is not None:
-    # 上传文件到数据库
-    upload_file_to_db(conn, file_path)
-else:
-    print("Failed to connect to the database.")
 
-# 关闭连接（如果打开）
-if conn:
-    conn.close()
 

filesys/upload.py

@@ -0,0 +1,18 @@
+from tkinter import filedialog
+import tkinter as tk
+from sql import *
+def upload_file_login():
+
+    new_window = tk.Tk()
+    new_window.title("欢迎来到上传界面")
+    new_window.geometry("600x200")
+
+    upload_file_button = tk.Button(new_window, text="上传图片", command=upload_image)
+    upload_file_button.pack()
+
+    back_button = tk.Button(new_window, text="返回", command=new_window.destroy)
+    back_button.pack()
+
+    new_window.mainloop()
+
+