BOARDS/CARDPUTER CircuitPython/projects/text_editor.py

# GUIDE
# type freely! CTRL+D clears the text.
# CTRL + S to Save
# CTRL + O to Open
# DESIGNED TO WORK WITH SD CARD !!! 
# backspace, space, and return function as expected.

# keys not recognised WILL print "UNKNOWN 0x??",
# unless PRINT_UNKNOWN_KEYS is set to False
# UP, DOWN, LEFT, RIGHT will print,
# unless PRINT_COMMAND_KEYS is set to False

# TODO:
# UP and DOWN scrolling
# look into folders in SD card
######

import board
import terminalio
from adafruit_display_text import label
from keyboard import Keyboard

import busio
import sdcardio
import storage
import os

def file_menu():
    files_text = ""
    index = 0
    for file in os.listdir("/sd"):
        filename = str(file)
        if ".txt" in filename:
            if index == open_cursor:
                filename = "> " + filename
            files_text = files_text + filename+"\n"
            index += 1
    files_group.text = files_text

## SD Card set up for saving text to SD card
spi = busio.SPI(board.SD_SCK, MOSI=board.SD_MOSI, MISO=board.SD_MISO)
cs = board.SD_CS

try:
    sdcard = sdcardio.SDCard(spi, cs)
    vfs = storage.VfsFat(sdcard)

    storage.mount(vfs, "/sd", readonly=False) # access files on sd card here
    SD_status = "SD Card Mounted"
except OSError:
    SD_status = "No SD card found"
    pass # SD card not inserted/found
#########################################

keyb = Keyboard()

PRINT_UNKNOWN_KEYS = False
PRINT_COMMAND_KEYS = False
COMMAND_KEYS = ["UP", "DOWN", "LEFT", "RIGHT"]

page = "editor"

intro_text = "Press CTRL+D to clear text\n"+SD_status+"\n\n"
intro_text_area = label.Label(terminalio.FONT, text=intro_text, color="C3B")
intro_text_area.x = 10
intro_text_area.y = 10

text = ""
text_area = label.Label(terminalio.FONT, text=text)
text_area.x = 0
text_area.y = 30

board.DISPLAY.root_group = intro_text_area
intro_text_area.append(text_area)

filename_text = ""
save_text = "Save File - Enter Filename\nOverwriting Enabled"
save_group = label.Label(terminalio.FONT, text=save_text, color="C3B")
save_group.x = 10
save_group.y = 10
filename_group = label.Label(terminalio.FONT, text=filename_text)
filename_group.x = 0
filename_group.y = 60
save_group.append(filename_group)

files_text = ""
open_text = "Open File - Select using arrow keys"
open_group = label.Label(terminalio.FONT, text=open_text, color="C3B")
open_group.x = 10
open_group.y = 10
files_group = label.Label(terminalio.FONT, text=files_text)
files_group.x = 0
files_group.y = 30
open_group.append(files_group)

open_cursor = 0

while True:
    key = keyb.scan()

    if key == "BACKSP":
        if page == "editor":
            text = text[:-1]
            text_area.text = text
        elif page == "save":
            filename_text = filename_text[:-1]
            filename_group.text = filename_text
        
    elif key == "CTRLD":
        if page == "editor":
            text = ""
            text_area.text = text
        elif page == "save":
            filename_text = ""
            filename_group.text = filename_text
        
    elif key == "CTRLS":
        board.DISPLAY.root_group = save_group
        page = "save"
        filename_group.text=filename_text
        
    elif key == "CTRLO":
        board.DISPLAY.root_group = open_group
        page="open"
        file_menu()
        
    elif key == "ESC":
        if page == "editor":
            pass
        elif page == "save" or page == "open":
            board.DISPLAY.root_group = intro_text_area
        page = "editor"

        
    else:
        ## The actual editor
        if page == "editor":
            if "UNKNOWN" in key and PRINT_UNKNOWN_KEYS == False:
                pass
            elif key in COMMAND_KEYS and PRINT_COMMAND_KEYS == False:
                pass
            else:
                ## WRAPPING
                ## calculate wrapping when entering a character 
                lines = text.split("\n")
                current_line = lines[len(lines)-1]
                if len(current_line) > 35:
                    text=text+"\n" # <- start a new line here
                ##############################################
                text = text+key
                text_area.text = text

        ## The save file menu - for typing the filename
        elif page == "save":
            if key == "\n":
                with open("/sd/"+filename_text+".txt", "w") as f:
                    f.write(text)
                    f.close()
                board.DISPLAY.root_group = intro_text_area
                page = "editor"
            elif not keyb.is_alphanumeric(key):
                pass # dont allow special characters to be entered
            else:
                filename_text = filename_text+key
                filename_group.text=filename_text

        ## The open file menu - for selecting a file
        elif page == "open":
            if key == "UP":
                if open_cursor == 0:
                    pass
                else:
                    open_cursor -= 1
                    file_menu()
            elif key == "DOWN":
                if open_cursor == len(os.listdir("/sd"))-1:
                    pass
                else:
                    open_cursor += 1
                    file_menu()
            elif key == "\n":
                text = ""
                fn = os.listdir("/sd")[open_cursor]
                with open("/sd/"+fn, "r") as f:
                    lines = f.readlines()
                for line in lines:
                    text = text + line
                text_area.text = text
                filename_text = fn.split(".txt")[0]
                filename_group.text=filename_text
                page = "editor"
                board.DISPLAY.root_group = intro_text_area