shell脚本实现股票查询

g84ch

    博客地址:http://blog.chinaunix.net/uid-18757596-id-3085105.html
    偶然在网上看到sina有一个查询股票信息的api，随即在linux下用shell写了个小脚本查询股票信息，方便工作之余关心一下股票行情，而且在linux终端里面查看，不容易被boss发现哦（我还故意把字体颜色调成了深蓝色，不仔细看，看不出来的哦，boss同事还以为你在认真工作写代码呢，嘿嘿）。小脚本分两个部分，checkstock.sh和stock_num.dat，把想要查询的股票代码（格式为：sh601699 或者 sz000001）添加到stock_num.dat中去，运行checkstock.sh即可查询。
    顺便总结一下，写脚本时候碰到的一些问题和解决方法,以备日后查阅，也望各位大牛指点新招妙招：
1. 浮点数的比较：
    两种方法：1. ds=`echo "1.23" "2.33" | awk '{if($1>=$2) {print 1;} else {print 0;}}'`
                               2. ds=`echo "$1 >= $2" | bc`
    因为test只能比较整数，浮点数的比较借用了bc和awk，通过上述两句，ds的值为0或者1，再做判断。

2. 数组作为参数传给函数：
    见函数prase_info()中的第一句， 如下：
     eval $(echo $1 | awk '{split($0,array,","; for(i in array) print "hs_info_array["i"]="array}')
通过这个语句，可以把参数$1(字符串数组)通过awk赋给shell数组hs_info_array 。

3. 传参数不改变字符数组的结构
    见函数print_array(),传值的时候用 print_array "${STOCK_NUM[@]}",这里用@，结构不变，用*，则数组中的字符串则会被拆分，比如，STOCK_NUM[1]="aaa   bbb" 如果传值时用*，则会被分成两个，一个是aaa，一个是bbb。而用@，则仍然是“aaa   bbb”

配置文件stock_num.dat如下：

1.     sh601699
   
2.     sh600600
   
3.     sz000002

复制代码

shell脚本checkstock.sh如下：

1.     #!/bin/bash
   
2. 
   
3.     ##################################################################################
   
4.     # Read stock number data from ./stock_num_file.txt
   
5.     #
   
6.     #
   
7.     #
   
8.     ###################################################################################
   
9.     STOCK_NUM_FILE="./stock_num.dat"
   
10.     ITEM_COLOR="\033[34m"
    
11.     COLOR_RESET="\033[0m"
    
12.     COLOR_RED="\033[31m"
    
13.     COLOR_GREEN="\033[32m"
    
14. 
    
15.     function ERRTRAP()
    
16.     {
    
17.             echo "[LINE:$1] Error: Comand or Function exited with status $?"
    
18. 
    
19.     }
    
20. 
    
21.     function print_array()
    
22.     {
    
23.             i=0
    
24.             for elem in "$@"
    
25.             do
    
26.                     echo $((i++)) $elem
    
27.             done
    
28.     }
    
29. 
    
30.     function read_data_file()
    
