# from __future__ import unicode_literals, division
import time, datetime, requests, itchat
from itchat.content import *

"""
Explanation: 如何使用和开发微信聊天机器人的系列教程
A workshop to develop & use an intelligent and interactive chat-bot in WeChat
WeChat is a popular social media app, which has more than 800 million monthly active users.
<img src='http://www.kudosdata.com/wp-content/uploads/2016/11/cropped-KudosLogo1.png' width=30% style="float: right;">
<img src='reference/WeChat_SamGu_QR.png' width=10% style="float: right;">
http://www.KudosData.com
by: Sam.Gu@KudosData.com
April 2017 ========== Scan the QR code to become trainer's friend in WeChat ========>>
第一课：使用微信问答机制
Lesson 1: Basic usage of WeChat Python API

使用和开发微信个人号聊天机器人：一种Python编程接口 (Use WeChat Python API)
用微信App扫QR码图片来自动登录 (Log-in, contact scan, and processing of text, image, file, video, etc)
查找指定联系人或群组 (Scan ccontact list)
发送信息（文字、图片、文件、音频、视频等） (Send message: text, image, file, voice, video, etc)
接收信息 (Receive message, and keep 'listening')
自动回复 (Receive message and then automaticaly reply)
自定义复杂消息处理，例如：信息存档、回复群组中被@的消息 (Advanced message processing and reply)

导入需要用到的一些功能程序库：
End of explanation
"""


itchat.auto_login(hotReload=True) # hotReload=True: 退出程序后暂存登陆状态。即使程序关闭，一定时间内重新开启也可以不用重新扫码。
# itchat.auto_login(enableCmdQR=-2) # enableCmdQR=-2: 命令行显示QR图片

"""
Explanation: * 用微信App扫QR码图片来自动登录
End of explanation
"""


# 获取自己的用户信息，返回自己的属性字典
friend = itchat.search_friends()
print(friend)

print('NickName  : %s' % friend['NickName'])
print('Alias A-ID: %s' % friend['Alias'])
print('RemarkName: %s' % friend['RemarkName'])
print('UserName  : %s' % friend['UserName'])

# 获取任何一项等于name键值的用户。
# 'NickName' 昵称, set by that friend, changeable
# 'Alias' ID微信号 = wechatAccount, one time set by that friend, cannot change
# 'RemarkName' 备注名, set by current login account owner, changeable by login account owner
# 注意：返回可能包含多个朋友。为什么呢？

friend = itchat.search_friends(name=u'Sam Gu')
# friend = itchat.search_friends(name=u'Mr. R')
# friend = itchat.search_friends(name=u'Ms. S')

for i in range(0, len(friend)):
    print('NickName  : %s' % friend[i]['NickName'])
    print('Alias A-ID: %s' % friend[i]['Alias'])
    print('RemarkName: %s' % friend[i]['RemarkName'])
    print('UserName  : %s' % friend[i]['UserName'])

# 获取分别对应相应键值的用户。

# friend = itchat.search_friends(nickName=u'Sam Gu')
# friend = itchat.search_friends(wechatAccount=u'Sam Gu')
friend = itchat.search_friends(remarkName=u'Sam Gu')
# friend = itchat.search_friends(userName=u'Sam Gu')

for i in range(0, len(friend)):
    print('NickName  : %s' % friend[i]['NickName'])
    print('Alias A-ID: %s' % friend[i]['Alias'])
    print('RemarkName: %s' % friend[i]['RemarkName'])
    print('UserName  : %s' % friend[i]['UserName'])

# 查找群组
# group = itchat.search_chatrooms(name=u'Data Science')
group = itchat.search_chatrooms(name=u'陪聊妹UAT')

for i in range(0, len(group)):
    print('NickName  : %s' % group[i]['NickName'])
    print('Alias A-ID: %s' % group[i]['Alias'])
    print('RemarkName: %s' % group[i]['RemarkName'])
    print('UserName  : %s' % group[i]['UserName'])
    print('Is Owner? : %s ( 0 for No | 1 for Yes )' % group[0]['IsOwner'])
    print('Is Admin? : %s' % group[i]['IsAdmin'])
    print('')

