|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102 |
- import pickle, os
- import logging
-
- import requests
-
- from ..config import VERSION
- from ..returnvalues import ReturnValue
- from ..storage import templates
- from .contact import update_local_chatrooms, update_local_friends
- from .messages import produce_msg
-
- logger = logging.getLogger('itchat')
-
- def load_hotreload(core):
- core.dump_login_status = dump_login_status
- core.load_login_status = load_login_status
-
- async def dump_login_status(self, fileDir=None):
- fileDir = fileDir or self.hotReloadDir
- try:
- with open(fileDir, 'w') as f:
- f.write('itchat - DELETE THIS')
- os.remove(fileDir)
- except:
- raise Exception('Incorrect fileDir')
- status = {
- 'version' : VERSION,
- 'loginInfo' : self.loginInfo,
- 'cookies' : self.s.cookies.get_dict(),
- 'storage' : self.storageClass.dumps()}
- with open(fileDir, 'wb') as f:
- pickle.dump(status, f)
- logger.debug('Dump login status for hot reload successfully.')
-
- async def load_login_status(self, fileDir,
- loginCallback=None, exitCallback=None):
- try:
- with open(fileDir, 'rb') as f:
- j = pickle.load(f)
- except Exception as e:
- logger.debug('No such file, loading login status failed.')
- return ReturnValue({'BaseResponse': {
- 'ErrMsg': 'No such file, loading login status failed.',
- 'Ret': -1002, }})
-
- if j.get('version', '') != VERSION:
- logger.debug(('you have updated itchat from %s to %s, ' +
- 'so cached status is ignored') % (
- j.get('version', 'old version'), VERSION))
- return ReturnValue({'BaseResponse': {
- 'ErrMsg': 'cached status ignored because of version',
- 'Ret': -1005, }})
- self.loginInfo = j['loginInfo']
- self.loginInfo['User'] = templates.User(self.loginInfo['User'])
- self.loginInfo['User'].core = self
- self.s.cookies = requests.utils.cookiejar_from_dict(j['cookies'])
- self.storageClass.loads(j['storage'])
- try:
- msgList, contactList = self.get_msg()
- except:
- msgList = contactList = None
- if (msgList or contactList) is None:
- self.logout()
- await load_last_login_status(self.s, j['cookies'])
- logger.debug('server refused, loading login status failed.')
- return ReturnValue({'BaseResponse': {
- 'ErrMsg': 'server refused, loading login status failed.',
- 'Ret': -1003, }})
- else:
- if contactList:
- for contact in contactList:
- if '@@' in contact['UserName']:
- update_local_chatrooms(self, [contact])
- else:
- update_local_friends(self, [contact])
- if msgList:
- msgList = produce_msg(self, msgList)
- for msg in msgList: self.msgList.put(msg)
- await self.start_receiving(exitCallback)
- logger.debug('loading login status succeeded.')
- if hasattr(loginCallback, '__call__'):
- await loginCallback(self.storageClass.userName)
- return ReturnValue({'BaseResponse': {
- 'ErrMsg': 'loading login status succeeded.',
- 'Ret': 0, }})
-
- async def load_last_login_status(session, cookiesDict):
- try:
- session.cookies = requests.utils.cookiejar_from_dict({
- 'webwxuvid': cookiesDict['webwxuvid'],
- 'webwx_auth_ticket': cookiesDict['webwx_auth_ticket'],
- 'login_frequency': '2',
- 'last_wxuin': cookiesDict['wxuin'],
- 'wxloadtime': cookiesDict['wxloadtime'] + '_expired',
- 'wxpluginkey': cookiesDict['wxloadtime'],
- 'wxuin': cookiesDict['wxuin'],
- 'mm_lang': 'zh_CN',
- 'MM_WX_NOTIFY_STATE': '1',
- 'MM_WX_SOUND_STATE': '1', })
- except:
- logger.info('Load status for push login failed, we may have experienced a cookies change.')
- logger.info('If you are using the newest version of itchat, you may report a bug.')
|