# -*- coding: utf-8 -*-
#在某个网站上搜索指定字符串
def count_string(weburl,string_tobe_find):
    import urllib
    try:
        web = urllib.urlopen(weburl).read()
    except:
        return -1
    string = str(web)
    return string.count(string_tobe_find)

#发送邮件提醒管理员
def send_mail(sub,content):
    import smtplib
    from email.mime.text import MIMEText
    to_list=['XXXXX@XXX.com']
    mail_host="smtp.163.com"  #设置服务器
    mail_user="XXXXXXX"    #用户名
    mail_pass="XXXXXXX"   #口令
    mail_postfix="163.com"  #发件箱的后缀

me="提醒"+"<"+mail_user+"@"+mail_postfix+">"
    msg = MIMEText(content,_subtype='plain',_charset='gb2312')
    msg['Subject'] = sub
    msg['From'] = me
    msg['To'] = ";".join(to_list)
    try:
        server = smtplib.SMTP()
        server.connect(mail_host)
        server.login(mail_user,mail_pass)
        server.sendmail(me, to_list, msg.as_string())
        server.close()
        return True
    except Exception, e:
        print str(e)
        return False

#检查函数，调用上面的函数
def check():
    import time
    status = count_string('http://acm.sdut.edu.cn/web/status.php',"waiting")
    if status == -1:
        print time.ctime(time.time()),"OJ页面无法打开","请检查问题"
        if send_mail("OJ页面无法打开","请检查问题！\n\tfrom ma6174") == False:
            print '邮件发送失败'
            return False
        time.sleep(600)
    elif status > 1 :
        print time.ctime(time.time()),"OJ又waiting了","请尽快修复"
        if send_mail("OJ又waiting了","请尽快修复！\n\tfrom ma6174") == False:
            print '邮件发送失败'
            return False
        time.sleep(600)
    else:
        print time.ctime(time.time()),"\tOJ运行正常!"

#主函数
if __name__=='__main__':
    print "\tOJ运行检查程序 --- BY ma6174"
    from time import sleep
    while True:
        check()

sleep(60)