commands/management/commands/qiniuonce.py

@@ -10,19 +10,22 @@ 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
              

                +        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']:
              

                +                    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