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
|
#!/usr/bin/python
'''
Program reads from stdin and color asked substrings.
Example usage:
make all 2>&1 | colorize.py green CC red warning
cat /etc/fstab | colorize.py red defaults
Author: Aapo Rantalainen
License: GPL3+
Design:
If user gives keyword which is substring of another keyword
e.g Input contains 123456
* '2345' and '34'
* '345' and '34'
-> '34' triggers
(because when '4' is read '34' is only completed keyword)
*'34' and '234'.
-> 234 triggers, because it is lengthiest keyword
*If you are looking for many words (spacebar between), use:
colorize.py red 'file system'
'''
import sys
#easy words for colors
colorcodes = {
'black' : '\033[30m',
'red' : '\033[31m',
'green' : '\033[32m',
'yellow' : '\033[33m',
'blue' : '\033[34m',
'magenta' : '\033[35m',
'cyan' : '\033[36m',
'white' : '\033[37m' }
#Reset all attributes
ENDC = '\033[0m'
#empty dictionary:
colorbook = {}
#usage: print colors
if len(sys.argv) >= 2:
if sys.argv[1] == '--help':
print "Usage: "+sys.argv[0]+" color keyword color keyword"
print "\nColors in use:"
for k, v in colorcodes.iteritems():
print v+k+ENDC
sys.exit("")
#There must be even number of parameters
if len(sys.argv)%2 == 0:
sys.exit(sys.argv[0]+": Must provide even numbers of parameters")
#first parameter is the name of application.
for i in range(1,len(sys.argv),2):
color = sys.argv[i]
if color in colorcodes:
keyword = sys.argv[i+1]
asked_color = colorcodes[color]
colorbook[keyword]= asked_color
else:
sys.exit(sys.argv[0]+": Can't understand color: "+color)
#Read from STDIN (read one line, process it and read next)
for line in iter(sys.stdin.readline, ""):
start=0
#scroll
for end in range(0,len(line)):
current_word = line[start:end]
candidate_word = [] #empty list
for keyword, v in colorbook.iteritems():
if keyword in current_word:
candidate_word.append(keyword)
#use lengthiest keyword, if any
if len(candidate_word) is not 0:
max_length=0
selected_word=""
for ii in candidate_word:
if len(ii) > max_length:
selected_word=ii
max_length = len(ii)
#selected_word=candidate_word[0]
color = colorbook[selected_word]
#print "color", color
current_word = current_word.replace(selected_word, color+selected_word+ENDC)
#print current_word,
sys.stdout.write(current_word)
start=end
current_word="" #because if we just read last character from the row, it will be printed again
print current_word
|
