main 函数重构，添加群助手功能，新版本到：0.3.0

Author: sfyc23

everyday_wechat/__init__.py

@@ -4,3 +4,5 @@
 # Author: snow
 
 
+script_name = 'EverydayWechat'
+__version__ = '0.1.013'

everyday_wechat/_config.yaml

@@ -6,17 +6,22 @@
 is_forced_switch: False
 
 #---------------------------------------自动回复功能设置--------------------------------------start
-auto_relay_info:
+auto_reply_info:
   # 是否开启自动回复
-  is_auto_relay: True
+  is_auto_reply: True
 
-  # 指定自动回复的好友名单。（填：文件传输助手，可回复自己，测试效果）
-  # - '好友1'
-  # - '好友2'
-  auto_reply_names:
-    - '诗风'
+  # 是否自动回复所有好友(慎重启动！),开启后，仅仅有黑名单的成员才不会自动回复
+  is_auto_reply_all: False
+
+  # 白名单，is_auto_reply_all: False 生效。此名单的用户才可自动回复。（填：文件传输助手，可回复自己，测试效果）
+  auto_reply_white_list:
+    - 'happy'
     - '文件传输助手'
 
+  # 黑名单，is_auto_reply_all:True 生效。此名单不会自动回复。
+  auto_reply_black_list:
+    - '小号'
+
   # 机器人渠道（1: 图灵机器人，2: 一个AI ,3 : 青云客，4 腾讯智能闲聊，5:天行机器人,6 海知智能)
   bot_channel: 3
 
@@ -81,29 +86,49 @@ alarm_info:
       horescope: "处女座"
       # 落款（可空），落款参考：['厌物', '你脚下的蚂蚁', '专说骗人的诳话者', '黄天霸', 'cxk', '魔鬼的叔父', '哺乳类脊椎动物之一', '名字写在水上的人', 'BIG BAD WOLF', '你的兄弟']
       sweet_words: '你脚下的蚂蚁'
-
+      is_tomorrow: False
 
     #  如果你需要另一套方案，如下面复制即可
     #  如不需要，则删除或注解下面所有的数据
-#    - group_name: 'EverydayWechat 交流群'
-#      wechat_name: '古典'
-#      city_name: '漳州'
-#      start_date: '2011-11-11'
-#      start_date_msg: '分手后的第{}天'
-#      dictum_channel: 6
-#      sweet_words: '名字写在水上的人'
-#      alarm_timed: "23:02"
+    - group_name: 'EverydayWechat 交流群'
+      wechat_name: '古典'
+      city_name: '漳州'
+      start_date: '2011-11-11'
+      start_date_msg: '分手后的第{}天'
+      dictum_channel: 4
+      sweet_words: '名字写在水上的人'
+      alarm_timed: "23:02"
+      is_tomorrow: True
 #---------------------------------------定时提醒功能设置--------------------------------------end
 
 
 #---------------------------群聊助手设置（已完成，正在整理。下个版本发布）--------------------------start
-auto_reply_group_conf:
+group_helper_conf:
   is_open: True # 开启群助手
-  # 白名单群
-  group_names:
+  is_all: False # 是否对所有群开启。当开启时，只有黑名单的名单才不受影响（慎重开启！）
+
+  # 白名单用户。当 is_all：Fase。只处理这个群里的消息
+  group_name_white_list:
+    - '蕉群'
+    - '群机器人测试群'
+
+  # 黑名单用户。当 is_all ：True 。这个群里的用户不受影响。
+  group_name_black_list:
     - 'EverydayWechat 交流群'
-  # 艾特标记，只有艾特自己，才发送对应消息。
-  is_at: True # 艾特标记
-  weather: True # 是否开启天气查询功能
+
+  is_at: True # 艾特标记。只有当别人艾特机器人，才会处理消息（慎重关闭！）
+  is_auto_reply: False # 开启群自动回复（慎重开启！）
+  is_weather: True # 是否开启天气查询功能
+  is_calendar: True # 是否开启万历年
+  is_rubbish: True # 是否开启垃圾查询（待定TODO）
+
 #---------------------------群聊助手设置（已完成，正在整理。下个版本发布）--------------------------end
 
