Python下非阻塞式读取键盘输入

Python下非阻塞式读取键盘输入

在Python编程中,有时需要实时地读取键盘输入,但是正常的输入方式会阻塞程序执行,导致不能及时响应。这时可以使用一些技巧实现非阻塞式获取键盘输入。

本文将介绍在Python中如何通过tty和termios模块实现非阻塞读取键盘输入。主要分为以下几个步骤:

导入模块

导入所需的模块,包括tty、termios和select。

• tty模块可以设置终端模式
• termios可以保存和恢复终端属性
• select实现输入监听

import tty
import termios
import select
1
2
3

设置终端为原始模式

调用tty.setraw()将终端设置为原始模式,这样就可以不用按回车,直接读取键盘输入。

tty.setraw(sys.stdin.fileno())
1

监听输入

使用select.select()来监听标准输入stdin,并设置超时时间,例如0.1秒。

rlist, _, _ = select.select([sys.stdin], [], [], 0.1)
1

读取输入

在超时时间内如果有键盘输入,则读取一个字符并保存到变量key中;如果超时没有输入,key设置为空字符串。

if rlist:
    key = sys.stdin.read(1)
else:
    key = ''
1
2
3
4

恢复终端属性

最后使用termios.tcsetattr()将终端设置恢复原先的属性,避免影响后续的输入输出。

termios.tcsetattr(sys.stdin, termios.TCSADRAIN, settings)
1

返回结果

返回key变量,调用函数的地方就可以得到键盘输入或者超时的提示。

通过上述步骤,就可以定期检查键盘输入,如果有输入则返回,没有则不会阻塞程序执行。这样我们可以方便地实现游戏、控制界面等对键盘输入有实时响应需求的应用。

总之,使用tty、termios和select模块可以在Python中轻松实现非阻塞式读取键盘输入,构建出实时响应的交互程序。

全部代码

import tty #导入tty模块,用于设置终端模式
import sys #导入系统模块
import termios #导入termios模块,用于保存和恢复终端属性
import select #导入select模块,用于实现输入监听

settings = termios.tcgetattr(sys.stdin) #获取标准输入终端属性

def getKey():
    
    tty.setraw(sys.stdin.fileno()) #设置终端为原始模式
    
    rlist, _, _ = select.select([sys.stdin], [], [], 0.1) 
    #监听标准输入,超时0.1秒  
    
    if rlist: #若监听到输入
        
        key = sys.stdin.read(1) 
        #读取一个字符
        
    else:
        
        key = '' #否则设置为空字符
        
    termios.tcsetattr(sys.stdin, termios.TCSADRAIN, settings) 
    #恢复标准输入终端属性
    
    return key #返回键盘输入内容
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

键盘控制小车

代码

#!/usr/bin/env python

# Copyright (c) 2011, Willow Garage, Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#    * Redistributions of source code must retain the above copyright
#      notice, this list of conditions and the following disclaimer.
#    * Redistributions in binary form must reproduce the above copyright
#      notice, this list of conditions and the following disclaimer in the
#      documentation and/or other materials provided with the distribution.
#    * Neither the name of the Willow Garage, Inc. nor the names of its
#      contributors may be used to endorse or promote products derived from
#       this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.

import rospy

from geometry_msgs.msg import Twist

import sys, select, termios, tty

msg = """
Control Your Turtlebot!
---------------------------
Moving around:
   u    i    o  [ ]
   j    k    l
   m    ,    .

q/z : increase/decrease max speeds by 10%
w/x : increase/decrease only linear speed by 10%
e/c : increase/decrease only angular speed by 10%
space key, k : force stop
anything else : stop smoothly

CTRL-C to quit
"""

moveBindings = {
        'i':(1,0,0),
        'o':(1,-1,0),
        'j':(0,1,0),
        'l':(0,-1,0),
        'u':(1,1,0),
        ',':(-1,0,0),
        '.':(-1,1,0),
        'm':(-1,-1,0),
        '[':(0,0,1),
        ']':(0,0,-1),
           }

