#!/usr/bin/env python
# -*- coding: utf-8 -*-
#latest edit: 2009.06.19 
 
#to ensure the utf8 encoding environment
import sys
default_encoding = 'utf-8'
if sys.getdefaultencoding() != default_encoding:
    reload(sys)
    sys.setdefaultencoding(default_encoding)
 
import urllib
import base64
import re
import time                                 #time zone convertion
 
from google.appengine.api import urlfetch
from google.appengine.ext import db
 
def time_from_0_to_8(timestr,timezone=8): 
 
    TIMEFORMAT="%a %b %d %X +0000 %Y" 
    #Sat Jan 03 23:08:54 +0000 2009 
    ISOTIMEFORMAT='%Y-%m-%d %X' 
    x=time.strptime(timestr, TIMEFORMAT) 
    m=time.mktime(x)+60*60*timezone 
    p=time.strftime(ISOTIMEFORMAT,time.localtime(m)) 
    return p
 
def timestr_add_mins(timestr,mins=0,future=""):
    '''
    receive time str, format like: 2009-06-10 13:00:33
    and integer mins
    return seconds;
    '''
    ISOTIMEFORMAT='%Y-%m-%d %X'
    x=time.strptime(timestr, ISOTIMEFORMAT)
    m=time.mktime(x)+60*mins
    p=time.strftime(ISOTIMEFORMAT,time.localtime(m))
    return p
 
def add_time(p_time="",send_time=""):
    ISOTIMEFORMAT='%Y-%m-%d %X'
 
    now=time.localtime()
    #print now.tm_year
    p=""
    match = re.search(
	    r"""(?:(?P<year>\d{4})-(?=\d+))?
	(?:(?:(?P<month>\d{1,2})-)?(?P<date>\d{1,2})\s+)?
	(?P<hour>\d{1,2}):(?P<mins>\d{1,2}) |
	(?P<min3>\d+)
	    """, p_time, re.VERBOSE)
    if match:
        min3 = match.group("min3")
        if min3:
            x=time.strptime(send_time, ISOTIMEFORMAT)
            x=time.mktime(x)+int(min3)*60
            x=time.strftime(ISOTIMEFORMAT,time.localtime(x))
            return x
        year = match.group("year")
        if year==None:
            year=now.tm_year
        month = match.group("month")
        date = match.group("date")
        if month==None :
            month=now.tm_mon
        if date==None:
            date=now.tm_mday
        hour=match.group("hour")
        mins=match.group("mins")
        return """%s-%02d-%02d %02d:%02d:00"""%(year,int(month),int(date),int(hour),int(mins))
 
 
class Msgs(db.Model):
    msg_id = db.StringProperty()
    time = time = db.StringProperty() #the excution time; format: 2009-01-31 23:59
    send_time = db.StringProperty()
    d = db.BooleanProperty(default=False)
    p = db.BooleanProperty(default=False)
    q = db.BooleanProperty(default=False)
    task= db.TextProperty()
    user_id = db.StringProperty()
    user_name = db.StringProperty()
    done = db.BooleanProperty(default=False)
    def print_all(self):
        print "ID:%s in %s mins was %s, DPD:%d%d%d, by %s(%s). Task:%s" % \
        (self.msg_id,self.time,self.send_time,self.d,self.p,\
        self.done,self.user_name,self.user_id,self.task)
 
def reply_msg(user,sn,msg,reply_id=""):
 
    auth=base64.b64encode(user+":"+sn)        
    auth='Basic '+auth   
    form_fields = {
        "status": msg,
        "in_reply_to_status_id":reply_id        
    }
    form_data = urllib.urlencode(form_fields)
 
    result = urlfetch.fetch(url="""http://api.fanfou.com/statuses/update.xml""",\
            payload=form_data,
            headers={'Authorization':auth},
            method=urlfetch.POST
            )
 
    if result.status_code != 200:
        print "Send Msg:%s Error!"%msg
 
 
def private_msg(user,sn,msg,to_id):
    auth=base64.b64encode(user+":"+sn)        
    auth='Basic '+auth   
    form_fields = {
        "text": msg,
        "user":to_id
    }
    form_data = urllib.urlencode(form_fields)
 
    result = urlfetch.fetch(url="""http://api.fanfou.com/direct_messages/new.xml""",\
            payload=form_data,
            headers={'Authorization':auth},
            method=urlfetch.POST
            )
 
    if result.status_code != 200:
        print "Send Private Msg:%s Error!"%msg
 
 
