Download.py 文件源码

python
阅读 24 收藏 0 点赞 0 评论 0

项目:alfred-workflows 作者: mttjhn 项目源码 文件源码
def downloadFile():
    if len(sys.argv) > 1:
        query = sys.argv[1]
    else:
        query = None
    url = query + '+'
    response = urllib2.urlopen(url)
    # Parse URL for the file extension
    returnUrl = response.geturl()
    if 'filename=' in returnUrl:
        # Looks like there's a filename in the return URL!
        nS = returnUrl.find('filename=')+9
        nE = returnUrl.find('&', nS)
        urlFileName = urllib2.unquote(returnUrl[nS:nE])
        eS = urlFileName.rfind('.') + 1
        extension = urlFileName[eS:]
        # Let's infer the type from the name
        type = ''
        # Check to see if it's a screencast
        if 'Capture' in urlFileName:
            type = 'screencast'
        elif 'Shot' not in urlFileName:
            type = 'file'
    else:
        # If we can't get the file name, assume it's a PNG
        extension = 'png'
        type = ''
    fileName = getNextFileName(extension, type)
    with open(fileName, 'wr') as file:
        file.write(response.read())

# Run the file download method!
评论列表


问题


面经


文章

微信
公众号

扫码关注公众号