Esempio n. 1
0
import os
import urllib2
import commands
from Queue import Queue
from lib.ParseConfig import Config
from lib.log import logger_root
from lib.UrlMerge import URLMerge
from lib.mailtome import send_mail

# 本地部署API调用URL,需根据实际URL配置
api_get_prefix = 'http://xxx/videos'  # 下载列表API前缀

# 调用Config类,解析配置文件中的配置
config = Config()
school_id = config.get_school_id()  # 学校ID
prefixs = config.get_prefixs()  # mp4,gif,srt下载连接前缀,目前是七牛空间的
dirs = config.get_dirs()  # mp4,gif,srt各类文件下载后存放路径

api_get_url = api_get_prefix + '/' + school_id + '/api'  # 当前学校的下载列表API链接


def init_queue(url_dicts):
    queue = Queue()
    for fk in url_dicts.keys():
        for dk in url_dicts[fk].keys():
            dir = dirs[fk + '_dir'] + '/' + dk + '/'  # 实际文件存放路径
            if not os.path.exists(dir):
                os.makedirs(dir)
            for url in url_dicts[fk][dk]:
                queue.put((dir, url))  # 生成下载队列
    return queue