def send_email(report_name):
cf = Config()
logger = Log.get_logger()
report_file = os.path.join(cf.get_report_dir(), report_name)
# 2. 配置email正文MIMEText
with open(report_file, 'rb') as f:
email_body = f.read()
msg = MIMEText(email_body, 'html', 'utf-8')
# 3. 配置email头部信息
msg['Subject'] = Header(cf.get_email('subject'), 'utf-8')
msg['From'] = cf.get_email('smtp_user')
msg['To'] = cf.get_email('receiver').replace(",", ";")
# 4. 连接smtp服务器，发送邮件
try:
smtp = smtplib.SMTP()
smtp.connect(cf.get_email('smtp_host'))
smtp.login(cf.get_email('smtp_user'), cf.get_email('smtp_passwd'))
for receiver in cf.get_email('receiver').split(","):
smtp.sendmail(cf.get_email('smtp_user'), receiver.strip(), msg.as_string())
smtp.quit()
logger.info("测试报告邮件已发送")
except Exception as ex:
logger.error(str(ex))
import cv2
import glob
import json
import numpy as np
import os
import pyyolo
from util.log import Log
LOG = Log.get_logger(log_level='i')
class Model(object):
def __init__(self):
LOG.info('init model...')
conf_file = os.path.join(os.path.dirname(os.path.abspath(__file__)),
'conf.json')
assert os.path.exists(conf_file), '{} config file not exists, please' \
'copy it from conf.json.tmpl'
with open(conf_file) as f:
self.params = json.load(f)
def _init_model(self):
LOG.info('init general python yolo model')
pyyolo.init(self.params['darknet_path'],
os.path.join('../', self.params['datacfg']),
os.path.join('../', self.params['cfgfile']),
self.params['weightfile'])
@staticmethod
def _init_frame():
"""
import os
import sys
import time
import Queue
import urllib
import threading
sys.path.append("..")
from util.log import Log
from util.config import Config
from util.config import project_dir
cf = Config()
log = Log()
logger = log.get_logger()
class LinkConsumer(threading.Thread):
def __init__(self, name, data_queue):
threading.Thread.__init__(self, name=name)
self.data_queue = data_queue
def run(self):
while not self.data_queue.empty():
link = self.data_queue.get()
self._save(link)
def _save(self, link):
output_dir = os.path.join(project_dir,
cf.get('output_directory')).strip()
def __init__(self):
self.logger = Log.get_logger()
self.http = Http()
self.db = DB()
def __init__(self):
self.cf = Config()
self.timeout=float(self.cf.get_http('time_out'))
self.logger = Log.get_logger()
self.env = self.url=''
self.headers = self.params = self.data = self.files = {}