# -*- 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):

        dirs = ['media/file', 'media/user/complement_code']

        for dir in dirs:
            files = os.listdir(dir)

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