84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
#!/usr/bin/env python2
|
|
# -*- coding: UTF-8 -*-
|
|
# File: msg.py
|
|
# Date: Sun Nov 23 20:44:33 2014 +0800
|
|
# Author: Yuxin Wu <[email protected]>
|
|
|
|
from datetime import datetime
|
|
from bs4 import BeautifulSoup
|
|
from .utils import ensure_bin_str, ensure_unicode
|
|
|
|
TYPE_MSG = 1
|
|
TYPE_IMG = 3
|
|
TYPE_SPEAK = 34
|
|
TYPE_VIDEO = 43
|
|
TYPE_EMOJI = 47
|
|
TYPE_LOCATION = 48
|
|
TYPE_LINK = 49 # link share or file from web
|
|
TYPE_VOIP = 50
|
|
TYPE_SYSTEM = 10000
|
|
|
|
class WeChatMsg(object):
|
|
FIELDS = ["msgSvrId","type","isSend","createTime","talker","content","imgPath"]
|
|
|
|
@staticmethod
|
|
def filter_type(tp):
|
|
if tp in [TYPE_SYSTEM] or tp > 10000 or tp < 0:
|
|
return True
|
|
return False
|
|
|
|
def __init__(self, row):
|
|
""" row: a tuple corresponding to FIELDS"""
|
|
assert len(row) == len(WeChatMsg.FIELDS)
|
|
for f, v in zip(WeChatMsg.FIELDS, row):
|
|
setattr(self, f, v)
|
|
self.createTime = datetime.fromtimestamp(self.createTime / 1000)
|
|
self.talker_name = None
|
|
if self.content:
|
|
self.content = ensure_unicode(self.content)
|
|
|
|
def msg_str(self):
|
|
# TODO: fix more types
|
|
if self.type == TYPE_LOCATION:
|
|
soup = BeautifulSoup(self.content)
|
|
loc = soup.find('location')
|
|
label = loc['label']
|
|
ret = label
|
|
try:
|
|
poiname = loc['poiname']
|
|
if poiname:
|
|
ret = poiname
|
|
except:
|
|
pass
|
|
return ret + " ({},{})".format(loc['x'], loc['y'])
|
|
elif self.type == TYPE_VOIP:
|
|
return "REQUEST VIDEO CHAT"
|
|
elif self.type == TYPE_LINK:
|
|
soup = BeautifulSoup(self.content)
|
|
url = soup.find('url').text
|
|
if not url:
|
|
title = soup.find('title').text
|
|
assert title, "No title or url found in TYPE_LINK"
|
|
return u"FILE:{}".format(title)
|
|
return u"URL:{}".format(url)
|
|
elif self.type == TYPE_VIDEO:
|
|
return "VIDEO"
|
|
else:
|
|
return self.content
|
|
|
|
def __repr__(self):
|
|
ret = u"{}|{}:{}:{}".format(
|
|
self.type,
|
|
(self.talker if not self.talker_name else self.talker_name) \
|
|
if not self.isSend else 'me',
|
|
self.createTime,
|
|
ensure_unicode(self.msg_str())).encode('utf-8')
|
|
if self.imgPath:
|
|
ret = u"{}|img:{}".format(ensure_unicode(ret.strip()), self.imgPath)
|
|
return ret.encode('utf-8')
|
|
else:
|
|
return ret
|
|
|
|
def __lt__(self, r):
|
|
return self.createTime < r.createTime
|