Advanced Lube pump timing and memory for oil mist spindles

loadrt oneshot count=1

#central lubrication
loadusr -W /home/cnc/linuxcnc/configs/haas_probebasic/lube.py
net lube-run lube.run => hm2_7i95.0.ssr.00.out-02 # your output for the lube pump
net spindle-cw   lube.spindle_status <= 

#!/usr/bin/env python3
import linuxcnc, hal, time, os, json
# File to store cycle state
STATE_FILE = "/home/cnc/linuxcnc/configs/haas_probebasic/lube_pump_state.json"  #location of the json file that will automatically be generated
# Initialize HAL component
lube = hal.component("lube")
lube.newpin("run", hal.HAL_BIT, hal.HAL_OUT)
lube.newpin("delay", hal.HAL_BIT, hal.HAL_OUT)
lube.newpin("spindle_status", hal.HAL_BIT, hal.HAL_IN)
lube.newpin("reset", hal.HAL_BIT, hal.HAL_IN)
lube.ready()
# Pump timing settings
PUMP_RUN_DURATION = 10  # Seconds HOW LONG THE PUMP WILL RUN
REST_DURATION = 1800  # 30 minutes (in seconds) HOW LONG THE PUMP CYCLE TIME 
# Initialize variables
pump_running = False
resting = False
elapsed_rest_time = 0
remaining_rest_time = REST_DURATION
start_time = None
# Load previous state if exists
if os.path.exists(STATE_FILE):
    try:
        with open(STATE_FILE, "r") as f:
            state = json.load(f)
            elapsed_rest_time = state.get("elapsed_rest_time", 0)
            resting = state.get("resting", False)
            # Ensure remaining rest time continues exactly where it left off
            remaining_rest_time = max(0, REST_DURATION - elapsed_rest_time)
            if resting:
                print(f"Machine restarted. Pump remains in rest mode for {remaining_rest_time:.2f} seconds.")
            else:
                remaining_rest_time = REST_DURATION  # Reset if not previously resting
    except Exception as e:
        print(f"Error loading state: {e}")
print("Lube system initialized.")
try:
    while True:
        time.sleep(0.1)  # Reduce CPU usage
        # Check spindle status
        if lube["spindle_status"]:
            if not pump_running and not resting:
                pump_running = True
                lube["run"] = 1
                start_time = time.time()
                print("Pump running.")
            # Stop pump after run duration
            if pump_running and (time.time() - start_time >= PUMP_RUN_DURATION):
                pump_running = False
                resting = True
                lube["run"] = 0
                lube["delay"] = 1  # Indicate pump resting
                start_time = time.time()
                print("Pump resting.")
        # Handle rest period
        if resting:
            if lube["spindle_status"]:
                elapsed_rest_time += time.time() - start_time
                start_time = time.time()
                if elapsed_rest_time >= REST_DURATION:
                    resting = False
                    lube["delay"] = 0  # Rest complete
                    elapsed_rest_time = 0
                    print("Rest complete. Pump ready to run again.")
            else:
                start_time = time.time()  # Pause timer if spindle stops
        # Save state periodically
        with open(STATE_FILE, "w") as f:
            json.dump(
                {
                    "elapsed_rest_time": elapsed_rest_time,
                    "resting": resting,
                },
                f,
            )
        # Handle reset signal
        if lube["reset"]:
            print("Reset signal received.")
            resting = False
            elapsed_rest_time = 0
            lube["reset"] = 0  # Reset the reset signal
except KeyboardInterrupt:
    print("Shutting down lube system.")
    lube["run"] = 0
    raise SystemExit