speedBindings={
        'q':(1.1,1.1),
        'z':(.9,.9),
        'w':(1.1,1),
        'x':(.9,1),
        'e':(1,1.1),
        'c':(1,.9),
          }

def getKey():
    tty.setraw(sys.stdin.fileno())
    rlist, _, _ = select.select([sys.stdin], [], [], 0.1)
    if rlist:
        key = sys.stdin.read(1)
    else:
        key = ''

    termios.tcsetattr(sys.stdin, termios.TCSADRAIN, settings)
    return key

speed = .2
speed_y = .2
turn = 1

def vels(speed,turn):
    return "currently:\tspeed %s\tturn %s " % (speed,turn)

if __name__=="__main__":
    settings = termios.tcgetattr(sys.stdin)

    rospy.init_node('robint_teleop')
    pub = rospy.Publisher('/cmd_vel/joy_pub', Twist, queue_size=1)
    # pub_gazebo = rospy.Publisher('cmd_vel', Twist, queue_size=5)
    rate = rospy.Rate(50) # 10hz

    x = 0
    y = 0
    th = 0
    status = 0
    count = 0
    acc = 0.1
    target_speed = 0
    target_speed_y = 0
    target_turn = 0
    control_speed = 0
    control_speed_y = 0
    control_turn = 0
    try:
        print(msg)
        print(vels(speed,turn))
        while(1):
            key = getKey()
            if key in moveBindings.keys():
                x = moveBindings[key][0]
                y = moveBindings[key][2]
                th = moveBindings[key][1]
                count = 0
            elif key in speedBindings.keys():
                speed = speed * speedBindings[key][0]
                speed_y = speed_y * speedBindings[key][0]
                turn = turn * speedBindings[key][1]
                count = 0

                print(vels(speed,turn))
                if (status == 14):
                    print(msg)
                status = (status + 1) % 15
            elif key == ' ' or key == 'k' :
                x = 0
                y = 0
                th = 0
                control_speed = 0
                control_speed_y = 0
                control_turn = 0
            else:
                count = count + 1
                if count > 4:
                    x = 0
                    y = 0
                    th = 0
                if (key == '\x03'):
                    break

            target_speed = speed * x
            target_speed_y = speed_y * y
            target_turn = turn * th

            if target_speed > control_speed:
                control_speed = min( target_speed, control_speed + 0.02 )
            elif target_speed < control_speed:
                control_speed = max( target_speed, control_speed - 0.02 )
            else:
                control_speed = target_speed

            if target_speed_y > control_speed_y:
                control_speed_y = min( target_speed_y, control_speed_y + 0.02 )
            elif target_speed_y < control_speed_y:
                control_speed_y = max( target_speed_y, control_speed_y - 0.02 )
            else:
                control_speed_y = target_speed_y

            if target_turn > control_turn:
                control_turn = min( target_turn, control_turn + 0.1 )
            elif target_turn < control_turn:
                control_turn = max( target_turn, control_turn - 0.1 )
            else:
                control_turn = target_turn

            twist = Twist()
            twist.linear.x = control_speed; twist.linear.y = control_speed_y; twist.linear.z = 0
            twist.angular.x = 0; twist.angular.y = 0; twist.angular.z = control_turn
            pub.publish(twist)
            # rate.sleep()
            # pub_gazebo.publish(twist)

            #print("loop: {0}".format(count))
            #print("target: vx: {0}, wz: {1}".format(target_speed, target_turn))
            #print("publihsed: vx: {0}, wz: {1}".format(twist.linear.x, twist.angular.z))

    except Exception as e:
        print(e)

    finally:
        twist = Twist()
        twist.linear.x = 0; twist.linear.y = 0; twist.linear.z = 0
        twist.angular.x = 0; twist.angular.y = 0; twist.angular.z = 0
        pub.publish(twist)
        # pub_gazebo.publish(twist)

    termios.tcsetattr(sys.stdin, termios.TCSADRAIN, settings)


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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197