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
|
#!/usr/bin/python2
#client for msg_daemon
import socket
import sys
import argparse
import pickle
import time
# Create a UDS socket
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
# Connect the socket to the port where the server is listening
server_address = '/run/msg_socket'
def send_message(message):
#print >>sys.stderr, 'connecting to %s' % server_address
try:
sock.connect(server_address)
except socket.error, msg:
print >>sys.stderr, msg
sys.exit(1)
#try:
sock.sendall(message)
#amount_received = 0
#amount_expected = len(message)
print "Waiting for response..."
data = sock.recv(1024)
print data
#finally:
#print >>sys.stderr, 'error occured closing socket'
sock.close()
def usage():
line = '''
Help screen:
msg_client.py is used to add,remove, or list items in the queue of messages that
will be displayed on screen. Optionaly messages can be given an identifier or "tag".
This identifier does not have to be unique.
Items are processed in the order they arrive, based off their slot number.
The lowest slot number will be displayed first.
After the message is displayed it is removed from the queue.
ADD:
To add items to the queue:
msg_client.py --msg "My first message" --tag "tag1"
REMOVE:
Removing items are based off the tag.
To remove all items from the queue that match the tag "tag1":
msg_client.py --clear --tag tag1
To remove all items from the queue:
msg_client.py --clear
Listing items in the queue or to get the total count:
msg_client.py --print_list
> -------------
> msg : slot 1
> slot : 1
> cmd : msg
> tag : None
> timeout : None
To immediately stop displaying the current message, use the kill command.
msg_client.py --kill
'''
print line
sys.exit(0)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('--tag', action='store', dest='tag',help='message tag')
parser.add_argument('--timeout', action='store', dest='timeout',help='time to display message')
action = parser.add_mutually_exclusive_group(required=True)
action.add_argument('--msg', action='store', dest='msg', help='Add message to the queue')
action.add_argument('--clear', action='store_true', help='Remove message from queue that match tag')
action.add_argument('--kill', action='store_true', help='Kill current msg thats displayed')
action.add_argument('--print_list', action='store_true', help='Print current queue')
action.add_argument('--usage', action='store_true', help='Print help screen')
results = parser.parse_args()
resultsdict = vars(results)
if results.msg :
cmd = "msg"
elif results.clear:
cmd = "clear"
elif results.kill:
cmd = "kill"
elif results.print_list:
cmd = "list"
elif results.usage:
usage()
arg_dict = {'cmd':cmd , 'msg':resultsdict['msg'] , 'tag':resultsdict['tag'] , 'timeout':resultsdict['timeout']}
send_message(pickle.dumps(arg_dict))
#sys.exit(0)
|
