#!/usr/bin/env python3
# coding: utf-8

import logging

from wxpy import *

logger = logging.getLogger()
logger.setLevel(logging.INFO)

robot = Robot('wxpy.pkl')
group = ensure_one(robot.groups().search('wxpy 交流群'))


@robot.register(msg_types=(FRIENDS, TEXT))
def add_to_group(msg):

    if not isinstance(msg.chat, (Group, MP)) and 'wxpy' in msg.text.lower():
        logger.info('received: {}'.format(msg))
        if msg.type == FRIENDS:
            logger.info('accepting user: {}'.format(msg.card))
            robot.accept_friend(msg.card)
            user = msg.card
        else:
            user = msg.chat

        if user in group:
            logger.info('{} is already in {}'.format(user, group))
            return '你已经加入 {} 啦'.format(group.nick_name)
        else:
            logger.info('inviting {} to {}'.format(user, group))
            group.add_members(user, use_invitation=True)


robot.start()