gaetalk.py

   1 #!/usr/bin/env python2
   2 # vim:fileencoding=utf-8
   3
   4 import re
   5 import logging
   6 import datetime
   7
   8 from google.appengine.ext import db
   9 from google.appengine.api import xmpp
  10
  11 import utils
  12 import config
  13
  14 helpre = re.compile(r'^\W{0,2}help$')
  15
  16 #用户所有资源离线时，会加上“完全”二字
  17 OFFLINE = u'离线'
  18 AWAY    = u'离开'
  19 XAWAY   = u'离开'
  20 BUSY    = u'忙碌'
  21 ONLINE  = u'在线'
  22 CHAT    = u'和我说话吧'
  23
  24 NEW        = u'加入'
  25 LEAVE      = u'退出'
  26 NICK       = u'昵称更改 (%s -> %s)'
  27 SNOOZE     = u'snooze %ds'
  28 BLACK      = u'禁言 %s %ds'
  29 BLACK_AUTO = u'被禁言 %ds'
  30 KICK       = u'删除 %s (%s)'
  31 ADMIN      = u'%s 成为管理员 (by %s)'
  32 UNADMIN    = u'%s 不再是管理员 (by %s)'
  33 NOTICE     = u'通告：%s'
  34 BLOCK      = u'封禁 %s，原因：%s'
  35 UNBLOCK    = u'解封 %s'
  36
  37 STATUS_CODE = {
  38   '':     ONLINE,
  39   'away': AWAY,
  40   'dnd':  BUSY,
  41   'xa':   XAWAY,
  42   'chat': CHAT,
  43 }
  44
  45 #状态的排序顺序
  46 STATUS_LIST = [CHAT, ONLINE, AWAY, XAWAY, BUSY, OFFLINE]
  47
  48 timezone = datetime.timedelta(hours=config.timezoneoffset)
  49
  50 class User(db.Model):
  51   jid = db.StringProperty(required=True, indexed=True)
  52   nick = db.StringProperty(required=True, indexed=True)
  53
  54   add_date = db.DateTimeProperty(auto_now_add=True)
  55   last_online_date = db.DateTimeProperty()
  56   last_offline_date = db.DateTimeProperty()
  57   last_speak_date = db.DateTimeProperty()
  58
  59   msg_count = db.IntegerProperty(required=True, default=0)
  60   msg_chars = db.IntegerProperty(required=True, default=0)
  61   credit = db.IntegerProperty(required=True, default=0)
  62
  63   black_before = db.DateTimeProperty(auto_now_add=True)
  64   snooze_before = db.DateTimeProperty()
  65
  66   avail = db.StringProperty(required=True)
  67   is_admin = db.BooleanProperty(required=True, default=False)
  68   resources = db.StringListProperty(required=True)
  69   #允许他人私信？
  70   reject_pm = db.BooleanProperty(default=False)
  71
  72   prefix = db.StringProperty(required=True, default=config.default_prefix)
  73   nick_pattern = db.StringProperty(required=True, default='[%s]')
  74   nick_changed = db.BooleanProperty(required=True, default=False)
  75   intro = db.StringProperty()
  76
  77 class Log(db.Model):
  78   time = db.DateTimeProperty(auto_now_add=True, indexed=True)
  79   msg = db.StringProperty(required=True, multiline=True)
  80   jid = db.StringProperty()
  81   nick = db.StringProperty()
  82   type = db.StringProperty(required=True, indexed=True,
  83                            choices=set(['chat', 'member', 'admin', 'misc']))
  84
  85 class Group(db.Model):
  86   time = db.DateTimeProperty(auto_now_add=True)
  87   topic = db.StringProperty(multiline=True)
  88   status = db.StringProperty()
  89
  90 class BlockedUser(db.Model):
  91   jid = db.StringProperty(required=True, indexed=True)
  92   add_date = db.DateTimeProperty(auto_now_add=True)
  93   reason = db.StringProperty()
  94
  95 def log_msg(sender, msg):
  96   l = Log(jid=sender.jid, nick=sender.nick,
  97           type='chat', msg=msg)
  98   l.put()
  99
 100 def log_onoff(sender, action, resource=''):
 101   if resource:
 102     if action == OFFLINE and not sender.resources:
 103       msg = u'完全%s (%s)' % (action, resource)
 104     else:
 105       msg = '%s (%s)' % (action, resource)
 106   else:
 107     msg = action
 108   l = Log(jid=sender.jid, nick=sender.nick,
 109           type='member', msg=msg)
 110   l.put()
 111
 112 def log_admin(sender, action):
 113   l = Log(jid=sender.jid, nick=sender.nick,
 114           type='admin', msg=action)
 115   l.put()
 116
 117 def get_user_by_jid(jid):
 118   return User.gql('where jid = :1', jid.lower()).get()
 119
 120 def get_user_by_nick(nick):
 121   return User.gql('where nick = :1', nick).get()
 122
 123 def get_group_info():
 124   return Group.all().get()
 125
 126 def get_blocked_user(jid):
 127   return BlockedUser.gql('where jid = :1', jid.lower()).get()
 128
 129 def get_member_list():
 130   now = datetime.datetime.now()
 131   #一个查询中最多只能有一个不等比较
 132   l = User.gql('where avail != :1', OFFLINE)
 133   return [unicode(x.jid) for x in l \
 134           if x.snooze_before is None or x.snooze_before < now]
 135
 136 def send_to_all_except(jid, message):
 137   if isinstance(jid, str):
 138     jids = [x for x in get_member_list() if x != jid]
 139   else:
 140     jids = [x for x in get_member_list() if x not in jid]
 141   logging.debug(jid)
 142   logging.debug(jids)
 143   try:
 144     xmpp.send_message(jids, message)
 145   except xmpp.InvalidJidError:
 146     pass
 147
 148 def send_to_all(message):
 149   jids = get_member_list()
 150   xmpp.send_message(jids, message)
 151
 152 def send_status(jid):
 153   if get_blocked_user(jid):
 154     xmpp.send_presence(jid, status=u'您已经被本群封禁')
 155     return
 156
 157   grp = get_group_info()
 158   if grp is None or not grp.status:
 159     xmpp.send_presence(jid)
 160   else:
 161     xmpp.send_presence(jid, status=grp.status)
 162
 163 def handle_message(msg):
 164   jid = msg.sender.split('/')[0]
 165
 166   sender = get_blocked_user(jid)
 167   if sender is not None:
 168     msg.reply(u'您已经被本群封禁，原因为 %s。' % sender.reason)
 169     return
 170
 171   sender = get_user_by_jid(jid)
 172   if sender is None:
 173     msg.reply('很抱歉，出错了，请尝试更改状态或者重新添加好友。')
 174     return
 175
 176   if msg.body.startswith('?OTR:'):
 177     msg.reply('不支持 OTR 加密！')
 178     return
 179
 180   if msg.body in config.blocked_away_messages:
 181     msg.reply('系统认为您的客户端正在自动发送离开消息。如果您认为这并不正确，请向管理员反馈。')
 182     return
 183
 184   if len(msg.body) > 500 or msg.body.count('\n') > 5:
 185     msgbody = config.post_code(msg.body)
 186     if msgbody:
 187       msgbody += '/text' # 默认当作纯文本高亮
 188       msg.reply(u'内容过长，已贴至 %s 。' % msgbody)
 189       firstline = ''
 190       lineiter = iter(msg.body.split('\n'))
 191       try:
 192         while not firstline:
 193           firstline = lineiter.next()
 194       except StopIteration:
 195         pass
 196       if len(firstline) > 40:
 197         firstline = firstline[:40]
 198       msgbody += '\n' + firstline + '...'
 199     else:
 200       logging.warn(u'贴代码失败，代码长度 %d' % len(msg.body))
 201       msg.reply('由于技术限制，每条消息最长为 500 字。大段文本请贴 paste 网站。\n'
 202                 '如 http://paste.ubuntu.org.cn/ http://slexy.org/\n'
 203                )
 204       return
 205     ch = None
 206   else:
 207     msgbody = msg.body
 208     if sender.is_admin or sender.jid == config.root:
 209       ch = AdminCommand(msg, sender)
 210     else:
 211       ch = BasicCommand(msg, sender)
 212
 213   if not ch or not ch.handled:
 214     now = datetime.datetime.now()
 215     if sender.black_before is not None \
 216        and sender.black_before > now:
 217       if (datetime.datetime.today()+timezone).date() == \
 218          (sender.black_before+timezone).date():
 219         format = '%H时%M分%S秒'
 220       else:
 221         format = '%m月%d日 %H时%M分%S秒'
 222       msg.reply('你已被禁言至 ' \
 223                 + (sender.black_before+timezone).strftime(format))
 224       return
 225
 226     sender.last_speak_date = now
 227     sender.snooze_before = None
 228     try:
 229       sender.msg_count += 1
 230       sender.msg_chars += len(msgbody)
 231     except TypeError:
 232       sender.msg_count = 1
 233       sender.msg_chars = len(msgbody)
 234     sender.put()
 235     body = utils.removelinks(msgbody)
 236     for u in User.gql('where avail != :1', OFFLINE):
 237       if u.snooze_before is not None and u.snooze_before >= now:
 238         continue
 239       if u.jid == sender.jid:
 240         continue
 241       try:
 242         message = '%s %s' % (
 243           u.nick_pattern % sender.nick,
 244           body
 245         )
 246         xmpp.send_message(u.jid, message)
 247       except xmpp.InvalidJidError:
 248         pass
 249     log_msg(sender, msgbody)
 250
 251 def try_add_user(jid, show=OFFLINE, resource=''):
 252   '''使用 memcache 作为锁添加用户'''
 253   u = get_blocked_user(jid)
 254   if u is not None:
 255     xmpp.send_message(jid, u'您已经被本群封禁，原因为 %s。' % u.reason)
 256     return
 257
 258   L = utils.MemLock('add_user')
 259   L.require()
 260   try:
 261     u = get_user_by_jid(jid)
 262     if u is not None:
 263       return
 264     u = add_user(jid, show, resource)
 265   finally:
 266     L.release()
 267   if show != OFFLINE:
 268     log_onoff(u, show, resource)
 269   logging.info(u'%s added', jid)
 270
 271 def add_user(jid, show=OFFLINE, resource=''):
 272   '''resource 在 presence type 为 available 里使用'''
 273   nick = jid.split('@')[0]
 274   # same user name with different domains are possible
 275   while get_user_by_nick(nick):
 276     nick += '_'
 277   u = User(jid=jid.lower(), avail=show, nick=nick)
 278   if show != OFFLINE:
 279     u.last_online_date = datetime.datetime.now()
 280   if resource:
 281     u.resources.append(resource)
 282   u.put()
 283   log_onoff(u, NEW)
 284   logging.info(u'%s 已经加入' % jid)
 285   send_status(jid)
 286   xmpp.send_message(jid, u'欢迎 %s 加入！要获取使用帮助，请输入 %shelp，要获知群主题，请输入 %stopic。' % (
 287     u.nick, u.prefix, u.prefix))
 288   return u
 289
 290 def del_user(jid, by_cmd=False):
 291   l = User.gql('where jid = :1', jid.lower())
 292   for u in l:
 293     if u.jid == config.root:
 294       xmpp.send_message(jid, u'root 用户：离开前请确定你已做好善后工作！')
 295     log_onoff(u, LEAVE)
 296     if by_cmd:
 297       logging.info(u'%s (%s) 已经离开 (通过使用命令)' % (u.nick, u.jid))
 298     else:
 299       logging.info(u'%s (%s) 已经离开' % (u.nick, u.jid))
 300     u.delete()
 301
 302 class BasicCommand:
 303   handled = True
 304   def __init__(self, msg, sender):
 305     self.sender = sender
 306     self.msg = msg
 307
 308     if helpre.match(msg.body):
 309       self.do_help()
 310     elif msg.body.startswith(sender.prefix):
 311       cmd = msg.body[len(sender.prefix):].split()
 312       try:
 313         handle = getattr(self, 'do_' + cmd[0])
 314       except AttributeError:
 315         msg.reply(u'错误：未知命令 %s' % cmd[0])
 316       except IndexError:
 317         msg.reply(u'错误：无命令')
 318       except UnicodeEncodeError:
 319         msg.reply(u'错误：命令名解码失败。此问题在 GAE 升级其 Python 到 3.x 后方能解决。')
 320       else:
 321         handle(cmd[1:])
 322         logging.debug('%s did command %s' % (sender.jid, cmd[0]))
 323     else:
 324       self.handled = False
 325
 326   def get_msg_part(self, index):
 327     '''返回消息中第 index 个单词及其后的所有字符'''
 328     return self.msg.body[len(self.sender.prefix):].split(None, index)[-1]
 329
 330   def do_online(self, args):
 331     '''在线成员列表。可带一个参数，指定在名字中出现的一个子串。'''
 332     r = []
 333     pat = args[0] if args else None
 334     now = datetime.datetime.now()
 335     l = User.gql('where avail != :1', OFFLINE)
 336     for u in l:
 337       m = u.nick
 338       if pat and m.find(pat) == -1:
 339         continue
 340       status = u.avail
 341       if status != ONLINE:
 342         m += u' (%s)' % status
 343       if u.snooze_before is not None and u.snooze_before > now:
 344         m += u' (snoozing)'
 345       if u.black_before is not None and u.black_before > now:
 346         m += u' (已禁言)'
 347       r.append(unicode('* ' + m))
 348     r.sort()
 349     n = len(r)
 350     if pat:
 351       r.insert(0, u'在线成员列表（包含子串 %s）:' % pat)
 352       r.append(u'共 %d 人。' % n)
 353     else:
 354       r.insert(0, u'在线成员列表:')
 355       r.append(u'共 %d 人在线。' % n)
 356     self.msg.reply(u'\n'.join(r).encode('utf-8'))
 357
 358   def do_lsadmin(self, args):
 359     '''管理员列表'''
 360     r = []
 361     now = datetime.datetime.now()
 362     l = User.gql('where is_admin = :1', True)
 363     for u in l:
 364       m = u.nick
 365       status = u.avail
 366       if status != ONLINE:
 367         m += u' (%s)' % status
 368       if u.snooze_before is not None and u.snooze_before > now:
 369         m += u' (snoozing)'
 370       if u.black_before is not None and u.black_before > now:
 371         m += u' (已禁言)'
 372       r.append(unicode('* ' + m))
 373     r.sort()
 374     n = len(r)
 375     r.insert(0, u'管理员列表:')
 376     r.append(u'共 %d 位管理员。' % n)
 377     self.msg.reply(u'\n'.join(r).encode('utf-8'))
 378
 379   def do_lsblocked(self, args):
 380     '''列出被封禁用户名单'''
 381     r = []
 382     l = BlockedUser.all()
 383     for u in l:
 384       r.append(unicode('* %s (%s, %s)' % (u.jid,
 385                                           utils.strftime(u.add_date, timezone),
 386                                           u.reason)
 387                       )
 388               )
 389     r.sort()
 390     n = len(r)
 391     r.insert(0, u'封禁列表:')
 392     r.append(u'共 %d 个 JID 被封禁。' % n)
 393     self.msg.reply(u'\n'.join(r).encode('utf-8'))
 394
 395   def do_chatty(self, args):
 396     '''消息数排行'''
 397     r = []
 398     for u in User.gql('ORDER BY msg_count ASC'):
 399       m = u.nick
 400       m = u'* %s:\t%5d条，共 %s' % (
 401         u.nick, u.msg_count,
 402         utils.filesize(u.msg_chars))
 403       r.append(m)
 404     n = len(r)
 405     r.insert(0, u'消息数量排行:')
 406     r.append(u'共 %d 人。' % n)
 407     self.msg.reply(u'\n'.join(r).encode('utf-8'))
 408
 409   def do_nick(self, args):
 410     '''更改昵称，需要一个参数，不能使用大部分标点符号'''
 411     if len(args) != 1:
 412       self.msg.reply('错误：请给出你想到的昵称（不能包含空格）')
 413       return
 414
 415     q = get_user_by_nick(args[0])
 416     if q is not None:
 417       self.msg.reply('错误：该昵称已被使用，请使用其它昵称')
 418     elif not utils.checkNick(args[0]):
 419       self.msg.reply('错误：非法的昵称')
 420     else:
 421       if not config.nick_can_change and self.sender.nick_changed:
 422         self.msg.reply('乖哦，你已经没机会再改昵称了')
 423         return
 424       old_nick = self.sender.nick
 425       log_onoff(self.sender, NICK % (old_nick, args[0]))
 426       self.sender.nick = args[0]
 427       self.sender.nick_changed = True
 428       self.sender.put()
 429       send_to_all_except(self.sender.jid,
 430         (u'%s 的昵称改成了 %s' % (old_nick, args[0])).encode('utf-8'))
 431       self.msg.reply('昵称更改成功！')
 432   do_nick.__doc__ += '，最长 %d 字节' % config.nick_maxlen
 433
 434   def do_whois(self, args):
 435     '''查看用户信息，参数为用户昵称'''
 436     if len(args) != 1:
 437       self.msg.reply('错误：你想知道关于谁的信息？')
 438       return
 439
 440     u = get_user_by_nick(args[0])
 441     if u is None:
 442       self.msg.reply(u'Sorry，查无此人。')
 443       return
 444
 445     now = datetime.datetime.now()
 446     status = u.avail
 447     addtime = (u.add_date + timezone).strftime('%Y年%m月%d日 %H时%M分').decode('utf-8')
 448     allowpm = u'否' if u.reject_pm else u'是'
 449     if u.snooze_before is not None and u.snooze_before > now:
 450       status += u' (snoozing)'
 451     if u.black_before is not None and u.black_before > now:
 452       status += u' (已禁言)'
 453     r = []
 454     r.append(u'昵称：\t%s' % u.nick)
 455     if self.sender.is_admin:
 456       r.append(u'JID：\t%s' % u.jid)
 457     r.append(u'状态：\t%s' % status)
 458     r.append(u'消息数：\t%d' % u.msg_count)
 459     r.append(u'消息总量：\t%s' % utils.filesize(u.msg_chars))
 460     r.append(u'加入时间：\t%s' % addtime)
 461     r.append(u'接收私信：\t%s' % allowpm)
 462     r.append(u'自我介绍：\t%s' % u.intro)
 463     self.msg.reply(u'\n'.join(r).encode('utf-8'))
 464
 465   def do_help(self, args=()):
 466     '''显示本帮助。参数 long 显示详细帮助，也可指定命令名。'''
 467     doc = []
 468     prefix = self.sender.prefix
 469
 470     if len(args) > 1:
 471       self.msg.reply('参数错误。')
 472       return
 473     arg = args[0] if args else None
 474
 475     if arg is None or arg == 'long':
 476       for b in self.__class__.__bases__ + (self.__class__,):
 477         for c, f in b.__dict__.items():
 478           if c.startswith('do_'):
 479             if arg is None:
 480               doc.append(u'%s%s:\t%s' % (prefix, c[3:], f.__doc__.decode('utf-8').\
 481                                          split(u'，', 1)[0].split(u'。', 1)[0]))
 482             else:
 483               doc.append(u'%s%s:\t%s' % (prefix, c[3:], f.__doc__.decode('utf-8')))
 484       doc.sort()
 485       if arg is None:
 486         doc.insert(0, u'** 命令指南 **\n(当前命令前缀 %s，可设置。使用 %shelp long 显示详细帮助)' % (prefix, prefix))
 487       else:
 488         doc.insert(0, u'** 命令指南 **\n(当前命令前缀 %s，可设置)' % prefix)
 489       doc.append(u'要离开，直接删掉好友即可。')
 490       doc.append(u'Gtalk 客户端用户要离开请使用 quit 命令。')
 491       self.msg.reply(u'\n'.join(doc).encode('utf-8'))
 492     else:
 493       try:
 494         handle = getattr(self, 'do_' + arg)
 495       except AttributeError:
 496         self.msg.reply(u'错误：未知命令 %s' % arg)
 497       except UnicodeEncodeError:
 498         self.msg.reply(u'错误：命令名解码失败。此问题在 GAE 升级其 Python 到 3.x 后方能解决。')
 499       else:
 500         self.msg.reply(u'%s%s:\t%s' % (prefix, arg, handle.__doc__.decode('utf-8')))
 501
 502   def do_topic(self, args=()):
 503     '''查看群主题'''
 504     grp = get_group_info()
 505     if grp is None or not grp.topic:
 506       self.msg.reply(u'没有设置群主题。')
 507     else:
 508       self.msg.reply(grp.topic)
 509
 510   def do_iam(self, args):
 511     '''查看自己的信息'''
 512     u = self.sender
 513     addtime = (u.add_date + timezone).strftime('%Y年%m月%d日 %H时%M分').decode('utf-8')
 514     allowpm = u'否' if u.reject_pm else u'是'
 515     r = []
 516     r.append(u'昵称：\t%s' % u.nick)
 517     r.append(u'JID：\t%s' % u.jid)
 518     r.append(u'资源：\t%s' % u' '.join(u.resources))
 519     r.append(u'消息数：\t%d' % u.msg_count)
 520     r.append(u'消息总量：\t%s' % utils.filesize(u.msg_chars))
 521     r.append(u'加入时间：\t%s' % addtime)
 522     r.append(u'命令前缀：\t%s' % u.prefix)
 523     r.append(u'接收私信：\t%s' % allowpm)
 524     r.append(u'自我介绍：\t%s' % u.intro)
 525     self.msg.reply(u'\n'.join(r).encode('utf-8'))
 526
 527   def do_m(self, args):
 528     '''发私信，需要昵称和内容两个参数。私信不会以任何方式被记录。用户可使用 set 命令设置是否接收私信。'''
 529     if len(args) < 2:
 530       self.msg.reply('请给出昵称和内容。')
 531       return
 532
 533     target = get_user_by_nick(args[0])
 534     if target is None:
 535       self.msg.reply('Sorry，查无此人。')
 536       return
 537
 538     if target.reject_pm:
 539       self.msg.reply('很抱歉，对方不接收私信。')
 540       return
 541
 542     msg = self.get_msg_part(2)
 543     msg = u'_私信_ %s %s' % (target.nick_pattern % self.sender.nick, msg)
 544     xmpp.send_message(target.jid, msg)
 545
 546   def do_intro(self, arg):
 547     '''设置自我介绍信息'''
 548     if not arg:
 549       self.msg.reply('请给出自我介绍的内容。')
 550       return
 551
 552     msg = self.get_msg_part(1)
 553     u = self.sender
 554     try:
 555       u.intro = msg
 556     except db.BadValueError:
 557       # 过长文本已在 handle_message 中被拦截
 558       self.msg.reply('错误：自我介绍内容只能为一行。')
 559       return
 560
 561     u.put()
 562     self.msg.reply(u'设置成功！')
 563
 564   def do_snooze(self, args):
 565     '''暂停接收消息，参数为时间（默认单位为秒）。再次发送消息时自动清除'''
 566     if len(args) != 1:
 567       self.msg.reply('你想停止接收消息多久？')
 568       return
 569     else:
 570       try:
 571         n = utils.parseTime(args[0])
 572       except ValueError:
 573         self.msg.reply('Sorry，我无法理解你说的时间。')
 574         return
 575
 576     try:
 577       self.sender.snooze_before = datetime.datetime.now() + datetime.timedelta(seconds=n)
 578     except OverflowError:
 579       self.msg.reply('Sorry，你不能睡太久。')
 580       return
 581     self.sender.put()
 582     if n == 0:
 583       self.msg.reply('你已经醒来。')
 584     else:
 585       self.msg.reply(u'OK，停止接收消息 %s。' % utils.displayTime(n))
 586     log_onoff(self.sender, SNOOZE % n)
 587
 588   def do_offline(self, args):
 589     '''假装离线，让程序认为你的所有资源已离线。如在你离线时程序仍认为你在线，请使用此命令。'''
 590     del self.sender.resources[:]
 591     self.sender.avail = OFFLINE
 592     self.sender.last_offline_date = datetime.datetime.now()
 593     self.sender.put()
 594     self.msg.reply('OK，在下次你说你在线之前我都认为你已离线。')
 595
 596   def do_fakeresource(self, args):
 597     '''假装在线，人工加入一个新的资源，使程序认为你总是在线。使用 off 参数来取消。'''
 598     if args and args[0] == 'off':
 599       try:
 600         self.sender.resources.remove('fakeresouce')
 601         self.msg.reply('OK，fakeresouce 已取消。')
 602         self.sender.put()
 603       except ValueError:
 604         self.msg.reply('没有设置 fakeresouce。')
 605     else:
 606       try:
 607         self.sender.resources.index('fakeresouce')
 608         self.msg.reply('你已经设置了永远在线。')
 609       except ValueError:
 610         self.sender.resources.append('fakeresouce')
 611         self.msg.reply('OK，你将永远在线。')
 612         self.sender.put()
 613
 614   def do_old(self, args):
 615     '''聊天记录查询，可选一个数字参数。默认为最后20条。特殊参数 OFFLINE （不区分大小写）显示离线消息（最多 100 条）'''
 616     s = self.sender
 617     q = False
 618     if not args:
 619       q = Log.gql("WHERE type = 'chat' ORDER BY time DESC LIMIT 20")
 620     elif len(args) == 1:
 621       try:
 622         n = int(args[0])
 623         if n > 0:
 624           q = Log.gql("WHERE type = 'chat' ORDER BY time DESC LIMIT %d" % n)
 625       except ValueError:
 626         if args[0].upper() == 'OFFLINE':
 627           q = Log.gql("WHERE time < :1 AND time > :2 AND type = 'chat' ORDER BY time DESC LIMIT 100", s.last_online_date, s.last_offline_date)
 628         else:
 629           pass
 630     if q is not False:
 631       r = []
 632       q = list(q)
 633       q.reverse()
 634       if q:
 635         if datetime.datetime.today() - q[0].time > datetime.timedelta(hours=24):
 636           show_date = True
 637         else:
 638           show_date = False
 639       for l in q:
 640         message = '%s %s %s' % (
 641           utils.strftime(l.time, timezone, show_date),
 642           s.nick_pattern % l.nick,
 643           l.msg
 644         )
 645         r.append(message)
 646       if r:
 647         self.msg.reply(u'\n'.join(r).encode('utf-8'))
 648       else:
 649         self.msg.reply('没有符合的聊天记录。')
 650     else:
 651       self.msg.reply('Oops, 参数不正确哦。')
 652
 653   def do_set(self, args):
 654     '''设置参数。参数格式 key=value；不带参数以查看说明。'''
 655     #注意：选项名/值中不能包含空格
 656     if len(args) != 1:
 657       doc = []
 658       for c, f in self.__class__.__dict__.items():
 659         if c.startswith('set_'):
 660           doc.append(u'* %s:\t%s' % (c[4:], f.__doc__.decode('utf-8')))
 661       for b in self.__class__.__bases__:
 662         for c, f in b.__dict__.items():
 663           if c.startswith('set_'):
 664             doc.append(u'* %s:\t%s' % (c[4:], f.__doc__.decode('utf-8')))
 665       doc.sort()
 666       doc.insert(0, u'设置选项：')
 667       self.msg.reply(u'\n'.join(doc).encode('utf-8'))
 668     else:
 669       msg = self.msg
 670       cmd = args[0].split('=', 1)
 671       if len(cmd) == 1 or cmd[1] == '':
 672         msg.reply(u'错误：请给出选项值')
 673         return
 674       try:
 675         handle = getattr(self, 'set_' + cmd[0])
 676       except AttributeError:
 677         msg.reply(u'错误：未知选项 %s' % cmd[0])
 678       except IndexError:
 679         msg.reply(u'错误：无选项')
 680       except UnicodeEncodeError:
 681         msg.reply(u'错误：选项名解码失败。此问题在 GAE 升级其 Python 到 3.x 后方能解决。')
 682       else:
 683         handle(cmd[1])
 684
 685   def do_quit(self, args):
 686     '''删除用户数据。某些自称“不作恶”的公司的客户端会不按协议要求发送删除好友的消息，请 gtalk 官方客户端用户使用此命令退出。参见 http://xmpp.org/rfcs/rfc3921.html#rfc.section.6.3 。'''
 687     del_user(self.sender.jid)
 688     self.msg.reply(u'OK.')
 689
 690   def set_prefix(self, arg):
 691     '''设置命令前缀'''
 692     self.sender.prefix = arg
 693     self.sender.put()
 694     self.msg.reply(u'设置成功！')
 695
 696   def set_nickpattern(self, arg):
 697     '''设置昵称显示格式，用 %s 表示昵称的位置'''
 698     try:
 699       arg % 'test'
 700     except (TypeError, ValueError):
 701       self.msg.reply(u'错误：不正确的格式')
 702       return
 703
 704     self.sender.nick_pattern = arg
 705     self.sender.put()
 706     self.msg.reply(u'设置成功！')
 707
 708   def set_allowpm(self, arg):
 709     '''设置是否接收私信，参数为 y（接收）或者 n（拒绝）'''
 710     if arg not in 'yn':
 711       self.msg.reply(u'错误的参数。')
 712       return
 713
 714     if arg == 'y':
 715       self.sender.reject_pm = False
 716     else:
 717       self.sender.reject_pm = True
 718     self.sender.put()
 719     self.msg.reply(u'设置成功！')
 720
 721 class AdminCommand(BasicCommand):
 722   def do_kick(self, args):
 723     '''删除某人。他仍可以重新加入。'''
 724     if len(args) != 1:
 725       self.msg.reply('请给出昵称。')
 726       return
 727
 728     target = get_user_by_nick(args[0])
 729     if target is None:
 730       self.msg.reply('Sorry，查无此人。')
 731       return
 732
 733     if target.jid == config.root:
 734       self.msg.reply('不能删除 root 用户')
 735       return
 736
 737     targetjid = target.jid
 738     targetnick = target.nick
 739     target.delete()
 740     self.msg.reply((u'OK，删除 %s。' % target.nick).encode('utf-8'))
 741     send_to_all_except(self.sender.jid, (u'%s 已被删除。' % target.nick) \
 742                        .encode('utf-8'))
 743     xmpp.send_message(targetjid, u'你已被管理员从此群中删除，请删除该好友。')
 744     log_admin(self.sender, KICK % (targetnick, targetjid))
 745
 746   def do_quiet(self, args):
 747     '''禁言某人，参数为昵称和时间（默认单位秒）'''
 748     if len(args) != 2:
 749       self.msg.reply('请给出昵称和时间。')
 750       return
 751     else:
 752       try:
 753         n = utils.parseTime(args[1])
 754       except ValueError:
 755         self.msg.reply('Sorry，我无法理解你说的时间。')
 756         return
 757
 758     target = get_user_by_nick(args[0])
 759     if target is None:
 760       self.msg.reply('Sorry，查无此人。')
 761       return
 762
 763     target.black_before = datetime.datetime.now() + datetime.timedelta(seconds=n)
 764     target.put()
 765     self.msg.reply(u'OK，禁言 %s %s。' % (target.nick, utils.displayTime(n)))
 766     send_to_all_except((self.sender.jid, target.jid),
 767                        (u'%s 已被禁言 %s。' % (target.nick, utils.displayTime(n))) \
 768                        .encode('utf-8'))
 769     xmpp.send_message(target.jid, u'你已被管理员禁言 %s。' % utils.displayTime(n))
 770     log_admin(self.sender, BLACK % (target.nick, n))
 771
 772   def do_notice(self, arg):
 773     '''发送群通告。只会发给在线的人，包括 snoozing 者。'''
 774     if not arg:
 775       self.msg.reply('请给出群通告的内容。')
 776       return
 777
 778     msg = self.msg.body[len(self.sender.prefix):].split(None, 1)[-1]
 779
 780     l = User.gql('where avail != :1', OFFLINE)
 781     log_admin(self.sender, NOTICE % msg)
 782     for u in l:
 783       try:
 784         xmpp.send_message(u.jid, u'通告：' + msg)
 785       except xmpp.InvalidJidError:
 786         pass
 787
 788   def do_topic(self, args=()):
 789     '''查看或设置群主题'''
 790     grp = get_group_info()
 791     if not args:
 792       if grp is None or not grp.topic:
 793         self.msg.reply(u'没有设置群主题。')
 794       else:
 795         self.msg.reply(grp.topic)
 796     else:
 797       grp = get_group_info()
 798       if grp is None:
 799         grp = Group()
 800       grp.topic = self.msg.body[len(self.sender.prefix):].split(None, 1)[-1]
 801       grp.put()
 802       self.msg.reply(u'群主题已更新。')
 803
 804   def do_admin(self, args):
 805     '''将某人添加为管理员'''
 806     if len(args) != 1:
 807       self.msg.reply(u'请给出昵称。')
 808       return
 809
 810     target = get_user_by_nick(args[0])
 811     if target is None:
 812       self.msg.reply(u'Sorry，查无此人。')
 813       return
 814
 815     if target.is_admin:
 816       self.msg.reply(u'%s 已经是管理员了。' % target.nick)
 817       return
 818
 819     target.is_admin = True
 820     target.put()
 821     send_to_all_except(target.jid,
 822                        (u'%s 已成为管理员。' % target.nick) \
 823                        .encode('utf-8'))
 824     xmpp.send_message(target.jid, u'你已是本群管理员。')
 825     log_admin(self.sender, ADMIN % (target.nick, self.sender.nick))
 826
 827   def do_unadmin(self, args):
 828     '''取消某人管理员的权限'''
 829     if len(args) != 1:
 830       self.msg.reply(u'请给出昵称。')
 831       return
 832
 833     target = get_user_by_nick(args[0])
 834     if target is None:
 835       self.msg.reply(u'Sorry，查无此人。')
 836       return
 837
 838     if not target.is_admin:
 839       self.msg.reply(u'%s 不是管理员。' % target.nick)
 840       return
 841
 842     target.is_admin = False
 843     target.put()
 844     send_to_all_except(target.jid,
 845                        (u'%s 已不再是管理员。' % target.nick) \
 846                        .encode('utf-8'))
 847     xmpp.send_message(target.jid, u'你已不再是本群管理员。')
 848     log_admin(self.sender, UNADMIN % (target.nick, self.sender.nick))
 849
 850   def do_block(self, args):
 851     '''封禁某个 ID，参数为用户昵称或者 ID（如果不是已经加入的 ID 的话），以及封禁原因'''
 852     if len(args) < 2:
 853       self.msg.reply(u'请给出要封禁的用户和原因。')
 854       return
 855
 856     target = get_user_by_nick(args[0])
 857     reason = self.msg.body[len(self.sender.prefix):].split(None, 2)[-1]
 858     if target is None:
 859       jid = args[0]
 860       name = jid
 861       fullname = name
 862     else:
 863       jid = target.jid
 864       name = target.nick
 865       fullname = '%s (%s)' % (name, jid)
 866     u = BlockedUser.gql('where jid = :1', jid).get()
 867     if u is not None:
 868       self.msg.reply(u'此 JID 已经被封禁。')
 869       return
 870
 871     if jid == config.root:
 872       self.msg.reply('不能封禁 root 用户')
 873       return
 874
 875     if target:
 876       target.delete()
 877     u = BlockedUser(jid=jid, reason=reason)
 878     u.put()
 879
 880     send_to_all_except(self.sender.jid,
 881                        (u'%s 已被本群封禁，理由为 %s。' % (name, reason)) \
 882                        .encode('utf-8'))
 883     self.msg.reply(u'%s 已被本群封禁，理由为 %s。' % (fullname, reason))
 884     xmpp.send_message(jid, u'你已被本群封禁，理由为 %s。' % reason)
 885     xmpp.send_presence(jid, status=u'您已经被本群封禁')
 886     log_admin(self.sender, BLOCK % (fullname, reason))
 887
 888   def do_unblock(self, args):
 889     '''解封某个 ID'''
 890     if len(args) != 1:
 891       self.msg.reply(u'请给出要解封用户的 JID。')
 892       return
 893
 894     target = get_blocked_user(args[0])
 895     if target is None:
 896       self.msg.reply(u'封禁列表中没有这个 JID。')
 897       return
 898
 899     target.delete()
 900     send_to_all((u'%s 已被解除封禁。' % args[0]) \
 901                 .encode('utf-8'))
 902     log_admin(self.sender, UNBLOCK % args[0])
 903
 904   def do_groupstatus(self, arg):
 905     '''设置群状态'''
 906     grp = get_group_info()
 907     if grp is None:
 908       grp = Group()
 909     grp.status = self.msg.body[len(self.sender.prefix):].split(None, 1)[-1]
 910     grp.put()
 911     for u in User.all():
 912       xmpp.send_presence(u.jid, status=grp.status)
 913     self.msg.reply(u'设置成功！')