3 [! @sdZddlZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z ddlmZddlmZdd lmZmZdd lmZejd Zejejdaed d dddgZedd ddddddgZdddZgZe gdZe gdZ e!dddZ"dZ#ej$dj%e#e#ej&Z'ej$dej(ej&BZ)d d!Z*d"d#Z+d$d%Z,d&d'Z-d(d)Z.d*d+Z/eed,d-d.Z0dCd/d0Z1d1d2d3giZ2d4d5Z3d6d7Z4ed e)d8d Z5ed9d:d;dd:d;d?d@Z7e dAdBZ dS)DzB .tell and .remind from Karkat, Infobot-ized * depends: database N)utc)commandinitprocess_privmsg)CaseInsensitiveDefaultDict)CommandException) namedtuple)datetime timedelta) search_datesremindTellid from_nickmessagedateReminderto_nickchannelbegintsendtsTUTC)ZRETURN_AS_TIMEZONE_AWAREZTIMEZONE)default)timespancCs4|dkr dS|dkrdS|dkr$d S|dkr0dSdS)Nweekwkwi: daydiQhourhrhiminuteminm<r)rrr)rr)r r!r")r#r$r%)rr'r'!/opt/Infobot/plugins/reminders.pyget_seconds_for_timespan!sr)zT(?:(?P\d+) ?(?Pweek|wk|w|day|d|hour|hr|h|minute|min|m|second|sec|s)s?)zin {}( ?\+? ?{})*za^ \.(?:tell|remind)\s (?P me|@? [A-z0-9]+)\s (?P.+)$cCst|jt||||dS)N)tellsappendr)rtellidrrrr'r'r(add_tell6sr-cGst|}tjt}||jkr*t||dS|j|}tj|jt||fd}||_ d|_ |j t j |t|jj |tjd||jdS)N)argsTz'Added reminder %r (fires in %f seconds))rr nowrrremind_handler threadingTimer total_secondsreminderdaemonstarttimersr+ remindersrloggerdebug)botr.r4 current_timedeltatimerr'r'r( add_reminder:s     r?cCs(tjdj}x|D] }t|qWdS)NzWSELECT to_nick, tellid, from_nick, message, begints FROM tells WHERE fulfilled = false;)dbexecutefetchallr-)datatellr'r'r( db_get_tellsJs rEcCs.tjdj}x|D]}t|f|qWdS)NzgSELECT id, to_nick, from_nick, message, channel, begints, endts FROM reminders WHERE fulfilled = false;)r@rArBr?)r;rCr4r'r'r(db_get_remindersOs rFcCs~tjtjt|j}dj|jjd|jj ||jj |j |jj dj|j |jj dj|}|j ||t jd|jdS)Nu{} {}: {} · {} · {}u| ✉ |zfrom {}u⌚ {}z4UPDATE tells SET fulfilled = true WHERE tellid = %s;)humanize naturaltimer r/rrformatstylegreentealrrnoticer@rAr)r;nickchanrD timestringoutputr'r'r( send_tellTs    rRcCs2|tkr.xt|D]}t||||qWt|=dS)N)r*rR)r;rNrOrDr'r'r( tell_handlerbsrS)tdreturncCstt|jdS)N)seconds)r roundr3)rTr'r'r(round_to_secondshsrXcCstjttjt|j}dj|jj d|jj |j |jj |j |jj dj|j |jj dj|}|j|j|tjd|jfdS)Nu{} {}: {} · {} · {}u| ✉ |zfrom {}u⌚ {}z4UPDATE reminders SET fulfilled = true WHERE id = %s;)rGrHrXr r/rrrIrJrKrLrrrrMrr@rAr)r;r4Z late_timerPrQr'r'r(r0ks   r0PRIVMSGcCst|ft|ddS)N)rSr)br%r'r'r({sr\cCs|dS)Nrr')timesr'r'r( time_select~sr^cCstj|}tjd||s"dSt|jd|jd}d}x$|D]\}}|t|t|7}qDWt|d}tjd||j dt j t |ffS)NzFound timestamp match: numberspanr)rVz timedelta: ) SIMPLE_SPAN_REsearchr9r:zipZcapturesr)intr groupr r/r)msgr% date_partslengthnumr`tr'r'r( get_simple_tss  rkc Cs`|d}|d}t|}|dkr$|}|s8t|dgtd}||krDdnd}|jd}|rt|\} } |j| d j} tjd | d | tj t } | | krt d t j d ||| || fjd} t|| ||| || | t| | }|j|d|d|d| dnV|} t j d||| fjd}t|jt||| tj t |j|d|d|ddS)NrNmessage_with_timemeen) languagessettingsZthemZyou@zTo cut: z End TS: z9You're trying to start a reminder for a date in the past.ziINSERT INTO reminders (from_nick, to_nick, message, channel, endts) VALUES (%s,%s,%s,%s,%s) RETURNING id;rz I'll remind z in z. To cancel, send .rmcancel .zSINSERT INTO tells (to_nick, from_nick, message) VALUES (%s,%s,%s) RETURNING tellid;z I'll tell z that. To cancel, send .tcancel )rkr dp_settingsstripr^replacer9r:r r/rrr@rAfetchoner?rXrfr*r+r)r;rNrOgrargrrlr]ZpronounZcutrrutcnowZ reminderidr=r,r'r'r(r s4    "Zrmcancelrs)cmdcharc Csy t|}Wn tk r,tdddYnXtjd|fj}|sLtd|\}}}} | tjtkrntd||kr~tdtjd|fx"t D]} | j j |kr| j PqW|j |d d S) z> .rmcancel - Cancels your started reminder, if possible. zreminder id missing or invalid.T)send_doczGSELECT from_nick, to_nick, message, endts FROM reminders WHERE id = %s;zthat reminder does not exist.z that reminder has already fired.z!you did not create that reminder!z$DELETE FROM reminders WHERE id = %s;z Reminder cancelled successfully.N)rd ValueErrorrr@rArwr r/rr7r4rcancelrf) r;rNrOryZridrowrrrrr>r'r'r(cancel_reminders&     rZtcancelc Csy t|}Wn tk r,tdddYnXtjd|fj}|sLtd|\}}}} } | rftd||krvtdtjd|fx t|D]} | j|kr| } PqWt|j| |j |d d S) z1 .tcancel - Cancels your tell, if possible. ztell id missing or invalid.T)r|zTSELECT to_nick, from_nick, message, begints, fulfilled FROM tells WHERE tellid = %s;zthat tell does not exist.z that tell has already been sent.zyou did not create that tell!z$DELETE FROM tells WHERE tellid = %s;zTell cancelled successfully.N) rdr}rr@rArwr*rremoverf) r;rNrOrytidrrrrrZ fulfilledrDZ to_remover'r'r( cancel_tells(   rcs<jdatjjjfddjjjdddS)Nr@cstS)N)rFr')r;r'r(r\szinit..cSst||jdd|S)N!r)rSsplit)r;hostrr'r'r(r\s)rCr@rEeventsWelcomeregisterJoin)r;r')r;r(rs r)N)8__doc__rGregexr1loggingZpytzrZutil.decoratorsrrrZ util.dictrZCIDDZ util.datar collectionsr r r Zdateparser.searchr getLoggerr9setLevelDEBUGr@rrrtr7r8r*strr)Z TIMESPANScompilerIZV1raXZREMIND_TELL_REr-r?rErFrRrSrXr0 __callbacks__r^rkr rrr'r'r'r(sV               ,