# -*- coding: utf-8 -*-

import os

from django_six import CompatibilityBaseCommand

from utils.qiniucdn import upload_file_path


class Command(CompatibilityBaseCommand):
    def handle(self, *args, **options):

        files = os.listdir('media/file')

        for file in files:
            print(file)
            ext = file.split('.')[-1]
            if not ext:
                continue
            if ext.lower() not in ['jpg', 'jpeg', 'png']:
                continue
            try:
                file_path = 'media/file/{}'.format(file)
                if os.path.exists(file_path):
                    upload_file_path(file_path, key=file, bucket='tamron')
                    os.remove(file_path)
            except Exception:
                pass