# -*-coding:utf-8-*-
import itchat
import itchat.content as itcontent

# 登錄網(wǎng)頁微信，hotReload=True 能讓登錄時間加長
itchat.auto_login(hotReload=True)

# 記錄公眾號機器人小冰的UserName
mps = itchat.search_mps(name='小冰')
AI = mps[0]['UserName']
# print(AI)

# 記錄自己的UserName，不然發(fā)送消息會發(fā)兩遍
username = itchat.get_friends()
user = username[0]['UserName']
# print(username)

# 記錄好友列表里好友的 UserName
friendsname = [friend['UserName'] for friend in username if friend['UserName'] != user]
# print(friendsname)

groupname = itchat.get_chatrooms()
groups = [group['UserName'] for group in groupname]


# 這個說來話長~~，有興趣的可以去上網(wǎng)查查
@itchat.msg_register(itcontent.TEXT, isFriendChat=True, isMpChat=True, isGroupChat=True)
def simple_reply(msg, FriendList=[]):
  Fromuser = msg['FromUserName']

  # 如果是AI而且列表不為空，就將AI發(fā)給自己的消息轉(zhuǎn)發(fā)給發(fā)送消息者
  if msg['FromUserName'] == AI and FriendList:
    # print(msg['FromUserName'])
    itchat.send(msg['Text'], toUserName=FriendList[-1])

  elif Fromuser in friendsname:
    if '新年' in msg['Text']:
      return '新年快樂，祝您身體健康，萬事勝意。'

    # 記錄發(fā)送消息者入FriendList中
    elif Fromuser not in FriendList and msg['Text'] == '小小冰真漂亮':
      FriendList.append(Fromuser)
      return '通信建立成功'

    # 第一次發(fā)送消息過來，回復以下內(nèi)容
    elif Fromuser in FriendList:
      if msg['Text'] in ['小小冰再見', '小小冰晚安', '小小冰下次聊']:
        FriendList.remove(Fromuser)
        return '再見，和您聊天十分開心，希望您今天過得愉快！'
      else:
        FriendList.append(Fromuser)
        itchat.send(msg['Text'], toUserName=AI)

    else:
      text = '''Mr.D先生現(xiàn)在不在，我是助手AI，有要事請撥打號碼：xxxxxxxxxxx。如果想和我聊天，那就大聲地說"小小冰真漂亮
            （回復‘小小冰再見/小小冰晚安/小小冰下次聊'可結(jié)束此次聊天。）"'''
      return text

  elif Fromuser in groups:
    if msg.isAt:
      if '新年' in msg['Content']:
        return '新年快樂，祝您身體健康，萬事勝意。'
      elif Fromuser not in FriendList and msg['Content'] == '小小冰真漂亮':
        FriendList.append(Fromuser)
        return '通信建立成功'
      elif Fromuser in FriendList:
        if msg['Content'] in ['小小冰再見', '小小冰晚安', '小小冰下次聊']:
          FriendList.remove(Fromuser)
          return '再見，和您聊天十分開心，希望您今天過得愉快！'
        else:
          FriendList.append(Fromuser)
          itchat.send(msg['Content'], toUserName=AI)
      else:
        text = '''Mr.D先生現(xiàn)在不在，我是助手AI，有要事請撥打號碼：xxxxxxxxxxx。如果想和我聊天，那就大聲地說"小小冰真漂亮
            （回復‘小小冰再見/小小冰晚安/小小冰下次聊'可結(jié)束此次聊天。）"'''
        return text

    elif msg['Text'] == '小小冰真漂亮':
      FriendList.append(Fromuser)
      return '通信建立成功'

    elif Fromuser in FriendList:
      if msg['Text'] in ['小小冰再見', '小小冰晚安', '小小冰下次聊']:
        FriendList.clear()
        return '再見，和您聊天十分開心，希望您今天過得愉快！'
      elif '新年' in msg['Text']:
        return '新年快樂，祝您身體健康，萬事勝意。'
      else:
        FriendList.append(Fromuser)
        itchat.send(msg['Text'], toUserName=AI)

        # 如果是自己發(fā)送消息，則清空列表
  elif Fromuser == user:
    FriendList.clear()

  # 其他公眾號信息，就通知一聲給微信文件助手
  else:
    itchat.send('公眾號信息', toUserName='filehelper')


@itchat.msg_register([itcontent.PICTURE, itcontent.RECORDING, itcontent.VIDEO, itcontent.MAP], isFriendChat=True,
           isGroupChat=True, isMpChat=True)
def return_text(msg):
  text = '我不具備識別語音與圖片等功能，請說普通話。'
  return text


itchat.run()