"""
Explanation: * 查找指定联系人或群组
使用search_friends方法可以搜索用户，有几种搜索方式： 
1.仅获取自己的用户信息 
2.获取昵称'NickName'、微信号'Alias'、备注名'RemarkName'中的任何一项等于name键值的用户 
3.获取分别对应相应键值的用户
End of explanation
"""


# 文字
reply = itchat.send(u'别来无恙啊！\n发送时间:\n{:%Y-%b-%d %H:%M:%S}'.format(datetime.datetime.now()), friend[0]['UserName']) 
print(reply['BaseResponse']['ErrMsg'])

# 图片
reply = itchat.send_image('./reference/WeChat_SamGu_QR.png', friend[0]['UserName']) 
print(reply['BaseResponse']['ErrMsg'])

# 文件
reply = itchat.send_file('./reference/logo.pdf', friend[0]['UserName']) 
print(reply['BaseResponse']['ErrMsg'])

# 音频（语音可以先转成MP3）
reply = itchat.send_file('./reference/audio.mp3', friend[0]['UserName']) 
print(reply['BaseResponse']['ErrMsg'])

# 视频
reply = itchat.send_video('./reference/video.mp4', friend[0]['UserName']) 
print(reply['BaseResponse']['ErrMsg'])

# 发送信息去群组： group[0]['UserName']
# 文字
reply = itchat.send(u'别来无恙啊！\n发送时间:\n{:%Y-%b-%d %H:%M:%S}'.format(datetime.datetime.now()), group[0]['UserName']) 
print(reply['BaseResponse']['ErrMsg'])

"""
Explanation: * 发送信息（文字、图片、文件、音频、视频等）
End of explanation
"""


# itchat.auto_login(hotReload=True) # hotReload=True: 退出程序后暂存登陆状态。即使程序关闭，一定时间内重新开启也可以不用重新扫码。

@itchat.msg_register(itchat.content.TEXT)
def text_reply(msg):
    print(msg['Text'])

# 长期有效地运行（术语叫做：开始监听）
itchat.run()

"""
Explanation: * 接收信息
显示发给自己的文本消息：
End of explanation
"""


# interupt, then re-login
itchat.auto_login(hotReload=True)

@itchat.msg_register(itchat.content.TEXT)
def text_reply(msg):
    print(msg['Text'])
    return u'谢谢亲[嘴唇]我收到 I received:\n' + msg['Text']

itchat.run()

"""
Explanation: 回复发给自己的文本消息：
End of explanation
"""


# interupt, then re-login
itchat.auto_login(hotReload=True)

# 如果收到[TEXT, MAP, CARD, NOTE, SHARING]类的信息，会自动回复：
@itchat.msg_register([TEXT, MAP, CARD, NOTE, SHARING]) # 文字、位置、名片、通知、分享
def text_reply(msg):
    itchat.send('%s: %s' % (msg['Type'], msg['Text']), msg['FromUserName'])

# 如果收到[PICTURE, RECORDING, ATTACHMENT, VIDEO]类的信息，会自动保存：
@itchat.msg_register([PICTURE, RECORDING, ATTACHMENT, VIDEO]) # 图片、语音、文件、视频
def download_files(msg):
    msg['Text'](msg['FileName'])
    return '@%s@%s' % ({'Picture': 'img', 'Video': 'vid'}.get(msg['Type'], 'fil'), msg['FileName'])

# 如果收到新朋友的请求，会自动通过验证添加加好友，并主动打个招呼：幸会幸会！Nice to meet you!
@itchat.msg_register(FRIENDS)
def add_friend(msg):
    itchat.add_friend(**msg['Text']) # 该操作会自动将新好友的消息录入，不需要重载通讯录
    itchat.send_msg(u'幸会幸会！Nice to meet you!', msg['RecommendInfo']['UserName'])

# 在群里，如果收到@自己的文字信息，会自动回复：
@itchat.msg_register(TEXT, isGroupChat=True)
def text_reply(msg):
    if msg['isAt']:
        itchat.send(u'@%s\u2005I received: %s' % (msg['ActualNickName'], msg['Content']), msg['FromUserName'])


itchat.run()

# interupt, then logout
itchat.logout() # 安全退出

"""
Explanation: * 自定义复杂消息处理，例如：信息存档、回复群组中被@的消息
End of explanation
"""