#get one page of to user's replies
def getPage(user,sn,since="",page=1):
    auth=base64.b64encode(user+":"+sn)
    auth='Basic '+auth   
    result = urlfetch.fetch(url="http://api.fanfou.com/statuses/replies.xml?since_id=%s&page=%d" % (since,int(page)),headers={'Authorization':auth})
 
    if result.status_code == 200:        
        return result.content
    else:
        return False
 
#parse replies
#this is a private function used only by parse.
def parseMsg(msg,send_time):
    #the msg is formated as: "@... d[pq] \d+ \w+"
    #return tuple (d,p,q,time,content), the first 3 are boolean type; time :integer;content:task
    match = re.search(r"@\S+\s+(?P<mode>\w+)\s+(?P<time>[-\d\s:]+)\s+(?P<todo>.*)", msg, re.DOTALL | re.VERBOSE)
    d=False
    p=False
    q=False
    time=0
    content=""
    if match:
        #print msg
        mode = match.group("mode")
        #print "mode: %s"%mode
        if mode.find("d")!=-1:
            d=True
        if mode.find("p")!=-1:
            p=True
        if mode.find("q")!=-1:
            q=True
        time=add_time(match.group("time"),send_time)
        content=match.group("todo")
        content=db.Text(content,"utf_8")
        return (d,p,q,time,content)
    else:
        return ()
 
 
def parse(content):
    #regex to parse the content 
    matches = re.findall(
    r"""(?sx)<created_at>([^<]+)</created_at>\s*        #0 msg created time
        <id>([^<]+)</id>\s*                                #1 msg id
        <text><!\[CDATA\[(.*?)\]\]></text>.*?            #2 msg content
        <id>([^<]+)</id>\s*                                #3 msg sender's id
        <name>([^<]+)</name>                            #4 msg sender's name
        """,content)
    numbers =0    
    for m in matches:      
        msg=Msgs()
        msg.msg_id=m[1]
        msg.send_time=time_from_0_to_8(m[0])
        msg.user_id=m[3]
        msg.user_name=m[4]
        x=parseMsg(m[2],msg.send_time) 
        if x:
            try:
                (msg.d,msg.p,msg.q,msg.time,msg.task)=x
                msg.put()
                numbers+=1
 
                if not msg.q:#q=True:
                    receipt='''@%s 收到，我将于%s提醒您。谢谢使用！'''%\
                        (msg.user_name,msg.time[:-3])                
                    reply_msg("timers","jianjian",receipt,msg.msg_id)
 
                #print msg.msg_id
            except:
                print "asign value error" 
                return numbers
    return numbers                
 
def get_latest_id():
    msg=db.GqlQuery("SELECT * FROM Msgs ORDER BY send_time DESC LIMIT 1")#send_time
    for m in msg:
        if m.msg_id!=None:
            return m.msg_id
        else:
            return ""
 
 
 
def display_all():
    msg=db.GqlQuery("SELECT * FROM Msgs ORDER BY send_time DESC")#send_time
    if not msg.count():
        return 
    for m in msg:
        m.print_all()
 
def get_records():
    msg=db.GqlQuery("SELECT * FROM Msgs")#send_time
    return msg.count()
 
#condition= WHERE....
def delete_all(condition=""):
    msg=db.GqlQuery("SELECT * FROM Msgs %s"%condition)#send_time
    for m in msg:
        db.delete(m)
 
 
 
def count_down(user,sn):
    msg=db.GqlQuery("SELECT * FROM Msgs WHERE done= :1",False)
    index=0
    now=time.gmtime()
    now=time.mktime(now)+3600*8
    ISOTIMEFORMAT='%Y-%m-%d %X'
    x=time.strftime(ISOTIMEFORMAT,time.gmtime(now))
 
    for m in msg:
        index+=1
        #m.time-=1
        print "task id: %s, exe time:%s vs now:%s; result: %s"%(m.msg_id,m.time,x,m.time<=x)
        if (m.time<=x):
            #m.done=True
            #d,p
            if m.d:                
                reply_msg(user,sn,'''@%s %s'''%(m.user_name,m.task),m.msg_id)
            if m.p:                
                private_msg(user,sn,m.task,m.user_id)
 
            m.done=True
            m.put()
#        m.print_all()
    print "there are %d tasks to do."%index
    #display_all()
 
 
def mainloop(user,sn,latest): 
    page=1
    while(1):
        msg=getPage(user,sn,latest,page)
        #print msg
        if msg:
            num=parse(msg)
            page+=1
            #page=page+1
            if num==0:
                break    
 
 
 
print ""
 
user="timers"
sn="jianjian"
 
#delete_all()
 
#display_all()
#rutine check
count_down(user,sn)
 
latest=get_latest_id()
if latest==None:
    latest=""
 
mainloop(user,sn,latest)