writing_manager.py

import tkinter as tk
import os
import configparser
import csv
from datetime import datetime
from pathlib import Path

# Read the root directory from the config file
config = configparser.ConfigParser()
config_dir = Path.home() / Path('.writing_manager')
config_file = config_dir / Path('config.ini')
print(f"Config file path: {config_file}")
if not config_file.exists():
    # Create the config directory if it doesn't exist
    config_file.parent.mkdir(parents=True, exist_ok=True)
    # Create a default config file
    with open(config_file, 'w') as f:
        f.write("[files]\n")
        # TODO: project_dir needs to be selected by the user
        f.write("project_dir = /path/to/your/files\n")
        # f.write("file_order_csv = /path/to/your/files/file_order.csv\n")
config.read(str(config_file))

project_dir = config.get('files', 'project_dir')
# file_order_csv = config.get('files', 'file_order_csv')
file_order_csv = config_dir / 'file_order.csv'

def read_file_order_from_csv(file_order_csv):
    with open(file_order_csv, 'r') as csvfile:
        reader = csv.reader(csvfile)
        data = list(reader)
        last_row = data[-1]
        file_order = last_row[1].split('|')

        return {file_order[i]: i for i in range(len(file_order))}


# file_order = read_file_order_from_csv(file_order_csv)

class DragDropListbox(tk.Listbox):
    def __init__(self, master, **kwargs):
        super().__init__(master, **kwargs)
        # self.bind("<Button-1>", self.start_drag)
        # self.bind("<B1-Motion>", self.do_drag)
        # self.bind("<ButtonRelease-1>", self.stop_drag)
        self.drag_data = {"index": None}

    def list_files_numerically(self, file_order):
        self.delete(0, tk.END)
        for file, i in file_order.items():
            self.insert(tk.END, f"{i+1}: {file}")

    def get_filename_without_number(self, list_item):
        return list_item.split(": ", 1)[1] if ": " in list_item else list_item

    def get_ordered_file_list(self):
        items = list(self.get(0, tk.END))
        filenames = [self.get_filename_without_number(item) for item in items]
        return filenames

    def start_drag(self, event):
        # Get the index of the clicked item
        self.drag_data["index"] = self.nearest(event.y)

    def do_drag(self, event):
        # Highlight the item being dragged
        self.selection_clear(0, tk.END)
        self.selection_set(self.drag_data["index"])

    def stop_drag(self, event):
        # Get the target index where the item is dropped
        target_index = self.nearest(event.y)
        source_index = self.drag_data["index"]

        if source_index != target_index:
            # Rearrange items in the listbox
            items = list(self.get(0, tk.END))
            item = items.pop(source_index)
            items.insert(target_index, item)

            # Update the listbox with the new order
            self.delete(0, tk.END)
            for i in items:
                self.insert(tk.END, i)

        # Clear drag data
        self.drag_data["index"] = None

