#!/usr/bin/env python
# -*- encoding: utf-8 -*-
# author: xu3352<xu3352@gmail.com>
# desc: qiniu object upload tools ("blog")

## import
from qiniu import Auth, put_file, etag, urlsafe_base64_encode
import qiniu.config
import os
from time import gmtime, strftime

## =============================================================fields
# Access Key and Secret Key
access_key = 'your_access_key'
secret_key = 'your_secret_key'

# qiniu access object
q = Auth(access_key, secret_key)

# bucket to upload
bucket_name = 'blog'
bucket_url_prefix = "http://on6gnkbff.bkt.clouddn.com"

# other params
blog_uploads="/Users/xuyinglong/Desktop/blog_uploads"
backup="/Users/xuyinglong/coding-python/blog/upload_backup"
logfile="/Users/xuyinglong/coding-python/blog/output.log"

## =============================================================functions
# backup file
def backup_files(f, t):
    # move file from blog_uploads to backup directory
    os.rename(blog_uploads + "/" + f, backup + "/" + t)

# write log
def log(f, info):
    output = open(logfile, "a")
    currtime = strftime("%Y-%m-%d %H:%M:%S", gmtime())
    output.write(currtime + " - file:" + f + " - info:" + info + "\n")
    output.close()

# postfix filter
def filter(name):
    return name.endswith('.png') or name.endswith(".jpg") or name.endswith(".zip")

# main
def upload_main():
    # jpg|png files in blog_uploads directory.
    imgfiles = [name for name in os.listdir(blog_uploads) if filter(name)]

    print("file count:"+ str(len(imgfiles)) )   # output console

    for f in imgfiles:
        currtime = strftime("%Y%m%d%H%M%S", gmtime())
        # key = filename with postfix
        key = currtime + "_" + f
        
        #gen a token
        token = q.upload_token(bucket_name, key, 3600)

        #upload local file with token
        ret, info = put_file(token, key, blog_uploads + '/' + f)

        url = bucket_url_prefix + "/" + key
        print(url)  # output console

        # backup file 
        backup_files(f, key)

        # log url to file
        log(f, url)

## =============================================================main
# call main
upload_main()