31.     {
    
32.             i=0
    
33.             if [ ! -e "./stock_num.dat" ]
    
34.             then
    
35.                     echo "ERROR: The stock_num.dat isn't existed!"
    
36.                     exit 0
    
37.             fi
    
38. 
    
39.             while read line
    
40.             do
    
41.                     STOCK_NUM[$i]=$line
    
42.                     let i+=1
    
43.             done < $STOCK_NUM_FILE
    
44.             for ((i=0; i < ${#STOCK_NUM[@]}; i++ ))
    
45.             do
    
46.             #echo "${STOCK_NUM[$i]}"
    
47.                     get_info ${STOCK_NUM[$i]}
    
48.             done
    
49.     }
    
50. 
    
51.     ############################################################################
    
52.     #
    
53.     # funciton: get_info(arg1)
    
54.     #
    
55.     # result: Get stock information from Internet
    
56.     #
    
57.     ############################################################################
    
58.     function get_info()
    
59.     {
    
60.             stock_info=`curl http://hq.sinajs.cn/list=$1 2>/dev/null | sed -n -e 's/var.*=\"//' -e 's/ //g' -e 's/\";$//p'`
    
61.     # echo "$stock_info"
    
62.     # s_name=`echo "$stock_info" | sed -n 's/,.*$//p'`
    
63.     # echo $s_name
    
64.     # stock_data=`echo "$stock_info" | sed -n 's/'$s_name',//p'`
    
65.     # echo $stock_data
    
66.             s_num=$1
    
67.             if [ "$stock_info" != "" ];then
    
68.                     prase_info $stock_info
    
69.                     display_info
    
70.             else
    
71.                     echo "Exception:Can't find any infomation about $s_num"
    
72.             fi
    
73.     }
    
74. 
    
75.     function prase_info()
    
76.     {
    
77.             eval $(echo $1 | awk '{split($0,array,","); for(i in array) print "stock_info_array["i"]="array[i]}')
    
78.             s_name=${stock_info_array[1]}
    
79.             s_today_open=${stock_info_array[2]}
    
80.             s_lastday_close=${stock_info_array[3]}
    
81.             s_current_price=${stock_info_array[4]}
    
82.             s_today_highest=${stock_info_array[5]}
    
83.             s_today_lowest=${stock_info_array[6]}
    
84.             s_dealed_stock_num=${stock_info_array[9]}
    
85.             s_dealed_stock_cash_num=${stock_info_array[10]}
    
86.             s_date=${stock_info_array[31]}
    
87.             s_time=${stock_info_array[32]}
    
88. 
    
89.             #####################################
    
90.             #${stock_info_array[11-20] buy1-buy5
    
91.             #${stock_info_array[21-30] sale1-sale5
    
92.             ######################################
    
93. 
    
94. 
    
95.     }
    
96. 
    
97.     ############################################################################
    
98.     #
    
99.     # funciton: print_pre_zero(arg1)
    
100.     #
     
101.     # result: add prefix zero for rate
     
102.     #
     
103.     ############################################################################
     
104.     function print_pre_zero()
     
105.     {
     
106.             local cmp_value_1=`echo "$1 < 0 && $1 > -1" | bc`
     
107.             local cmp_value_2=`echo "$1 >0 && $1 < 1" | bc`
     
108.             if [ $cmp_value_1 -eq "1" ]
     
109.             then
     
110.                     local tmp_val=`echo "scale=2;$1 / -1" | bc -l`
     
111.                     echo "-0$tmp_val"
     
112.             elif [ $cmp_value_2 -eq "1" ]
     
113.             then
     
114.                     echo "0$1"
     
115.             else
     
116.                     echo "$1"
     
117.             fi
     
118. 
     
119.     }
     
120. 
     
121.     ############################################################################
     
122.     #
     
123.     # funciton: cmp_display(arg1,arg2)
     
124.     #
     
125.     # result: print arg1 with color(red or green)
     
126.     #
     
127.     ############################################################################
     
128.     function cmp_display()
     
129.     {
     
130.     #############################################################################
     
131.     # Two methods to compare float:
     
132.     # 1. ds=`echo "0" "0" | awk '{if($1>=$2) {print 1;} else {print 0;}}'`
     
133.     # 2. ds=`echo "$1 >= $2" | bc`
     
134.     # echo $ds
     
135.     #############################################################################
     
136.             local cmp_value=`echo "$1 >= $2" | bc -l`
     
137.             if [ $cmp_value -eq "0" ]
     
138.             then
     
139.                     echo -e "$COLOR_GREEN $1 $COLOR_RESET"
     
140.             else
     
141.                     echo -e "$COLOR_RED $1 $COLOR_RESET"
     
142.             fi
     
143.     }
     
144. 
     
145. 
     
146.     ############################################################################
     
147.     #
     
148.     # funciton: cmp_display_rate(arg1,arg2)
     
149.     #
     
150.     # result: print arg1 with color(red or green) and add prefix zero.
     
151.     #
     
152.     ############################################################################
     
153.     function cmp_display_rate()
     
154.     {
     
155.             local tmp_rate=`echo "($1 - $2) / $2 * 100" | bc -l`
     
156.             local cmp_rate=`echo "scale=2;$tmp_rate/1" | bc`
     
157. 
     
158.             local cmp_value=`echo "$1 >= $2" | bc`
     
159. 
     
160.             if [ $cmp_value -eq "0" ]
     
161.             then
     
162.                     echo -e "$COLOR_GREEN $1 ($(print_pre_zero $(echo $1 - $2 | bc -l)) $(print_pre_zero $cmp_rate)%)$COLOR_RESET"
     
163.             else
     
164.                     echo -e "$COLOR_RED $1 (+$(print_pre_zero $(echo $1 - $2 | bc -l)) $(print_pre_zero $cmp_rate)%)$COLOR_RESET"
     
165.             fi
     
166. 
     
167. 
     
168.     }
     
169. 
     
170.     function display_info()
     
171.     {
     
172.             echo -e "$s_date"
     
173.             echo -e "$s_time"
     
174.             echo -e "$ITEM_COLOR STOCK: $s_name No. $s_num $COLOR_RESET"
     
175.             echo -e "$ITEM_COLOR Last Day Closed Price:$COLOR_RESET $s_lastday_close"
     
176.             echo -e "$ITEM_COLOR Today Open Price:$COLOR_RESET $(cmp_display $s_today_open $s_lastday_close)"
     
177.             echo -e "$ITEM_COLOR Curren Price:$COLOR_RESET $(cmp_display_rate $s_current_price $s_lastday_close)"
     
178.             echo -e "$ITEM_COLOR Highest Price:$COLOR_RESET $(cmp_display $s_today_highest $s_lastday_close)"
     
179.             echo -e "$ITEM_COLOR Lowest Price:$COLOR_RESET $(cmp_display $s_today_lowest $s_lastday_close)"
     
180.             echo -e "$ITEM_COLOR Dealed Stock Num(per 100):$COLOR_RESET $(expr $s_dealed_stock_num / 100)"
     
181.             echo -e "$ITEM_COLOR Dealed Stock Cash Num:$COLOR_RESET $s_dealed_stock_cash_num"
     
182.     }
     
183. 
     
184.     function display_hs_index()
     
185.     {
     
186.             eval $(echo $1 | awk '{split($0,array,","); for(i in array) print "hs_info_array["i"]="array[i]}')
     
187.             echo -e "$ITEM_COLOR Index:$COLOR_RESET ${hs_info_array[1]}$COLOR_RESET"
     
188.             echo -e "$ITEM_COLOR Today Open:$COLOR_RESET $(cmp_display ${hs_info_array[2]} ${hs_info_array[3]})"
     
189.             echo -e "$ITEM_COLOR Last CLosed:$COLOR_RESET ${hs_info_array[3]}"
     
190.             echo -e "$ITEM_COLOR Curren :$COLOR_RESET $(cmp_display_rate ${hs_info_array[4]} ${hs_info_array[3]})"
     
191.             echo -e "$ITEM_COLOR Highest:$COLOR_RESET $(cmp_display ${hs_info_array[5]} ${hs_info_array[3]})"
     
192.             echo -e "$ITEM_COLOR Lowest:$COLOR_RESET $(cmp_display ${hs_info_array[6]} ${hs_info_array[3]})"
     
193. 
     
194. 
     
195. 
     
196.     }
     
197. 
     
198.     function view_hs_index()
     
199.     {
     
200.             sh_info=`curl http://hq.sinajs.cn/list=sh000001 2>/dev/null | sed -n -e 's/var.*=\"//' -e 's/\";$//p'`
     
201.             if [ "$sh_info" != "" ];then
     
202.                     display_hs_index $sh_info
     
203.             else
     
204.                     echo "Exception:Can't find any infomation about $s_num"
     
205.             fi
     
206.             sz_info=`curl http://hq.sinajs.cn/list=sz399001 2>/dev/null | sed -n -e 's/var.*=\"//' -e 's/\";$//p'`
     
207.             if [ "$sz_info" != "" ];then
     
208.                     display_hs_index $sz_info
     
209.             else
     
210.                     echo "Exception:Can't find any infomation about $s_num"
     
211.             fi
     
212. 
     
213.     }
     
214. 
     
215.     view_hs_index
     
216.     read_data_file
     
217.     #trap "ERRTRAP $LINENO" ERR

复制代码

txnet

太帅了 学习了

blackold

学以致用，很好！

futurelinux

谢谢分享，膜拜！

rdcwayx

不错， 慢慢看。

这行：

1. stock_info=`curl http://hq.sinajs.cn/list=$1 2>/dev/null | sed -n -e 's/var.*=\"//' -e 's/ //g' -e 's/\";$//p'`

复制代码

是否直接可以：

1. stock_info=`curl http://hq.sinajs.cn/list=$1 2>/dev/null | cut -d \" -f2`

复制代码

或者

1. stock_info=`curl http://hq.sinajs.cn/list=$1 2>/dev/null | awk -F \" '{gsub(/ /,"",$2);print $2}' `

复制代码

g84ch

多谢版主指教，学习了。呵呵

回复 5# rdcwayx


   

zhaopingzi

$ ./checkstock.sh
Exception:Can't find any infomation about
Exception:Can't find any infomation about
Exception:Can't find any infomation about sh601699
Exception:Can't find any infomation about sh600600
Exception:Can't find any infomation about sz000002

jiejie455

回复 6# g84ch

或者直接 awk 'BEGIN{"curl http://hq.sinajs.cn/list='$1' 2>/dev/null"|getline;print $2}'


   

g84ch

回复 7# zhaopingzi


#./checkstock.sh
Index: 上证指数
Today Open:  2418.788
Last CLosed: 2428.487
Curren :  2424.853 (-3.634 -0.14%)
Highest:  2434.691
Lowest:  2418.334
Index: 深证成指
Today Open:  10011.119
Last CLosed: 10054.845
Curren :  10068.840 (+13.995 0.13%)
Highest:  10123.275
Lowest:  10011.119
2012-03-01
10:48:25
STOCK: 潞安环能 No. sh601699
Last Day Closed Price: 26.70
Today Open Price:  26.51
Curren Price:  27.12 (+0.42 1.57%)
Highest Price:  27.12
Lowest Price:  26.42
Dealed Stock Num(per 100): 47937
Dealed Stock Cash Num: 128807467
2012-03-01
10:48:28
STOCK: 柳工 No. sz000528
Last Day Closed Price: 14.20
Today Open Price:  14.15
Curren Price:  14.25 (+0.05 0.35%)
Highest Price:  14.35
Lowest Price:  14.14
Dealed Stock Num(per 100): 39039
Dealed Stock Cash Num: 55662717.84
2012-03-01
10:48:25
STOCK: 青岛啤酒 No. sh600600
Last Day Closed Price: 33.90
Today Open Price:  33.73
Curren Price:  33.93 (+0.03 0.08%)
Highest Price:  34.00
Lowest Price:  33.73
Dealed Stock Num(per 100): 3463
Dealed Stock Cash Num: 11707678
2012-03-01
10:48:28
STOCK: 万科Ａ No. sz000002
Last Day Closed Price: 8.28
Today Open Price:  8.24
Curren Price:  8.40 (+0.12 1.44%)
Highest Price:  8.47
Lowest Price:  8.22
Dealed Stock Num(per 100): 368564
Dealed Stock Cash Num: 309623305.93

g84ch

回复 8# jiejie455

貌似不行啊

1. awk 'BEGIN{"curl http://hq.sinajs.cn/list=sh601699 2>/dev/null"|getline;print $2}'
   
2. hq_str_sh601699="潞安环能,26.51,26.70,26.99,27.13,26.42,26.98,26.99,5119704,137629495,100,26.98,100,26.97,14200,26.96,1600,26.94,200,26.93,100,26.99,13960,27.00,1900,27.01,4500,27.02,14200,27.03,2012-03-01,10:54:00";
   

复制代码