class FileListApp:
    def __init__(self, root):
        self.root = root
        self.root.title("File List App")
        self.file_list = DragDropListbox(self.root, activestyle="dotbox")
        self.file_list.pack(padx=10, pady=10, 
                            # ipadx=10, ipady=10,
                            )

        # Populate the listbox with files in the desired order
        # self.file_list.list_files_numerically(read_file_order_from_csv(file_order_csv))
        self.list_files_in_order()
        # for file in file_order:
        #     self.file_list.insert(tk.END, file)

        # Bind double-click event to open the file
        self.file_list.bind('<Double-1>', self.open_file)

        # Bind Enter key press to open the selected file
        self.file_list.bind('<Return>', self.open_selected_file)

        # Create reorder button
        self.reorder_button = tk.Button(self.root, text="Reorder", command=self.toggle_reorder)
        self.reorder_button.pack(padx=10, pady=10)

        # Create reorder cancel button
        self.cancel_button = tk.Button(self.root, text="Cancel", command=self.cancel_reorder)
        self.cancel_button.pack_forget()

    def open_file(self, event):
        # Get the selected file
        selected_index = self.file_list.nearest(event.y)
        selected_file = self.file_list.get(selected_index)
        # Construct the full path to the file
        file_path = os.path.join(project_dir, selected_file)
        # Open the file using the default application
        os.startfile(file_path)

    def open_selected_file(self, event):
        # Get the selected file
        selected_index = self.file_list.curselection()
        if selected_index:
            selected_file = self.file_list.get(selected_index)
            # Construct the full path to the file
            file_path = os.path.join(project_dir, selected_file)
            # Open the file using the default application
            os.startfile(file_path)

    def move_up(self, event):
        # Get the selected file
        selected_index = self.file_list.curselection()
        if selected_index and selected_index[0] > 0:
            # Move the file up
            items = list(self.file_list.get(0, tk.END))
            item = items.pop(selected_index[0])
            items.insert(selected_index[0] - 1, item)
            self.file_list.delete(0, tk.END)
            for i in items:
                self.file_list.insert(tk.END, i)
            self.file_list.selection_clear(0, tk.END)
            new_index = selected_index[0] - 1
            self.file_list.activate(selected_index[0])
            self.file_list.see(new_index)

    def move_down(self, event):
        # Get the selected file
        selected_index = self.file_list.curselection()
        if selected_index and selected_index[0] < self.file_list.size() - 1:
            # Move the file down
            items = list(self.file_list.get(0, tk.END))
            item = items.pop(selected_index[0])
            items.insert(selected_index[0] + 1, item)
            self.file_list.delete(0, tk.END)
            for i in items:
                self.file_list.insert(tk.END, i)
            self.file_list.selection_clear(0, tk.END)
            new_index = selected_index[0] + 1
            self.file_list.selection_set(new_index)
            self.file_list.activate(selected_index[0])
            self.file_list.see(new_index)

    def enable_reordering(self):
        print("Enabling reordering")
        # allow dragging and dropping to reorder files
        self.file_list.bind("<Button-1>", self.file_list.start_drag)
        self.file_list.bind("<B1-Motion>", self.file_list.do_drag)
        self.file_list.bind("<ButtonRelease-1>", self.file_list.stop_drag)
        self.file_list.bind("<Alt-Up>", self.move_up)
        self.file_list.bind("<Alt-Down>", self.move_down)

    def save_new_order(self):
        print("Saving file order")

        # Get the current order of files
        filenames = self.file_list.get_ordered_file_list()
        # Write the order at the end of the CSV file
        with open(file_order_csv, 'a') as out:
            # get date in the format YYYY-MM-DDTH:MM:SS
            date = datetime.now().strftime("%Y-%m-%dT%H:%M:%S")
            out.write(f"{date},{'|'.join(filenames)}\n")

        # relist files with numbers corresponding to the new order
        self.list_files_in_order()

    def cancel_reorder(self):
        self.reorder_button['text'] = "Reorder"
        # self.reorder_button['command'] = self.toggle_reorder
        self.file_list.unbind("<Button-1>")
        self.file_list.unbind("<B1-Motion>")
        self.file_list.unbind("<ButtonRelease-1>")
        self.file_list.unbind('<Alt-Up>')
        self.file_list.unbind('<Alt-Down>')
        self.cancel_button.pack_forget()

        # Readjust to show the current order of files
        # file_order = read_file_order_from_csv(file_order_csv)
        # self.file_list.delete(0, tk.END)
        # self.file_list.list_files_numerically(file_order)
        self.list_files_in_order()
        # items = list(self.file_list.get(0, tk.END))
        # # Sort the items based on the predefined order
        # sorted_items = sorted(items, key=lambda x: file_order.get(x, float('inf')))
        # # Update the listbox with the new order
        # for item in sorted_items:
        #     self.file_list.insert(tk.END, item)

    def list_files_in_order(self):
        file_order = read_file_order_from_csv(file_order_csv)
        self.file_list.list_files_numerically(file_order)

    def toggle_reorder(self):
        if self.reorder_button['text'] == "Reorder":
            print("changing button to save")
            self.reorder_button['text'] = "Save"
            self.enable_reordering()
            self.cancel_button.pack(side=tk.LEFT, padx=10)
        else:
            print("changing button to reorder")
            self.reorder_button['text'] = "Reorder"
            self.save_new_order()
            print("removing cancel button")
            self.cancel_button.pack_forget()


if __name__ == "__main__":
    root = tk.Tk()
    app = FileListApp(root)
    root.mainloop()