+
+# start ----------------------------------- 数据库设置 ----------------------------------- start
+db_config: # 请安装 mongodb 数据库
+  is_open_db: True # 是否开启数据库功能
+  mongodb_conf:
+    'host': 'localhost'
+    'port': 27017
+#   end ----------------------------------- 数据库设置 ----------------------------------- end

everyday_wechat/control/bot/ruyiai.py

@@ -22,7 +22,7 @@ def get_ruyiai_bot(text, userId):
     """
     try:
         # config.init()
-        info = config.get('auto_relay_info')['ruyi_conf']
+        info = config.get('auto_reply_info')['ruyi_conf']
         app_key = info['app_key']
         if not app_key:
             print('海知智能 api_key 为空，请求失败')

everyday_wechat/main.py

@@ -15,7 +15,7 @@
 from itchat.content import (
     TEXT
 )
-from everyday_wechat.utils.common import md5_encode
+
 from everyday_wechat.utils.data_collection import (
     get_bot_info,
     get_weather_info,
@@ -25,8 +25,17 @@
     get_constellation_info
 )
 from everyday_wechat.utils import config
+from everyday_wechat.utils.itchat_helper import (
+    init_wechat_config,
+    set_system_notice,
+)
+from everyday_wechat.utils.group_helper import (
+    handle_group_helper
+)
+from everyday_wechat.utils.friend_helper import (
+    handle_friend
+)
 
-reply_userNames = []
 FILEHELPER_MARK = ['文件传输助手', 'filehelper']  # 文件传输助手标识
 FILEHELPER = 'filehelper'
 TIME_COMPILE = re.compile(r'^\s*([01]?[0-9]|2[0-3])\s*[：:\-]\s*([0-5]?[0-9])\s*$')
@@ -69,10 +78,10 @@ def _online():
         return False
 
     hotReload = not config.get('is_forced_switch', False)  # 切换微信号，重新扫码。
-    loginCallback = init_wechat
+    loginCallback = init_data
     exitCallback = exit_msg
     for _ in range(2):  # 尝试登录 2 次。
-        if platform.system() == 'Windows':
+        if platform.system() in ('Windows', 'Darwin'):
             itchat.auto_login(hotReload=hotReload, loginCallback=loginCallback, exitCallback=exitCallback)
             itchat.run(blockThread=True)
         else:
@@ -88,98 +97,25 @@ def _online():
     return False
 
 
-def init_wechat():
+def init_data():
     """ 初始化微信所需数据 """
     set_system_notice('登录成功')
-
-    # conf = get_yaml()
     itchat.get_friends(update=True)  # 更新好友数据。
     itchat.get_chatrooms(update=True)  # 更新群聊数据。
 
-    # 从config copy ，用于保存新的接口内容。
-    myset = config.copy()
-    # start---------------------------处理自动回复好友---------------------------start
-    relay = myset.get('auto_relay_info')
-    if relay.get('is_auto_relay'):
-        auto_reply_uuids = []
-        for name in relay.get('auto_reply_names'):
-            if name.lower() in FILEHELPER_MARK:  # 判断是否文件传输助手
-                if FILEHELPER not in reply_userNames:
-                    auto_reply_uuids.append(FILEHELPER)
-                continue
-            friend = get_friend(name)
-            if friend:
-                auto_reply_uuids.append(friend['UserName'])
-            else:
-                print('自动回复中的好友昵称『{}』有误。'.format(name))
-        relay['auto_reply_uuids'] = set(auto_reply_uuids)
-        print('已开启图灵自动回复...')
-    # end---------------------------处理自动回复好友---------------------------end
-
-    alarm = myset.get('alarm_info')
-    alarm_dict = {}
-    if alarm.get('is_alarm'):
-        for gi in alarm.get('girlfriend_infos'):
-            ats = gi.get('alarm_timed')
-            if not ats:
-                continue
-            uuid_list = []
-            # start---------------------------处理好友---------------------------start
-            friends = gi.get('wechat_name')
-            if isinstance(friends, str):
-                friends = [friends]
-            if isinstance(friends, list):
-                for name in friends:
-                    if name.lower() in FILEHELPER_MARK:  # 判断是否文件传输助手
-                        uuid_list.append(FILEHELPER)
-                        continue
-                    name_info = get_friend(name)
-                    if not name_info:
-                        print('用户昵称{}无效'.format(name))
-                    else:
-                        uuid_list.append(name_info['UserName'])
-            # end---------------------------处理好友---------------------------end
-
-            # start---------------------------群组处理---------------------------start
-            group_names = gi.get('group_name')
-            if isinstance(group_names, str):
-                group_names = [group_names]
-            if isinstance(group_names, list):
-                for name in group_names:
-                    name_info = get_group(name)
-                    if not name_info:
-                        print('定时任务中的群聊名称『{}』有误。'
-                              '(注意：必须要把需要的群聊保存到通讯录)'.format(name))
-                    else:
-                        uuid_list.append(name_info['UserName'])
-            # end---------------------------群组处理---------------------------end
-
-            # start---------------------------定时处理---------------------------start
-            if isinstance(ats, str):
-                ats = [ats]
-            if isinstance(ats, list):
-                for at in ats:
-                    times = TIME_COMPILE.findall(at)
-                    if not times:
-                        print('时间{}格式出错'.format(at))
-                        continue
-                    hour, minute = int(times[0][0]), int(times[0][1])
-                    temp_dict = {'hour': hour, 'minute': minute, 'uuid_list': uuid_list}
-                    temp_dict.update(gi)
-                    alarm_dict[md5_encode(str(temp_dict))] = temp_dict
-        #   end---------------------------定时处理---------------------------end
-        alarm['alarm_dict'] = alarm_dict
-
-    # 将解析的数据保存于config中。
-    config.update(myset)
-    # print(json.dumps(alarm_dict, ensure_ascii=False))
+    init_wechat_config()  # 初始化所有配置内容
 
     # 提醒内容不为空时，启动定时任务
+    alarm_dict = config.get('alarm_info').get('alarm_dict')
     if alarm_dict:
         init_alarm(alarm_dict)  # 初始化定时任务
 
 
 def init_alarm(alarm_dict):
+    """
+    初始化定时任务
+    :param alarm_dict: 定时相关内容
+    """
     # 定时任务
     scheduler = BackgroundScheduler()
     for key, value in alarm_dict.items():
@@ -197,98 +133,40 @@ def send_alarm_msg(key):
 
     gf = conf.get(key)
     # print(gf)
-    calendar_info = get_calendar_info(gf.get('calendar'))
+    is_tomorrow = gf.get('is_tomorrow', False)
+    calendar_info = get_calendar_info(gf.get('calendar'), is_tomorrow)
+    weather = get_weather_info(gf.get('city_name'), is_tomorrow)
+    horoscope = get_constellation_info(gf.get("horescope"), is_tomorrow)
     dictum = get_dictum_info(gf.get('dictum_channel'))
-    weather = get_weather_info(gf.get('city_name'))
     diff_time = get_diff_time(gf.get('start_date'), gf.get('start_date_msg'))
     sweet_words = gf.get('sweet_words')
-    horoscope = get_constellation_info(gf.get("horescope"))
-
     send_msg = '\n'.join(x for x in [calendar_info, weather, horoscope, dictum, diff_time, sweet_words] if x)
     # print('\n' + send_msg + '\n')
-
     if not send_msg or not is_online(): return
-
     uuid_list = gf.get('uuid_list')
     for uuid in uuid_list:
         time.sleep(1)
         itchat.send(send_msg, toUserName=uuid)
     print('\n定时内容:\n{}\n发送成功...\n\n'.format(send_msg))
-
     print('自动提醒消息发送完成...\n')
 
 
 @itchat.msg_register([TEXT])
 def text_reply(msg):
     """ 监听用户消息，用于自动回复 """
-    try:
-        # if not get_yaml().get('is_auto_relay'):
-        #     return
-        conf = config.get('auto_relay_info')
-        if not conf.get('is_auto_relay'):
-            return
-        # print(json.dumps(msg, ensure_ascii=False))
-        # 获取发送者的用户id
-        uuid = FILEHELPER if msg['ToUserName'] == FILEHELPER else msg.fromUserName
-        # 如果用户id是自动回复列表的人员
-        if uuid in conf.get('auto_reply_uuids'):
-            receive_text = msg.text  # 好友发送来的消息内容
-            # 好友叫啥，用于打印
-            nickName = FILEHELPER if uuid == FILEHELPER else msg.user.nickName
-            print('\n{}发来信息：{}'.format(nickName, receive_text))
-            reply_text = get_bot_info(receive_text, uuid)  # 获取自动回复
-            if reply_text:  # 如内容不为空，回复消息
-                time.sleep(random.randint(1, 2))  # 休眠一秒，保安全。想更快的，可以直接注释。
-                reply_text = reply_text if not uuid == FILEHELPER else '机器人回复：' + reply_text
-                itchat.send(reply_text, toUserName=uuid)
-                print('回复{}：{}'.format(nickName, reply_text))
-            else:
-                print('自动回复失败\n')
-    except Exception as exception:
-        print(str(exception))
+    handle_friend(msg)
+    # print(json.dumps(msg, ensure_ascii=False))
 
 
-def set_system_notice(text):
-    """
-    给文件传输助手发送系统日志。
-    :param text:str 日志内容
-    """
-    if text:
-        text = '系统通知：' + text
-        itchat.send(text, toUserName=FILEHELPER)
+@itchat.msg_register([TEXT], isGroupChat=True)
+def text_group(msg):
+    """ 监听用户消息，用于自动回复 """
+    handle_group_helper(msg)
 
 
 def exit_msg():
-    """ 项目中止提醒 """
-    set_system_notice('项目已断开连接')
-
-
-def get_group(gruop_name, update=False):
-    """
-    根据群组名获取群组数据
-    :param wechat_name:str, 群组名
-    :param update: bool 强制更新群组数据
-    :return: obj 单个群组信息
-    """
-    if update: itchat.get_chatrooms(update=True)
-    if not gruop_name: return None
-    groups = itchat.search_chatrooms(name=gruop_name)
-    if not groups: return None
-    return groups[0]
-
-
-def get_friend(wechat_name, update=False):
-    """
-    根据用户名获取用户数据
-    :param wechat_name: str 用户名
-    :param update: bool 强制更新用户数据
-    :return: obj 单个好友信息
-    """
-    if update: itchat.get_friends(update=True)
-    if not wechat_name: return None
-    friends = itchat.search_friends(name=wechat_name)
-    if not friends: return None
-    return friends[0]
+    """ 退出通知 """
+    print('程序已退出')
 
 
 if __name__ == '__main__':

everyday_wechat/utils/common.py

@@ -6,6 +6,9 @@
 import hashlib
 import json
 
+FILEHELPER_MARK = ['文件传输助手', 'filehelper']  # 文件传输助手标识
+FILEHELPER = 'filehelper'
+
 SPIDER_HEADERS = {
     'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; '
                   'WOW64; rv:60.0) Gecko/20100101 Firefox/60.0',
@@ -54,6 +57,8 @@ def is_json(resp):
 
 def md5_encode(text):
     """ 把數據 md5 化 """
+    if not isinstance(text,str):
+        text = str(text)
     md5 = hashlib.md5()
     md5.update(text.encode('utf-8'))
     encodedStr = md5.hexdigest().upper()
@@ -82,5 +87,5 @@ def get_constellation_name(date):
     return None
 
 if __name__ == '__main__':
-    # print (md5_encode('0'))
+    print (md5_encode('aeryou'))
     pass