blending-control/menu.py at master · jcbachmann/blending-control · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#!/usr/bin/env python

import subprocess
import os
import stacker
import dump_feeder
import reclaimer
import drive
import shaker
import time

run = True
lastStockpileName = "unknown"
options = [
#--- stockpile
["stack", lambda: stackFile()],
["dump feeder material", lambda: dump_feeder.dump1()],
["flush feeder material", lambda: dump_feeder.dump2()],
["reclaim", lambda: reclaimer.reclaim("../data/" + lastStockpileName)],
#--- feeder
["control feeder", "./shakekey.sh"],
["stop feeder", lambda: shaker.stop()],
#--- drive
["flush stockpile",  lambda: drive.flush()],
["drive to start pos", lambda: drive.moveToStockpile(0)],
["drive to end pos", lambda: drive.moveToStockpile(1)],
["drive to dump pos", lambda: drive.moveToMM(-400)],
["drive left", lambda: drive.run(-1000)],
["drive right", lambda: drive.run(1000)],
["drive stop", lambda: drive.stop()],
#--- other
["colorcount", lambda: os.system("colorcount")],
["alarm", "./alarm.sh"],
["poweroff", lambda: os.system("poweroff")],
["reboot", lambda: os.system("reboot")],
["close", lambda: stop()]
]

def stop():
	global run
	run = False
	return 0

def stackFile():
	global lastStockpileName
	print("\nStockpiles:")
	directoryFiles = os.listdir("../data")
	directoryFiles = sorted(list(filter(lambda x: x.endswith(".stockpile"), directoryFiles)))
	stockpileNum = 0
	for file in directoryFiles:
		stockpileNum += 1
		print(str(stockpileNum) + " - " + file)
	print(str(stockpileNum + 1) + " - return")
	choiceIndex = int(input("Choose your stockpile...\n"))
	if(choiceIndex == stockpileNum + 1):
		return 0
	else:
		lastStockpileName = directoryFiles[choiceIndex - 1]
	print(lastStockpileName)
	stacker.stack("../data/" + lastStockpileName)
	return 0

def printTopic(topic):
	print("--- " + topic + " ---")
	return 0

def showMenu():
	#print list
	index = 0
	printTopic("stockpile")
	for option in options:
		index += 1
		if(index == 5): printTopic("feeder")
		if(index == 7): printTopic("drive")
		if(index == 14): printTopic("other")
		print(str(index) + " - " + option[0])
	print("\n")
	response = int(input("Choose your choice...\n")) - 1
	os.system("cowsay " + str(options[response][0]))

	#execute
	program = options[response][1]
	if(isinstance(program, str)):
		os.system(program)
	else:
		drive.poll()
		program()
	os.system("sync")

drive.init()
while(run):
	showMenu()
drive.cleanup()