util.py

from typing import List
from dataclasses import dataclass

ORIGINAL_GCODE_FILE = "./originalLayer.gcode"
NEW_FILE_SUFFIX = "LayerTest.gcode"
PRINT_START_SIGN = ";TYPE:Bottom surface\n"
PRINT_END_SIGN = ";WIPE_START\n"

@dataclass
class FileLines:
	startPos: int	
	startLines: List[str]

	endPos: int
	endLines: List[str]

	drawLines: List[str]

def getAllFileLines():
	with open(ORIGINAL_GCODE_FILE) as fin:
		gcodeLines = fin.readlines()

	startPos = gcodeLines.index(PRINT_START_SIGN)
	startLines = gcodeLines[0:startPos+1]

	endPos = gcodeLines.index(PRINT_END_SIGN)
	endLines = gcodeLines[endPos:]

	drawLines = gcodeLines[startPos+1:endPos]
	drawLines = [dl for dl in drawLines if dl.startswith('G1')]

	return FileLines(
		startPos,
		startLines,
		endPos,
		endLines,
		drawLines
	)

def writeFileLines(outputLines, fileNamePrefix):
	with open(fileNamePrefix + NEW_FILE_SUFFIX, 'w') as fout:
		fout.writelines(outputLines)