Python邮件提醒脚本

2月份之前，iPhone官网货源不是很足，所以写了个提醒脚本，定时刷新页面，条件符合则发送邮件提醒，

第一次用Python干了点事，Python的简洁、开发效率、丰富的库，给我留下很深印象

#!/usr/bin/python
#-*- coding:utf-8 -*-
#import sys
import httplib
import smtplib
import time


#read mailing list from specified file
def read_mailing_list(filename):
    mailing_list = []
    f = file(filename, 'r')
    while True:
        line = f.readline();
        if len(line) > 0:
            mailing_list.append(line.strip())
        else:
            break
    f.close()
    return mailing_list;


#check if there is some iphone to buy
def check_iphone_status(resp_content):
    s = '预计发货时间'
    pos = resp_content.find(s)
    if pos != -1:
        tag1 = '<span>'
        tag2 = '</span>'
        pos1 = resp_content.find(tag1, pos)
        pos2 = resp_content.find(tag2, pos)
        if pos1 != -1 and pos2 != -1:
            dst = resp_content[pos1 + len(tag1) : pos2]
            dst = ''.join(dst.split())
            #print dst
            if dst == '暂无供应':
                #print 'no iPhone 4S'
                return -1
            else:
                print 'hello, iPhone 4S'
                return 0 
        else:
            return -1
    else:
        return -1


#send mail
def send_mail(_from, _to, _msg):
    mailsrv = smtplib.SMTP('smtp.163.com')
    mailsrv.login('xxx', 'xxxxxx')
    #mailsrv.ehlo()
    #mailsrv.starttls()
    #mailsrv.ehlo()
    mailsrv.sendmail(_from, _to, _msg)
    mailsrv.quit()
    

#script start to run
#read mailing list
mailing_list = read_mailing_list('mailing_list')
print 'mailing list: ' , mailing_list
mail_from = 'xxxx@163.com'
#mail_to = ','.join(mailing_list)
mail_msg = 'Subject:iPhone 4S in sale\n\nplease visit:\n \
            http://www.apple.com.cn\n \
            http://store.apple.com/cn\n \
            http://store.apple.com/cn/browse/home/shop_iphone/family/iphone'
mail_msg2 = 'Subject:iPhone 4S sell out\n\nplease visit:\n \
            http://www.apple.com.cn\n \
            http://store.apple.com/cn\n \
            http://store.apple.com/cn/browse/home/shop_iphone/family/iphone'

f = file('log', 'w')
f.write('mailing list:\n')
for item in mailing_list:
    f.write(item + '\n')
    print item
f.write('\n')
f.flush()

#send http request
while True:
    conn = httplib.HTTPConnection('store.apple.com')
    conn.request('GET', '/cn/browse/home/shop_iphone/family/iphone/iphone4s') 
    result = conn.getresponse()
    resp_status = result.status
    if resp_status == 200:
        resp_content = result.read()
        status = check_iphone_status(resp_content)
        if status == 0:
            #send_mail(mail_from, mailing_list, mail_msg)
            f.write('iPhone 4S in sale\n')
            f.flush()
            print 'iPhone 4S in sale'
        else:
            #send_mail(mail_from, mailing_list, mail_msg2)
            #print 'send mail done'
            f.write('iPhone 4S sell out\n')
            f.flush()
            print 'iPhone 4S sell out'
    conn.close()
    time.sleep(60)

f.close()