Python牛刀小试(四)--代码解析(邮件发送功能)

1.工具类python

点击(此处)折叠或打开

1. # -*-coding=utf-8-*-
   
2. __author__ = 'zhangshengdong'
   
3. '''
   
4. 个人技术博客:http://blog.chinaunix.net/uid/26446098.html
   
5. 联系方式: sdzhang@cashq.ac.cn
   
6. '''
   
7. class Toolkit():                          #工具类，后面会调用，用于分割例3的配置文件。
   
8. 
   
9.     @staticmethod
   
10.     def getUserData(cfg_file):
    
11.         f=open(cfg_file,'r')
    
12.         account={}
    
13.         for i in f.readlines():
    
14.             ctype,passwd=i.split('=')
    
15.             #print ctype
    
16.             #print passwd
    
17.             account[ctype.strip()]=passwd.strip()
    
18.         return account

2.价格比较策略代码

点击(此处)折叠或打开

1. #-*-coding=utf-8-*-
   
2. __author__ = 'zhangshengdong'
   
3. '''
   
4. 个人技术博客:http://blog.chinaunix.net/uid/26446098.html
   
5. 联系方式: sdzhang@cashq.ac.cn
   
6. '''
   
7. import os
   
8. import sys
   
9. import smtplib
   
10. import tushare as ts
    
11. from Ztoolkit import Toolkit as TK
    
12. from email.mime.text import MIMEText
    
13. from email import Encoders, Utils
    
14. 
    
15. reload(sys)
    
16. sys.setdefaultencoding('utf8')
    
17. 
    
18. class MailSend():
    
19.     def __init__(self, smtp_server, from_mail, password, to_mail):   #邮箱初始化配置的函数
    
20.         self.server = smtp_server
    
21.         self.username = from_mail.split("@")[0]
    
22.         self.from_mail = from_mail
    
23.         self.password = password
    
24.         self.to_mail = to_mail
    
25. 
    
26.     def send_txt(self, name, real_price,price, percent, status):    #信息内容排版的函数
    
27.         if 'up' == status:
    
28. # content = 'stock name:%s current price: %.2f higher than price up:%.2f , ratio:%.2f' % (name, real_price,price, percent)
    
29.              content = '证券名称:%s 当前价格: %.2f 高于 定价模型上限价格:%.2f , 可卖出 预计盈利率:%.2f' %(name, real_price,price, percent+15)
    
30.         if 'down' == status:
    
31. # content = 'stock name:%s current price: %.2f lower than price down:%.2f , 盈利率:%.2f' % (name, real_price,price, percent)
    
32.              content = '证券名称:%s 当前价格: %.2f 低于 定价模型下限价格:%.2f , 可买入 预计盈利率:%.2f' %(name, real_price,price, percent+15)
    
33.         content = content + '%'
    
34.         print content
    
35.         subject = '%s' % name
    
36.         self.msg = MIMEText(content, 'plain', 'utf-8')
    
37.         self.msg['to'] = self.to_mail
    
38.         self.msg['from'] = self.from_mail
    
39.         self.msg['Subject'] = subject
    
40.         self.msg['Date'] = Utils.formatdate(localtime=1)
    
41.         try:
    
42. 
    
43.             self.smtp = smtplib.SMTP_SSL(port=465)
    
44.             self.smtp.connect(self.server)
    
45.             self.smtp.login(self.username, self.password)
    
46.             self.smtp.sendmail(self.msg['from'], self.msg['to'], self.msg.as_string())
    
47.             self.smtp.quit()
    
48.             print "sent"
    
49.         except smtplib.SMTPException, e:
    
50.             print e
    
51.             return 0
    
52. 
    
53. def push_msg(name, real_price,price, percent, status):            #信息推送的函数
    
54.     cfg = TK.getUserData('data.cfg')
    
55.     from_mail = cfg['from_mail']
    
56.     password = cfg['password']
    
57.     to_mail = cfg['to_mail']
    
58.     obj = MailSend('smtp.qq.com', from_mail, password, to_mail)   ##这里调用了上述MailSend类
    
59.     obj.send_txt(name,real_price, price, percent, status)
    
60. 
    
61. def read_stock(name):
    
62.     f = open(name)
    
63.     stock_list = []
    
64. 
    
65.     for s in f.readlines():
    
66.         s = s.strip()
    
67.         row = s.split(';')
    
68.          #print row
    
69.         print "code :",row[0]
    
70.         print "price_down :",row[1]
    
71.         print "price_up :",row[2]
    
72.         stock_list.append(row)
    
73.     return stock_list
    
74. 
    
75. def compare_price(code, price_down, price_up):
    
76.     try:
    
77.         df = ts.get_realtime_quotes(code)
    
78.     except Exception, e:
    
79.         print e
    
80.         time.sleep(5)
    
81.         return 0
    
82.     real_price = df['price'].values[0]
    
83.     name = df['name'].values[0]
    
84.     print "stock name :",name
    
85.     real_price = float(real_price)
    
86.     print "current price :",real_price
    
87.     pre_close = float(df['pre_close'].values[0])
    
88.     print "close price :",pre_close
    
89.     if real_price >= price_up:
    
90.         percent_up = (real_price - price_up) / price_up * 100
    
91.         print 'percent : %.2f\n' % (percent_up),
    
92.         print '%s real_price %.2f higher than price_up %.2f , %.2f' % (name, real_price, price_up,percent_up),
    
93.         print '%'
    
94.         push_msg(name, real_price,price_up, percent_up, 'up')
    
95.         return 1
    
96.     if real_price = price_down:
    
97.         percent_down = (price_down - real_price) / price_down * 100
    
98.         print '%s real_price %.2f lower than price_down %.2f , %.2f' % (name, real_price, price_down,percent_down),
    
99.         print '%'
    
100.         push_msg(name, real_price,price_down, percent_down, 'down')
     
101.         return 1
     
102. 
     
103. def main():
     
104.     #read_stock('price.txt')
     
105.     # choice = input("Input your choice:\n")
     
106. 
     
107.     # if str(choice) == '1':
     
108.         stock_lists_price = read_stock('price.txt')
     
109.         while 1:
     
110.             t = 0
     
111.             for each_stock in stock_lists_price:
     
112.                 code = each_stock[0]
     
113.                 price_down = float(each_stock[1])
     
114.                 price_up = float(each_stock[2])
     
115.                 t = compare_price(code, price_down, price_up)
     
116.                 if t:
     
117.                     stock_lists_price.remove(each_stock)
     
118. 
     
119. if __name__ == '__main__':
     
120.     path=os.path.join(os.getcwd(),'data')
     
121.     if os.path.exists(path)==False:
     
122.          os.mkdir(path)
     
123.     os.chdir(path)
     
124. 
     
125.     main()

3.邮箱配置文件

点击(此处)折叠或打开

1. from_mail=xxxxxxx@qq.com
2. password=xxxxxxxx
3. to_mail=xxxxxxx@qq.com