# -*- coding:UTF-8 -*-
import itchat
from itchat.content import TEXT, SHARING, SYSTEM
import pymysql, time, pdb, json, logging
from db.mysql import MysqlDB
from simple_settings import settings
DEBUG = settings.DEBUG
logger = logging.getLogger()
robotId = 0
mysql_pool = MysqlDB()
itchat.auto_login(hotReload=True)
# Get all info about groups which are saved in the groupchat list
GroupInfo = itchat.get_chatrooms()
connection = mysql_pool.connection()
cur = connection.cursor()
for group in GroupInfo:
# get member list for each group
query_group = "INSERT INTO group_info (name, number_members, user_name) VALUES (%s, %s, %s)"
cur.execute(query_group,
(group['NickName'], group['MemberCount'], group['UserName']))
# logger.debug(group)
detailedChatroom = itchat.update_chatroom(group['UserName'],
detailedMember=True)
memberlist_group = detailedChatroom['MemberList']
for member in memberlist_group:
jsonMem = json.dumps(member)
# logger.debug(jsonMem)
# query = "INSERT INTO group_member_info(member_num, group_name, member_info) VALUES(%s, %s, %s)"
query_member = "INSERT INTO member_info(user_name, nick_name, province,city, keyword,headimgurl,snsflag, sex, pyinitial, attrstatus) VALUES(%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)"
cur.execute(query_member,
import logging
import json
from modules.authentication import Authentication
from modules.robot import Robot
from modules.slack import Slack
from modules.friend import Friend
import _thread
DEBUG = settings.DEBUG
logger = logging.getLogger(__name__)
robotId = 0
robotName = ""
robot = None
mysql_pool = MysqlDB()
def loginRobot():
logger.info("Robot login")
Slack.sendLoginNotice("Robot *{}* had login.".format(robotName, robotId))
sql = "insert into temp (temp) values (%s) "
connection = mysql_pool.connection()
cur = connection.cursor()
cur.execute(sql, ("Robot login"))
robot.releaseLoginRobot(robotId)