rpi_cooler_python/cooler_control.py at master · thrubovc/rpi_cooler_python · 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
#!/usr/bin/env python3

from time import sleep
import os
import RPi.GPIO as GPIO

pin = 18
maxTemp = 65
minTemp = 45
flagPrevious = 'OFF'

GPIO.setwarnings(False)
GPIO.setmode(GPIO.BCM)
GPIO.setup(pin, GPIO.OUT)


def getCPUtemperature():
    res = os.popen('/opt/vc/bin/vcgencmd measure_temp').readline()
    temp = (res.replace("temp=", "").replace("'C\n", ""))
    return temp


def writeFile(state):
    f = open('/home/pi/cooler/fan_state', 'w')
    f.write("Current CPU temperature: " + str(getCPUtemperature()) + "'C\n")
    f.write('The fan is ' + state + '\n')
    f.close()


def setPin(mode):
    GPIO.output(pin, mode)
    return()


def fanON():
    setPin(True)
    writeFile('ON')
    return()


def fanOFF():
    setPin(False)
    writeFile('OFF')
    return()


def setFlag():
    cpuTemp = float(getCPUtemperature())
    if cpuTemp > maxTemp:
        flag = 'ON'
        return(flag)
    elif cpuTemp < minTemp:
        flag = 'OFF'
        return(flag)
    else:
        if flagPrevious == 'OFF':
            flag = 'OFF'
            return(flag)
        else:
            flag = 'ON'
            return(flag)
    return(flag)


def execute():
    flag = str(setFlag())
    if os.path.exists('/home/pi/cooler/on'):
        fanON()
    else:
        if flag == 'ON':
            fanON()
        else:
            fanOFF()
        global flagPrevious
        flagPrevious = flag
        return(flagPrevious)


while True:
    execute()
    sleep(5)