def __init__(self, conf):
qrcodeId = "1"
self.qrcodePath = conf.QrcodePath(qrcodeId)
if conf.httpServerIP:
self.qrcodeServer = QrcodeServer(
conf.httpServerIP,
conf.httpServerPort,
self.qrcodePath,
qrcodeId
)
StartDaemonThread(self.qrcodeServer.Run)
else:
self.qrcodeServer = None
if conf.mailAccount:
self.mailAgent = MailAgent(
conf.mailAccount, conf.mailAuthCode, name='QQBot管理员'
)
if self.qrcodeServer:
html = ('<p>您的 QQBot 正在登录，请尽快用手机 QQ 扫描下面的二维码。'
'若二维码已过期，请重新打开本邮件。若您看不到二维码图片，请确保'
'图片地址 <a href="{0}">{0}</a> 可以通过公网访问。</p>'
'<p><img src="{0}"></p>').format(self.qrcodeServer.qrcodeURL)
else:
html = ('<p>您的 QQBot 正在登录，请尽快用手机 QQ 扫描下面的二维码。'
'若二维码已过期，请将本邮件设为已读邮件，之后 QQBot 会在'
'1~2分钟内将最新的二维码发送到本邮箱。</p>'
'<p>{{png}}</p>')
html += '<p>conf.user=%r, conf.qq=%r</p>' % (conf.user, conf.qq)
self.qrcodeMail = {
'to_addr': conf.mailAccount,
'html': html,
'subject': ('%s[%s]' % ('QQBot二维码', qrcodeId)),
'to_name': '我'
}
self.qrcode = LockedValue(None)
else:
self.mailAgent = None
self.cmdQrcode = conf.cmdQrcode
if self.cmdQrcode:
global Image
try:
from PIL import Image as i
import wcwidth
Image = i
except ImportError:
ERROR('需要安装 pillow,wcwidth 才能使用文本模式显示二维码')
sys.exit(1)
def __init__(self, conf):
qrcodeId = uuid.uuid4().hex
self.qrcodePath = conf.QrcodePath(qrcodeId)
if conf.httpServerIP:
self.qrcodeServer = QrcodeServer(conf.httpServerIP,
conf.httpServerPort,
os.path.dirname(self.qrcodePath))
self.qrcodeURL = self.qrcodeServer.qrcodeURL
else:
self.qrcodeServer = None
if conf.mailAccount:
self.mailAgent = MailAgent(conf.mailAccount,
conf.mailAuthCode,
name='QQBot管理员')
if self.qrcodeServer:
html = ('<p>您的 QQBot 正在登录，请尽快用手机 QQ 扫描下面的二维码。'
'若二维码已过期，请重新打开本邮件。若您看不到二维码图片，请确保'
'图片地址 <a href="{0}">{0}</a> 可以通过公网访问。</p>'
'<p><img src="{0}"></p>').format(self.qrcodeURL)
else:
html = ('<p>您的 QQBot 正在登录，请尽快用手机 QQ 扫描下面的二维码。'
'若二维码已过期，请将本邮件设为已读邮件，之后 QQBot 会在'
'1~2分钟内将最新的二维码发送到本邮箱。</p>'
'<p>{{png}}</p>')
self.qrcodeMail = {
'to_addr': conf.mailAccount,
'html': html,
'subject': ('%s[%s]' % ('QQBot二维码', qrcodeId)),
'to_name': '我'
}
self.qrcode = LockedValue(None)
else:
self.mailAgent = None
self.cmdQrcode = conf.cmdQrcode
class QrcodeManager(object):
def __init__(self, conf):
qrcodeId = uuid.uuid4().hex
self.qrcodePath = conf.QrcodePath(qrcodeId)
if conf.httpServerIP:
self.qrcodeServer = QrcodeServer(conf.httpServerIP,
conf.httpServerPort,
os.path.dirname(self.qrcodePath))
self.qrcodeURL = self.qrcodeServer.QrcodeURL(qrcodeId)
else:
self.qrcodeServer = None
if conf.mailAccount:
self.mailAgent = MailAgent(conf.mailAccount,
conf.mailAuthCode,
name='QQBot管理员')
if self.qrcodeServer:
html = ('<p>您的 QQBot 正在登录，请尽快用手机 QQ 扫描下面的二维码。'
'若二维码已过期，请重新打开本邮件。若您看不到二维码图片，请确保'
'图片地址 <a href="{0}">{0}</a> 可以通过公网访问。</p>'
'<p><img src="{0}"></p>').format(self.qrcodeURL)
else:
html = ('<p>您的 QQBot 正在登录，请尽快用手机 QQ 扫描下面的二维码。'
'若二维码已过期，请将本邮件设为已读邮件，之后 QQBot 会在'
'1~2分钟内将最新的二维码发送到本邮箱。</p>'
'<p>{{png}}</p>')
self.qrcodeMail = {
'to_addr': conf.mailAccount,
'html': html,
'subject': ('%s[%s]' % ('QQBot二维码', qrcodeId)),
'to_name': '我'
}
self.qrcode = LockedValue(None)
else:
self.mailAgent = None
self.cmdQrcode = conf.cmdQrcode
if self.cmdQrcode:
global Image
try:
from PIL import Image as i
import wcwidth
Image = i
except ImportError:
ERROR('需要安装 pillow,wcwidth 才能使用文本模式显示二维码')
sys.exit(1)
def Show(self, qrcode):
with open(self.qrcodePath, 'wb') as f:
f.write(qrcode)
if self.cmdQrcode:
try:
showCmdQRCode(self.qrcodePath)
except Exception as e:
WARN('无法以文本模式显示二维码图片 file://%s 。%s',
SYSTEMSTR2STR(self.qrcodePath), e)
if not (self.qrcodeServer or self.mailAgent or self.cmdQrcode):
try:
showImage(self.qrcodePath)
except Exception as e:
WARN('无法弹出二维码图片 file://%s 。%s', SYSTEMSTR2STR(self.qrcodePath),
e)
if self.qrcodeServer:
INFO('请使用浏览器访问二维码，图片地址： %s', self.qrcodeURL)
if self.mailAgent:
if self.qrcode.getVal() is None:
self.qrcode.setVal(qrcode)
# first show, start a thread to send emails
StartDaemonThread(self.sendEmail)
else:
self.qrcode.setVal(qrcode)
def sendEmail(self):
lastSubject = ''
while True:
if lastSubject != self.qrcodeMail['subject']:
qrcode = self.qrcode.getVal()
if qrcode is None:
break
qrcode = '' if self.qrcodeServer else qrcode
try:
with self.mailAgent.SMTP() as smtp:
smtp.send(png_content=qrcode, **self.qrcodeMail)
except Exception as e:
WARN('无法将二维码发送至邮箱%s %s', self.mailAgent.account, e)
else:
INFO('已将二维码发送至邮箱%s', self.mailAgent.account)
if self.qrcodeServer:
break
else:
lastSubject = self.qrcodeMail['subject']
else:
time.sleep(65)
qrcode = self.qrcode.getVal()
if qrcode is None:
break
try:
DEBUG('开始查询邮箱 %s 中的最近的邮件', self.mailAgent.account)
with self.mailAgent.IMAP() as imap:
lastSubject = imap.getSubject(-1)
except Exception as e:
WARN('查询邮箱 %s 中的邮件失败 %s', self.mailAgent.account, e)
else:
DEBUG('最近的邮件： %s', lastSubject)
def Destroy(self):
if self.mailAgent:
self.qrcode.setVal(None)
try:
os.remove(self.qrcodePath)
except OSError:
pass